[secure-mode] make allowlist interface

Author: ethnzhng

plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeAllowList.java

@@ -15,12 +15,17 @@
 import java.util.Set;
 
 /** A class for properties that are allowlisted in Secure Mode. */
-final class SecureModeAllowList {
+interface SecureModeAllowList {
 
     public static final Set<String> PROPERTIES_ALLOWLIST =
             Set.of(
                     "engine",
                     "job_queue_size",
+                    "max_idle_time",
+                    "batch_size",
+                    "max_batch_delay",
+                    "minWorkers",
+                    "maxWorkers",
                     "option.entryPoint",
                     "option.task",
                     "option.model_id",
@@ -61,6 +66,4 @@ final class SecureModeAllowList {
                     "option.max_lora_rank",
                     "option.lora_extra_vocab_size",
                     "option.max_cpu_loras");
-
-    private SecureModeAllowList() {}
 }

plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeUtils.java

@@ -108,9 +108,9 @@ public static void validateSecurity(ModelInfo<?, ?> modelInfo) throws IOExceptio
      */
     private static void validateProperties(ModelInfo<?, ?> modelInfo, Set<String> allowedKeys) {
         Properties prop = modelInfo.getProperties();
-        Set<String> allowedKeysSet = new HashSet<>(allowedKeys);
+        allowedKeys = new HashSet<>(allowedKeys);
         for (String key : prop.stringPropertyNames()) {
-            if (!allowedKeysSet.contains(key)) {
+            if (!allowedKeys.contains(key)) {
                 throw new IllegalConfigurationException(
                         "Property " + key + " is prohibited from being set in Secure Mode.");
             }