[Enhancement] Support RESTORE_CLUSTER_SNAPSHOT env var to enter the r…

…estore cluster snapshot procedure Signed-off-by: xiangguangyxg <[email protected]>
StarRocks · Feb 7, 2025 · a07e66a · a07e66a
1 parent 64d9e17
commit a07e66a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/fe/fe-core/src/main/java/com/starrocks/lake/snapshot/RestoreClusterSnapshotMgr.java b/fe/fe-core/src/main/java/com/starrocks/lake/snapshot/RestoreClusterSnapshotMgr.java
@@ -53,11 +53,17 @@ private RestoreClusterSnapshotMgr(String clusterSnapshotYamlFile) throws StarRoc
     public static void init(String clusterSnapshotYamlFile, String[] args) throws StarRocksException {
         for (String arg : args) {
             if (arg.equalsIgnoreCase("-cluster_snapshot")) {
-                LOG.info("FE start to restore from a cluster snapshot");
+                LOG.info("FE start to restore from a cluster snapshot (-cluster_snapshot)");
                 instance = new RestoreClusterSnapshotMgr(clusterSnapshotYamlFile);
                 return;
             }
         }
+
+        String restoreClusterSnapshotEnv = System.getenv("RESTORE_CLUSTER_SNAPSHOT");
+        if (restoreClusterSnapshotEnv != null && restoreClusterSnapshotEnv.equalsIgnoreCase("true")) {
+            LOG.info("FE start to restore from a cluster snapshot (RESTORE_CLUSTER_SNAPSHOT=true)");
+            instance = new RestoreClusterSnapshotMgr(clusterSnapshotYamlFile);
+        }
     }
 
     public static boolean isRestoring() {