fix TODO

cryostatio · Mar 23, 2023 · 4db6f60 · 4db6f60
1 parent 37dda49
commit 4db6f60
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/main/java/io/cryostat/agent/MainModule.java b/src/main/java/io/cryostat/agent/MainModule.java
@@ -113,8 +113,10 @@ public static WebServer provideWebServer(
             ScheduledExecutorService executor,
             @Named(ConfigModule.CRYOSTAT_AGENT_WEBSERVER_HOST) String host,
             @Named(ConfigModule.CRYOSTAT_AGENT_WEBSERVER_PORT) int port,
-            Lazy<Registration> registration) {
-        return new WebServer(remoteContexts, cryostat, executor, host, port, registration);
+            Lazy<Registration> registration,
+            @Named(ConfigModule.CRYOSTAT_AGENT_REGISTRATION_RETRY_MS) int registrationRetryMs) {
+        return new WebServer(
+                remoteContexts, cryostat, executor, host, port, registration, registrationRetryMs);
     }
 
     @Provides

diff --git a/src/main/java/io/cryostat/agent/WebServer.java b/src/main/java/io/cryostat/agent/WebServer.java
@@ -79,6 +79,7 @@ class WebServer {
     private final int port;
     private final Credentials credentials;
     private final Lazy<Registration> registration;
+    private final int registrationRetryMs;
     private HttpServer http;
     private volatile int credentialId = -1;
 
@@ -92,14 +93,16 @@ class WebServer {
             ScheduledExecutorService executor,
             String host,
             int port,
-            Lazy<Registration> registration) {
+            Lazy<Registration> registration,
+            int registrationRetryMs) {
         this.remoteContexts = remoteContexts;
         this.cryostat = cryostat;
         this.executor = executor;
         this.host = host;
         this.port = port;
         this.credentials = new Credentials();
         this.registration = registration;
+        this.registrationRetryMs = registrationRetryMs;
 
         this.agentAuthenticator = new AgentAuthenticator();
         this.requestLoggingFilter = new RequestLoggingFilter();
@@ -163,7 +166,7 @@ void generateCredentials(CompletableFuture<Void> future) throws NoSuchAlgorithmE
                                                     log.error("Cannot submit credentials", e);
                                                 }
                                             },
-                                            5000, // TODO reuse registrationRetryMs
+                                            registrationRetryMs,
                                             TimeUnit.MILLISECONDS);
                                 } else {
                                     future.complete(null);