rm comment

prestodb · Feb 18, 2025 · 0472c5a · 0472c5a
1 parent 9f4cdd1
commit 0472c5a
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/presto-router/src/main/java/com/facebook/presto/router/cluster/ClusterManager.java b/presto-router/src/main/java/com/facebook/presto/router/cluster/ClusterManager.java
@@ -90,6 +90,44 @@ public ClusterManager(RouterConfig config, @ForClusterManager ScheduledExecutorS
         });
     }
 
+    @PostConstruct
+    public void startConfigReloadTask()
+    {
+        File routerConfigFile = new File(routerConfig.getConfigFile());
+        scheduledExecutorService.scheduleAtFixedRate(() -> {
+            long newConfigUpdateTime = routerConfigFile.lastModified();
+            if (lastConfigUpdate.get() != newConfigUpdateTime) {
+                RouterSpec routerSpec = parseRouterConfig(routerConfig)
+                        .orElseThrow(() -> new PrestoException(CONFIGURATION_INVALID, "Failed to load router config"));
+                this.groups = ImmutableMap.copyOf(routerSpec.getGroups().stream().collect(toMap(GroupSpec::getName, group -> group)));
+                this.groupSelectors = ImmutableList.copyOf(routerSpec.getSelectors());
+                this.schedulerType = routerSpec.getSchedulerType();
+                this.scheduler = new SchedulerFactory(routerSpec.getSchedulerType()).create();
+                this.initializeServerWeights();
+                this.initializeMembersDiscoveryURI();
+                List<URI> allClusters = getAllClusters();
+                allClusters.forEach(uri -> {
+                    if (!remoteClusterInfos.containsKey(uri)) {
+                        log.info("Attaching cluster %s to the router", uri.getHost());
+                        remoteClusterInfos.put(uri, remoteInfoFactory.createRemoteClusterInfo(discoveryURIs.get(uri)));
+                        remoteQueryInfos.put(uri, remoteInfoFactory.createRemoteQueryInfo(discoveryURIs.get(uri)));
+                        log.info("Successfully attached cluster %s to the router. Queries will be routed to cluster after successful health check", uri.getHost());
+                    }
+                });
+                for (URI uri : remoteClusterInfos.keySet()) {
+                    if (!allClusters.contains(uri)) {
+                        log.info("Removing cluster %s from the router", uri.getHost());
+                        remoteClusterInfos.remove(uri);
+                        remoteQueryInfos.remove(uri);
+                        discoveryURIs.remove(uri);
+                        log.info("Successfully removed cluster %s from the router", uri.getHost());
+                    }
+                }
+                lastConfigUpdate.set(newConfigUpdateTime);
+            }
+        }, 0L, (long) 5, TimeUnit.SECONDS);
+    }
+
     public List<URI> getAllClusters()
     {
         return groups.values().stream()