Ericsson#699 Reload nodes

paulchandler · Nov 4, 2024 · cbe31b5 · cbe31b5
1 parent 7a5da7e
commit cbe31b5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/...tion/src/main/java/com/ericsson/bss/cassandra/ecchronos/application/spring/ECChronos.java b/...tion/src/main/java/com/ericsson/bss/cassandra/ecchronos/application/spring/ECChronos.java
@@ -83,6 +83,7 @@ public ECChronos(
                 .withRepairConfiguration(repairConfigurationProvider::get)
                 .withEccNodesSync(eccNodesSync)
                 .withJmxConnectionProvider(jmxConnectionProvider)
+                .withDistributedNativeConnectionProvider(nativeConnectionProvider)
                 .withTableReferenceFactory(myECChronosInternals.getTableReferenceFactory()));
 
         myECChronosInternals.addRunPolicy(myTimeBasedRunPolicy);

diff --git a/...ericsson/bss/cassandra/ecchronos/core/impl/repair/DefaultRepairConfigurationProvider.java b/...ericsson/bss/cassandra/ecchronos/core/impl/repair/DefaultRepairConfigurationProvider.java
@@ -71,7 +71,7 @@ public class DefaultRepairConfigurationProvider extends NodeStateListenerBase im
     private final ExecutorService myService;
     private EccNodesSync myEccNodesSync;
     private DistributedJmxConnectionProvider myJmxConnectionProvider;
-    private final DistributedNativeConnectionProvider myAgentNativeConnectionProvider = null;
+    private DistributedNativeConnectionProvider myAgentNativeConnectionProvider;
 
     /**
      * Default constructor.
@@ -92,6 +92,7 @@ private DefaultRepairConfigurationProvider(final Builder builder)
                 "Table reference factory must be set");
         myEccNodesSync = builder.myEccNodesSync;
         myJmxConnectionProvider = builder.myJmxConnectionProvider;
+        myAgentNativeConnectionProvider = builder.myAgentNativeConnectionProvider;
 
 
         setupConfiguration();
@@ -112,6 +113,9 @@ public void fromBuilder(final Builder builder)
         myRepairConfigurationFunction = builder.myRepairConfigurationFunction;
         myTableReferenceFactory = Preconditions.checkNotNull(builder.myTableReferenceFactory,
                 "Table reference factory must be set");
+        myEccNodesSync = builder.myEccNodesSync;
+        myJmxConnectionProvider = builder.myJmxConnectionProvider;
+        myAgentNativeConnectionProvider = builder.myAgentNativeConnectionProvider;
 
         setupConfiguration();
     }
@@ -519,6 +523,7 @@ public static class Builder
         private TableReferenceFactory myTableReferenceFactory;
         private EccNodesSync myEccNodesSync;
         private DistributedJmxConnectionProvider myJmxConnectionProvider;
+        private DistributedNativeConnectionProvider myAgentNativeConnectionProvider;
 
 
         /**
@@ -616,6 +621,12 @@ public Builder withEccNodesSync(final EccNodesSync eccNodesSync)
             myEccNodesSync = eccNodesSync;
             return this;
         }
+        public Builder withDistributedNativeConnectionProvider(final DistributedNativeConnectionProvider agentNativeConnectionProvider)
+        {
+            myAgentNativeConnectionProvider = agentNativeConnectionProvider;
+            return this;
+        }
+
 
         /**
          * Build with DistributedJmxConnectionProvider.