Prevent FCM bottlenecking.

app/src/main/java/org/thoughtcrime/securesms/gcm/FcmFetchService.java

@@ -0,0 +1,92 @@
+package org.thoughtcrime.securesms.gcm;
+
+import android.app.Service;
+import android.content.Intent;
+import android.os.Build;
+import android.os.IBinder;
+
+import androidx.annotation.Nullable;
+
+import com.google.firebase.messaging.RemoteMessage;
+
+import org.thoughtcrime.securesms.dependencies.ApplicationDependencies;
+import org.thoughtcrime.securesms.jobs.PushNotificationReceiveJob;
+import org.thoughtcrime.securesms.logging.Log;
+import org.thoughtcrime.securesms.util.concurrent.SerialMonoLifoExecutor;
+import org.thoughtcrime.securesms.util.concurrent.SignalExecutors;
+
+import java.util.concurrent.atomic.AtomicInteger;
+
+/**
+ * This service does the actual network fetch in response to an FCM message.
+ *
+ * Our goals with FCM processing are as follows:
+ * (1) Ensure some service is active for the duration of the fetch and processing stages.
+ * (2) Do not make unnecessary network requests.
+ *
+ * To fulfill goal 1, this service will not call {@link #stopSelf()} until there is no more running
+ * requests.
+ *
+ * To fulfill goal 2, this service will not enqueue a fetch if there are already 2 active fetches
+ * (or rather, 1 active and 1 waiting, since we use a single thread executor).
+ *
+ * Unfortunately we can't do this all in {@link FcmReceiveService} because it won't let us process
+ * the next FCM message until {@link FcmReceiveService#onMessageReceived(RemoteMessage)} returns,
+ * but as soon as that method returns, it could also destroy the service. By not letting us control
+ * when the service is destroyed, we can't accomplish both goals within that service.
+ */
+public class FcmFetchService extends Service {
+
+  private static final String TAG = Log.tag(FcmFetchService.class);
+
+  private static final SerialMonoLifoExecutor EXECUTOR = new SerialMonoLifoExecutor(SignalExecutors.UNBOUNDED);
+
+  private final AtomicInteger activeCount = new AtomicInteger(0);
+
+  @Override
+  public int onStartCommand(Intent intent, int flags, int startId) {
+    boolean performedReplace = EXECUTOR.enqueue(this::fetch);
+
+    if (performedReplace) {
+      Log.i(TAG, "Already have one running and one enqueued. Ignoring.");
+    } else {
+      int count = activeCount.incrementAndGet();
+      Log.i(TAG, "Incrementing active count to " + count);
+    }
+
+    return START_NOT_STICKY;
+  }
+
+  @Override
+  public void onDestroy() {
+    Log.i(TAG, "onDestroy()");
+  }
+
+  @Override
+  public @Nullable IBinder onBind(Intent intent) {
+    return null;
+  }
+
+  private void fetch() {
+    MessageRetriever retriever = ApplicationDependencies.getMessageRetriever();
+    boolean          success   = retriever.retrieveMessages(this, new RestStrategy(), new RestStrategy());
+
+    if (success) {
+      Log.i(TAG, "Successfully retrieved messages.");
+    } else {
+      if (Build.VERSION.SDK_INT >= 26) {
+        Log.w(TAG, "Failed to retrieve messages. Scheduling on the system JobScheduler (API " + Build.VERSION.SDK_INT + ").");
+        FcmJobService.schedule(this);
+      } else {
+        Log.w(TAG, "Failed to retrieve messages. Scheduling on JobManager (API " + Build.VERSION.SDK_INT + ").");
+        ApplicationDependencies.getJobManager().add(new PushNotificationReceiveJob(this));
+      }
+    }
+
+    if (activeCount.decrementAndGet() == 0) {
+      Log.e(TAG, "stopping");
+      stopSelf();
+    }
+  }
+}
+

app/src/main/java/org/thoughtcrime/securesms/gcm/FcmJobService.java

@@ -9,16 +9,14 @@ import android.content.Context;
 import androidx.annotation.NonNull;
 import androidx.annotation.RequiresApi;
 
-import org.thoughtcrime.securesms.ApplicationContext;
 import org.thoughtcrime.securesms.dependencies.ApplicationDependencies;
 import org.thoughtcrime.securesms.logging.Log;
-import org.thoughtcrime.securesms.notifications.MessageNotifier;
 import org.thoughtcrime.securesms.util.ServiceUtil;
 import org.thoughtcrime.securesms.util.TextSecurePreferences;
 import org.thoughtcrime.securesms.util.concurrent.SignalExecutors;
 
 /**
- * Pulls down messages. Used when we fail to pull down messages in {@link FcmService}.
+ * Pulls down messages. Used when we fail to pull down messages in {@link FcmReceiveService}.
  */
 @RequiresApi(26)
 public class FcmJobService extends JobService {

app/src/main/java/org/thoughtcrime/securesms/gcm/FcmReceiveService.java

@@ -1,25 +1,22 @@
 package org.thoughtcrime.securesms.gcm;
 
 import android.content.Context;
-import android.os.Build;
+import android.content.Intent;
 
 import androidx.annotation.NonNull;
 
 import com.google.firebase.messaging.FirebaseMessagingService;
 import com.google.firebase.messaging.RemoteMessage;
 
-import org.thoughtcrime.securesms.ApplicationContext;
 import org.thoughtcrime.securesms.dependencies.ApplicationDependencies;
 import org.thoughtcrime.securesms.jobs.FcmRefreshJob;
-import org.thoughtcrime.securesms.jobs.PushNotificationReceiveJob;
 import org.thoughtcrime.securesms.logging.Log;
 import org.thoughtcrime.securesms.registration.PushChallengeRequest;
 import org.thoughtcrime.securesms.util.TextSecurePreferences;
-import org.thoughtcrime.securesms.util.concurrent.SignalExecutors;
 
-public class FcmService extends FirebaseMessagingService {
+public class FcmReceiveService extends FirebaseMessagingService {
 
-  private static final String TAG = FcmService.class.getSimpleName();
+  private static final String TAG = FcmReceiveService.class.getSimpleName();
 
   @Override
   public void onMessageReceived(RemoteMessage remoteMessage) {
@@ -46,22 +43,7 @@ public class FcmService extends FirebaseMessagingService {
   }
 
   private static void handleReceivedNotification(Context context) {
-    MessageRetriever retriever = ApplicationDependencies.getMessageRetriever();
-    boolean          success   = retriever.retrieveMessages(context, new RestStrategy(), new RestStrategy());
-
-    if (success) {
-      Log.i(TAG, "Successfully retrieved messages.");
-    } else {
-      if (Build.VERSION.SDK_INT >= 26) {
-        Log.w(TAG, "Failed to retrieve messages. Scheduling on the system JobScheduler (API " + Build.VERSION.SDK_INT + ").");
-        FcmJobService.schedule(context);
-      } else {
-        Log.w(TAG, "Failed to retrieve messages. Scheduling on JobManager (API " + Build.VERSION.SDK_INT + ").");
-        ApplicationDependencies.getJobManager().add(new PushNotificationReceiveJob(context));
-      }
-    }
-
-    Log.i(TAG, "Processing complete.");
+    context.startService(new Intent(context, FcmFetchService.class));
   }
 
   private static void handlePushChallenge(@NonNull String challenge) {
@@ -69,4 +51,4 @@ public class FcmService extends FirebaseMessagingService {
 
     PushChallengeRequest.postChallengeResponse(challenge);
   }
-}
+}