Reveal property to enable auto_snapshot. (#1031) (#1033)

Netflix · Mar 25, 2023 · b3d4dd9 · b3d4dd9
1 parent 068dc51
commit b3d4dd9
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/priam/src/main/java/com/netflix/priam/config/IConfiguration.java b/priam/src/main/java/com/netflix/priam/config/IConfiguration.java
@@ -1171,6 +1171,9 @@ default long getCompressionTransitionEpochMillis() {
         return 0L;
     }
 
+    /** @return whether to enable auto_snapshot */
+    boolean getAutoSnapshot();
+
     /**
      * Escape hatch for getting any arbitrary property by key This is useful so we don't have to
      * keep adding methods to this interface for every single configuration option ever. Also

diff --git a/priam/src/main/java/com/netflix/priam/config/PriamConfiguration.java b/priam/src/main/java/com/netflix/priam/config/PriamConfiguration.java
@@ -789,4 +789,9 @@ public int getTargetMinutesToCompleteSnaphotUpload() {
     public double getRateLimitChangeThreshold() {
         return config.get(PRIAM_PRE + ".rateLimitChangeThreshold", 0.1);
     }
+
+    @Override
+    public boolean getAutoSnapshot() {
+        return config.get(PRIAM_PRE + ".autoSnapshot", false);
+    }
 }
diff --git a/priam/src/main/java/com/netflix/priam/tuner/StandardTuner.java b/priam/src/main/java/com/netflix/priam/tuner/StandardTuner.java
@@ -126,6 +126,7 @@ public void writeAllProperties(String yamlLocation, String hostname, String seed
         map.put(
                 "compaction_large_partition_warning_threshold_mb",
                 config.getCompactionLargePartitionWarnThresholdInMB());
+        map.put("auto_snapshot", config.getAutoSnapshot());
 
         List<?> seedp = (List) map.get("seed_provider");
         Map<String, String> m = (Map<String, String>) seedp.get(0);

diff --git a/priam/src/test/java/com/netflix/priam/config/FakeConfiguration.java b/priam/src/test/java/com/netflix/priam/config/FakeConfiguration.java
@@ -40,6 +40,7 @@ public class FakeConfiguration implements IConfiguration {
     private boolean skipUpdatingOthersIngressRules;
     private boolean skipIngressUnlessIPIsPublic;
     private long compressionTransitionEpochMillis;
+    private boolean autoSnapshot;
 
     public final Map<String, String> fakeProperties = new HashMap<>();
 
@@ -296,4 +297,14 @@ public void setCompressionTransitionEpochMillis(long transitionTime) {
     public long getCompressionTransitionEpochMillis() {
         return compressionTransitionEpochMillis;
     }
+
+    public FakeConfiguration setAutoSnapshot(boolean autoSnapshot) {
+        this.autoSnapshot = autoSnapshot;
+        return this;
+    }
+
+    @Override
+    public boolean getAutoSnapshot() {
+        return autoSnapshot;
+    }
 }