Add flatten_result_index_mapping field to Config (opensearch-project#…

…1276) * Add flatten_result_index_mapping field into Config Signed-off-by: Jackie Han <[email protected]> * apply spotless check Signed-off-by: Jackie Han <[email protected]> --------- Signed-off-by: Jackie Han <[email protected]>
kaituo · Aug 12, 2024 · 34b350e · 34b350e
1 parent eb7aad4
commit 34b350e
Show file tree

Hide file tree

Showing 18 changed files with 161 additions and 25 deletions.
diff --git a/src/main/java/org/opensearch/ad/model/ADTask.java b/src/main/java/org/opensearch/ad/model/ADTask.java
@@ -344,7 +344,8 @@ public static ADTask parse(XContentParser parser, String taskId) throws IOExcept
                 detector.getRules(),
                 detector.getCustomResultIndexMinSize(),
                 detector.getCustomResultIndexMinAge(),
-                detector.getCustomResultIndexTTL()
+                detector.getCustomResultIndexTTL(),
+                detector.getFlattenResultIndexMapping()
             );
         return new Builder()
             .taskId(parsedTaskId)

diff --git a/src/main/java/org/opensearch/ad/model/AnomalyDetector.java b/src/main/java/org/opensearch/ad/model/AnomalyDetector.java
@@ -150,6 +150,7 @@ public Integer getShingleSize(Integer customShingleSize) {
      * @param customResultIndexMinSize custom result index lifecycle management min size condition
      * @param customResultIndexMinAge custom result index lifecycle management min age condition
      * @param customResultIndexTTL custom result index lifecycle management ttl
+     * @param flattenResultIndexMapping flag to indicate whether to flatten result index mapping or not
      */
     public AnomalyDetector(
         String detectorId,
@@ -176,7 +177,8 @@ public AnomalyDetector(
         List<Rule> rules,
         Integer customResultIndexMinSize,
         Integer customResultIndexMinAge,
-        Integer customResultIndexTTL
+        Integer customResultIndexTTL,
+        Boolean flattenResultIndexMapping
     ) {
         super(
             detectorId,
@@ -203,7 +205,8 @@ public AnomalyDetector(
             historyIntervals,
             customResultIndexMinSize,
             customResultIndexMinAge,
-            customResultIndexTTL
+            customResultIndexTTL,
+            flattenResultIndexMapping
         );
 
         checkAndThrowValidationErrors(ValidationAspect.DETECTOR);
@@ -280,6 +283,7 @@ public AnomalyDetector(StreamInput input) throws IOException {
         this.customResultIndexMinSize = input.readOptionalInt();
         this.customResultIndexMinAge = input.readOptionalInt();
         this.customResultIndexTTL = input.readOptionalInt();
+        this.flattenResultIndexMapping = input.readOptionalBoolean();
     }
 
     public XContentBuilder toXContent(XContentBuilder builder) throws IOException {
@@ -345,6 +349,7 @@ public void writeTo(StreamOutput output) throws IOException {
         output.writeOptionalInt(customResultIndexMinSize);
         output.writeOptionalInt(customResultIndexMinAge);
         output.writeOptionalInt(customResultIndexTTL);
+        output.writeOptionalBoolean(flattenResultIndexMapping);
     }
 
     @Override
@@ -441,6 +446,7 @@ public static AnomalyDetector parse(
         Integer customResultIndexMinSize = null;
         Integer customResultIndexMinAge = null;
         Integer customResultIndexTTL = null;
+        Boolean flattenResultIndexMapping = null;
 
         ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.currentToken(), parser);
         while (parser.nextToken() != XContentParser.Token.END_OBJECT) {
@@ -575,6 +581,9 @@ public static AnomalyDetector parse(
                 case RESULT_INDEX_FIELD_TTL:
                     customResultIndexTTL = onlyParseNumberValue(parser);
                     break;
+                case FLATTEN_RESULT_INDEX_MAPPING:
+                    flattenResultIndexMapping = onlyParseBooleanValue(parser);
+                    break;
                 default:
                     parser.skipChildren();
                     break;
@@ -605,7 +614,8 @@ public static AnomalyDetector parse(
             rules,
             customResultIndexMinSize,
             customResultIndexMinAge,
-            customResultIndexTTL
+            customResultIndexTTL,
+            flattenResultIndexMapping
         );
         detector.setDetectionDateRange(detectionDateRange);
         return detector;
@@ -692,4 +702,11 @@ private static Integer onlyParseNumberValue(XContentParser parser) throws IOExce
         }
         return null;
     }
+
+    private static Boolean onlyParseBooleanValue(XContentParser parser) throws IOException {
+        if (parser.currentToken() == XContentParser.Token.VALUE_BOOLEAN) {
+            return parser.booleanValue();
+        }
+        return null;
+    }
 }
diff --git a/src/main/java/org/opensearch/ad/rest/handler/AbstractAnomalyDetectorActionHandler.java b/src/main/java/org/opensearch/ad/rest/handler/AbstractAnomalyDetectorActionHandler.java
@@ -239,7 +239,8 @@ protected AnomalyDetector copyConfig(User user, Config config) {
             detector.getRules(),
             config.getCustomResultIndexMinSize(),
             config.getCustomResultIndexMinAge(),
-            config.getCustomResultIndexTTL()
+            config.getCustomResultIndexTTL(),
+            config.getFlattenResultIndexMapping()
         );
     }
 

diff --git a/src/main/java/org/opensearch/forecast/model/ForecastTask.java b/src/main/java/org/opensearch/forecast/model/ForecastTask.java
@@ -342,7 +342,8 @@ public static ForecastTask parse(XContentParser parser, String taskId) throws IO
                 forecaster.getHistoryIntervals(),
                 forecaster.getCustomResultIndexMinSize(),
                 forecaster.getCustomResultIndexMinAge(),
-                forecaster.getCustomResultIndexTTL()
+                forecaster.getCustomResultIndexTTL(),
+                forecaster.getFlattenResultIndexMapping()
             );
         return new Builder()
             .taskId(parsedTaskId)

diff --git a/src/main/java/org/opensearch/forecast/model/Forecaster.java b/src/main/java/org/opensearch/forecast/model/Forecaster.java
@@ -134,7 +134,8 @@ public Forecaster(
         Integer historyIntervals,
         Integer customResultIndexMinSize,
         Integer customResultIndexMinAge,
-        Integer customResultIndexTTL
+        Integer customResultIndexTTL,
+        Boolean flattenResultIndexMapping
     ) {
         super(
             forecasterId,
@@ -161,7 +162,8 @@ public Forecaster(
             historyIntervals,
             customResultIndexMinSize,
             customResultIndexMinAge,
-            customResultIndexTTL
+            customResultIndexTTL,
+            flattenResultIndexMapping
         );
 
         checkAndThrowValidationErrors(ValidationAspect.FORECASTER);
@@ -303,6 +305,7 @@ public static Forecaster parse(
         Integer customResultIndexMinSize = null;
         Integer customResultIndexMinAge = null;
         Integer customResultIndexTTL = null;
+        Boolean flattenResultIndexMapping = null;
 
         ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.currentToken(), parser);
         while (parser.nextToken() != XContentParser.Token.END_OBJECT) {
@@ -431,6 +434,9 @@ public static Forecaster parse(
                 case RESULT_INDEX_FIELD_TTL:
                     customResultIndexTTL = parser.intValue();
                     break;
+                case FLATTEN_RESULT_INDEX_MAPPING:
+                    flattenResultIndexMapping = parser.booleanValue();
+                    break;
                 default:
                     parser.skipChildren();
                     break;
@@ -461,7 +467,8 @@ public static Forecaster parse(
             historyIntervals,
             customResultIndexMinSize,
             customResultIndexMinAge,
-            customResultIndexTTL
+            customResultIndexTTL,
+            flattenResultIndexMapping
         );
         return forecaster;
     }

diff --git a/src/main/java/org/opensearch/forecast/rest/handler/AbstractForecasterActionHandler.java b/src/main/java/org/opensearch/forecast/rest/handler/AbstractForecasterActionHandler.java
@@ -257,7 +257,8 @@ protected Config copyConfig(User user, Config config) {
             config.getHistoryIntervals(),
             config.getCustomResultIndexMinSize(),
             config.getCustomResultIndexMinAge(),
-            config.getCustomResultIndexTTL()
+            config.getCustomResultIndexTTL(),
+            config.getFlattenResultIndexMapping()
         );
     }
 

diff --git a/src/main/java/org/opensearch/timeseries/model/Config.java b/src/main/java/org/opensearch/timeseries/model/Config.java
@@ -79,6 +79,7 @@ public abstract class Config implements Writeable, ToXContentObject {
     public static final String RESULT_INDEX_FIELD_MIN_SIZE = "result_index_min_size";
     public static final String RESULT_INDEX_FIELD_MIN_AGE = "result_index_min_age";
     public static final String RESULT_INDEX_FIELD_TTL = "result_index_ttl";
+    public static final String FLATTEN_RESULT_INDEX_MAPPING = "flatten_result_index_mapping";
 
     protected String id;
     protected Long version;
@@ -118,6 +119,7 @@ public abstract class Config implements Writeable, ToXContentObject {
     protected Integer customResultIndexMinSize;
     protected Integer customResultIndexMinAge;
     protected Integer customResultIndexTTL;
+    protected Boolean flattenResultIndexMapping;
 
     public static String INVALID_RESULT_INDEX_NAME_SIZE = "Result index name size must contains less than "
         + MAX_RESULT_INDEX_NAME_SIZE
@@ -148,7 +150,8 @@ protected Config(
         Integer historyIntervals,
         Integer customResultIndexMinSize,
         Integer customResultIndexMinAge,
-        Integer customResultIndexTTL
+        Integer customResultIndexTTL,
+        Boolean flattenResultIndexMapping
     ) {
         if (Strings.isBlank(name)) {
             errorMessage = CommonMessages.EMPTY_NAME;
@@ -267,6 +270,7 @@ protected Config(
         this.customResultIndexMinSize = Strings.trimToNull(resultIndex) == null ? null : customResultIndexMinSize;
         this.customResultIndexMinAge = Strings.trimToNull(resultIndex) == null ? null : customResultIndexMinAge;
         this.customResultIndexTTL = Strings.trimToNull(resultIndex) == null ? null : customResultIndexTTL;
+        this.flattenResultIndexMapping = Strings.trimToNull(resultIndex) == null ? null : flattenResultIndexMapping;
     }
 
     public int suggestHistory() {
@@ -310,6 +314,7 @@ public Config(StreamInput input) throws IOException {
         this.customResultIndexMinSize = input.readOptionalInt();
         this.customResultIndexMinAge = input.readOptionalInt();
         this.customResultIndexTTL = input.readOptionalInt();
+        this.flattenResultIndexMapping = input.readOptionalBoolean();
     }
 
     /*
@@ -362,6 +367,7 @@ public void writeTo(StreamOutput output) throws IOException {
         output.writeOptionalInt(customResultIndexMinSize);
         output.writeOptionalInt(customResultIndexMinAge);
         output.writeOptionalInt(customResultIndexTTL);
+        output.writeOptionalBoolean(flattenResultIndexMapping);
     }
 
     public boolean invalidShingleSizeRange(Integer shingleSizeToTest) {
@@ -418,7 +424,8 @@ public boolean equals(Object o) {
             && Objects.equal(historyIntervals, config.historyIntervals)
             && Objects.equal(customResultIndexMinSize, config.customResultIndexMinSize)
             && Objects.equal(customResultIndexMinAge, config.customResultIndexMinAge)
-            && Objects.equal(customResultIndexTTL, config.customResultIndexTTL);
+            && Objects.equal(customResultIndexTTL, config.customResultIndexTTL)
+            && Objects.equal(flattenResultIndexMapping, config.flattenResultIndexMapping);
     }
 
     @Generated
@@ -445,7 +452,8 @@ public int hashCode() {
                 historyIntervals,
                 customResultIndexMinSize,
                 customResultIndexMinAge,
-                customResultIndexTTL
+                customResultIndexTTL,
+                flattenResultIndexMapping
             );
     }
 
@@ -494,6 +502,9 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
         if (customResultIndexTTL != null) {
             builder.field(RESULT_INDEX_FIELD_TTL, customResultIndexTTL);
         }
+        if (flattenResultIndexMapping != null) {
+            builder.field(FLATTEN_RESULT_INDEX_MAPPING, flattenResultIndexMapping);
+        }
         return builder;
     }
 
@@ -702,6 +713,10 @@ public Integer getCustomResultIndexTTL() {
         return customResultIndexTTL;
     }
 
+    public Boolean getFlattenResultIndexMapping() {
+        return flattenResultIndexMapping;
+    }
+
     /**
      * Identifies redundant feature names.
      *
@@ -756,6 +771,7 @@ public String toString() {
                 .append("customResultIndexMinSize", customResultIndexMinSize)
                 .append("customResultIndexMinAge", customResultIndexMinAge)
                 .append("customResultIndexTTL", customResultIndexTTL)
+                .append("flattenResultIndexMapping", flattenResultIndexMapping)
                 .toString();
     }
 }
diff --git a/src/test/java/org/opensearch/ad/AnomalyDetectorRestTestCase.java b/src/test/java/org/opensearch/ad/AnomalyDetectorRestTestCase.java
@@ -314,7 +314,8 @@ public ToXContentObject[] getConfig(String detectorId, BasicHeader header, boole
                 null,
                 detector.getCustomResultIndexMinSize(),
                 detector.getCustomResultIndexMinAge(),
-                detector.getCustomResultIndexTTL()
+                detector.getCustomResultIndexTTL(),
+                detector.getFlattenResultIndexMapping()
             ),
             detectorJob,
             historicalAdTask,
@@ -640,7 +641,8 @@ protected AnomalyDetector cloneDetector(AnomalyDetector anomalyDetector, String
             null,
             anomalyDetector.getCustomResultIndexMinSize(),
             anomalyDetector.getCustomResultIndexMinAge(),
-            anomalyDetector.getCustomResultIndexTTL()
+            anomalyDetector.getCustomResultIndexTTL(),
+            anomalyDetector.getFlattenResultIndexMapping()
         );
         return detector;
     }