Skip to content

Commit 0b1ec1e

Browse files
committed
Make push notification senders available via CommandDependencies
1 parent 2744d33 commit 0b1ec1e

File tree

5 files changed

+12
-12
lines changed

5 files changed

+12
-12
lines changed

service/src/main/java/org/whispersystems/textsecuregcm/workers/CommandDependencies.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@ record CommandDependencies(
7575
MessagesManager messagesManager,
7676
ClientPresenceManager clientPresenceManager,
7777
KeysManager keysManager,
78+
APNSender apnSender,
79+
FcmSender fcmSender,
7880
PushNotificationManager pushNotificationManager,
7981
PushNotificationExperimentSamples pushNotificationExperimentSamples,
8082
FaultTolerantRedisCluster cacheCluster,
@@ -254,6 +256,7 @@ static CommandDependencies build(
254256
configuration.getDynamoDbTables().getPushNotificationExperimentSamples().getTableName(),
255257
Clock.systemUTC());
256258

259+
environment.lifecycle().manage(apnSender);
257260
environment.lifecycle().manage(messagesCache);
258261
environment.lifecycle().manage(clientPresenceManager);
259262
environment.lifecycle().manage(new ManagedAwsCrt());
@@ -266,6 +269,8 @@ static CommandDependencies build(
266269
messagesManager,
267270
clientPresenceManager,
268271
keys,
272+
apnSender,
273+
fcmSender,
269274
pushNotificationManager,
270275
pushNotificationExperimentSamples,
271276
cacheCluster,

service/src/main/java/org/whispersystems/textsecuregcm/workers/ScheduledApnPushNotificationSenderServiceCommand.java

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,15 @@
55

66
package org.whispersystems.textsecuregcm.workers;
77

8-
import static com.codahale.metrics.MetricRegistry.name;
9-
108
import io.dropwizard.core.Application;
119
import io.dropwizard.core.cli.ServerCommand;
1210
import io.dropwizard.core.server.DefaultServerFactory;
1311
import io.dropwizard.core.setup.Environment;
1412
import io.dropwizard.jetty.HttpsConnectorFactory;
15-
import java.util.concurrent.ExecutorService;
1613
import net.sourceforge.argparse4j.inf.Namespace;
1714
import net.sourceforge.argparse4j.inf.Subparser;
1815
import org.whispersystems.textsecuregcm.WhisperServerConfiguration;
1916
import org.whispersystems.textsecuregcm.metrics.MetricsUtil;
20-
import org.whispersystems.textsecuregcm.push.APNSender;
21-
import org.whispersystems.textsecuregcm.push.FcmSender;
2217
import org.whispersystems.textsecuregcm.push.PushNotificationScheduler;
2318
import org.whispersystems.textsecuregcm.util.logging.UncaughtExceptionHandler;
2419

@@ -73,15 +68,9 @@ protected void run(Environment environment, Namespace namespace, WhisperServerCo
7368
});
7469
}
7570

76-
final ExecutorService pushNotificationSenderExecutor = environment.lifecycle().executorService(name(getClass(), "apnSender-%d"))
77-
.maxThreads(1).minThreads(1).build();
78-
79-
final APNSender apnSender = new APNSender(pushNotificationSenderExecutor, configuration.getApnConfiguration());
80-
final FcmSender fcmSender = new FcmSender(pushNotificationSenderExecutor, configuration.getFcmConfiguration().credentials().value());
8171
final PushNotificationScheduler pushNotificationScheduler = new PushNotificationScheduler(
82-
deps.pushSchedulerCluster(), apnSender, fcmSender, deps.accountsManager(), namespace.getInt(WORKER_COUNT), namespace.getInt(MAX_CONCURRENCY));
72+
deps.pushSchedulerCluster(), deps.apnSender(), deps.fcmSender(), deps.accountsManager(), namespace.getInt(WORKER_COUNT), namespace.getInt(MAX_CONCURRENCY));
8373

84-
environment.lifecycle().manage(apnSender);
8574
environment.lifecycle().manage(pushNotificationScheduler);
8675

8776
MetricsUtil.registerSystemResourceMetrics(environment);

service/src/test/java/org/whispersystems/textsecuregcm/workers/FinishPushNotificationExperimentCommandTest.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@ void setUp() {
7272
null,
7373
null,
7474
null,
75+
null,
76+
null,
7577
pushNotificationExperimentSamples,
7678
null,
7779
null,

service/src/test/java/org/whispersystems/textsecuregcm/workers/NotifyIdleDevicesWithoutMessagesCommandTest.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,8 @@ private TestNotifyIdleDevicesWithoutMessagesCommand(final MessagesManager messag
6565
null,
6666
null,
6767
null,
68+
null,
69+
null,
6870
null);
6971

7072
this.idleDeviceNotificationScheduler = idleDeviceNotificationScheduler;

service/src/test/java/org/whispersystems/textsecuregcm/workers/StartPushNotificationExperimentCommandTest.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,8 @@ public TestStartPushNotificationExperimentCommand(
6161
null,
6262
null,
6363
null,
64+
null,
65+
null,
6466
pushNotificationExperimentSamples,
6567
null,
6668
null,

0 commit comments

Comments
 (0)