Add validation for DocumentDB Collection Config

Signed-off-by: Dinu John <[email protected]>
dinujoh · May 14, 2024 · 9d6510d · 9d6510d
1 parent 2975e1a
commit 9d6510d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...rc/main/java/org/opensearch/dataprepper/plugins/mongo/configuration/CollectionConfig.java b/...rc/main/java/org/opensearch/dataprepper/plugins/mongo/configuration/CollectionConfig.java
@@ -1,6 +1,8 @@
 package org.opensearch.dataprepper.plugins.mongo.configuration;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
+import jakarta.validation.constraints.Max;
+import jakarta.validation.constraints.Min;
 import jakarta.validation.constraints.NotNull;
 import jakarta.validation.constraints.Pattern;
 
@@ -23,11 +25,17 @@ public class CollectionConfig {
     private boolean stream;
 
     @JsonProperty("partition_count")
+    @Min(1)
+    @Max(1000)
     private int partitionCount;
 
     @JsonProperty("export_batch_size")
+    @Min(1)
+    @Max(1_000_000)
     private int exportBatchSize;
     @JsonProperty("stream_batch_size")
+    @Min(1)
+    @Max(1_000_000)
     private int streamBatchSize;
 
     public CollectionConfig() {