enum string check

Signed-off-by: Maxwell Brown <[email protected]>
opensearch-project · Jan 3, 2025 · da25170 · da25170
1 parent c8636fd
commit da25170
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/...org/opensearch/dataprepper/plugins/sink/opensearch/configuration/ActionConfiguration.java b/...org/opensearch/dataprepper/plugins/sink/opensearch/configuration/ActionConfiguration.java
@@ -12,6 +12,14 @@ public class ActionConfiguration {
     @JsonProperty("type")
     private OpenSearchBulkActions type;
 
+    @AssertTrue(message = "type must be one of index, create, update, upsert, delete")
+    boolean isTypeValid() {
+        if (type == null) {         //type will be null if the string doesnt match one of the enums
+            return true;
+        }
+        return false;
+    }
+
     public String getType() {
         return type.toString();
     }

diff --git a/...rg/opensearch/dataprepper/plugins/sink/opensearch/configuration/OpenSearchSinkConfig.java b/...rg/opensearch/dataprepper/plugins/sink/opensearch/configuration/OpenSearchSinkConfig.java
@@ -139,6 +139,14 @@ public boolean getEnableRequestCompression() {
     @JsonProperty("action")
     private OpenSearchBulkActions action = OpenSearchBulkActions.INDEX;
 
+    @AssertTrue(message = "action must be one of index, create, update, upsert, delete")
+    boolean isActionValid() {
+        if (action == null) {         //action will be null if the string doesn't match one of the enums
+            return true;
+        }
+        return false;
+    }
+
     public String getAction() {
         return action.toString();
     }