diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/workers/AbstractSinglePassCrawlAccountsCommand.java b/service/src/main/java/org/whispersystems/textsecuregcm/workers/AbstractSinglePassCrawlAccountsCommand.java index 539efe54..0606de55 100644 --- a/service/src/main/java/org/whispersystems/textsecuregcm/workers/AbstractSinglePassCrawlAccountsCommand.java +++ b/service/src/main/java/org/whispersystems/textsecuregcm/workers/AbstractSinglePassCrawlAccountsCommand.java @@ -24,6 +24,7 @@ import reactor.core.scheduler.Schedulers; public abstract class AbstractSinglePassCrawlAccountsCommand extends EnvironmentCommand { private CommandDependencies commandDependencies; + private Namespace namespace; private final Logger logger = LoggerFactory.getLogger(getClass()); @@ -53,14 +54,19 @@ public abstract class AbstractSinglePassCrawlAccountsCommand extends Environment return commandDependencies; } + protected Namespace getNamespace() { + return namespace; + } + @Override protected void run(final Environment environment, final Namespace namespace, final WhisperServerConfiguration configuration) throws Exception { UncaughtExceptionHandler.register(); - MetricsUtil.configureRegistries(configuration, environment); - commandDependencies = CommandDependencies.build(getName(), environment, configuration); + + this.namespace = namespace; + this.commandDependencies = CommandDependencies.build(getName(), environment, configuration); final int segments = Objects.requireNonNull(namespace.getInt(SEGMENT_COUNT)); @@ -75,7 +81,6 @@ public abstract class AbstractSinglePassCrawlAccountsCommand extends Environment } finally { commandStopListener.stop(); } - } @Override