[Feature] Allow order control in createAggConfig

Add new optional boolean param that adds the new `AggConfig` to the beginning of the array rather than the end. Makes it easier to work with Pie or other visualizations with `Schemas` that set `mustBeFirst` Signed-off-by: Josh Romero <[email protected]>
opensearch-project · Dec 29, 2022 · cfab551 · cfab551
1 parent bd52024
commit cfab551
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 4 deletions.
diff --git a/src/plugins/data/common/search/aggs/agg_configs.test.ts b/src/plugins/data/common/search/aggs/agg_configs.test.ts
@@ -117,7 +117,7 @@ describe('AggConfigs', () => {
       expect(ac.aggs).toHaveLength(3);
     });
 
-    it('adds new AggConfig entries to AggConfigs by default', () => {
+    it('adds new AggConfig entries to end of AggConfigs by default', () => {
       const configStates = [
         {
           enabled: true,
@@ -136,6 +136,7 @@ describe('AggConfigs', () => {
         schema: 'split',
       });
       expect(ac.aggs).toHaveLength(2);
+      expect(ac.aggs[1].schema).toBe('split');
     });
 
     it('does not add an agg to AggConfigs if addToAggConfigs: false', () => {
@@ -161,6 +162,55 @@ describe('AggConfigs', () => {
       );
       expect(ac.aggs).toHaveLength(1);
     });
+
+    it('adds new AggConfig entries to beginning of AggConfigs if mustBeFirst: true', () => {
+      const configStates = [
+        {
+          enabled: true,
+          type: 'histogram',
+          params: {},
+        },
+      ];
+
+      const ac = new AggConfigs(indexPattern, configStates, { typesRegistry });
+      expect(ac.aggs).toHaveLength(1);
+
+      ac.createAggConfig(
+        {
+          enabled: true,
+          type: 'terms',
+          params: {},
+          schema: 'split',
+        },
+        { mustBeFirst: true }
+      );
+      expect(ac.aggs).toHaveLength(2);
+      expect(ac.aggs[0].schema).toBe('split');
+    });
+
+    it('does not add an agg to AggConfigs if addToAggConfigs: false and mustBeFirst: true', () => {
+      const configStates = [
+        {
+          enabled: true,
+          type: 'histogram',
+          params: {},
+        },
+      ];
+
+      const ac = new AggConfigs(indexPattern, configStates, { typesRegistry });
+      expect(ac.aggs).toHaveLength(1);
+
+      ac.createAggConfig(
+        {
+          enabled: true,
+          type: 'terms',
+          params: {},
+          schema: 'split',
+        },
+        { addToAggConfigs: false, mustBeFirst: true }
+      );
+      expect(ac.aggs).toHaveLength(1);
+    });
   });
 
   describe('#getRequestAggs', () => {

diff --git a/src/plugins/data/common/search/aggs/agg_configs.ts b/src/plugins/data/common/search/aggs/agg_configs.ts
@@ -130,10 +130,10 @@ export class AggConfigs {
 
   createAggConfig = <T extends AggConfig = AggConfig>(
     params: CreateAggConfigParams,
-    { addToAggConfigs = true } = {}
+    { addToAggConfigs = true, mustBeFirst = false } = {}
   ) => {
     const { type } = params;
-    let aggConfig;
+    let aggConfig: AggConfig;
 
     if (params instanceof AggConfig) {
       aggConfig = params;
@@ -145,8 +145,12 @@ export class AggConfigs {
       });
     }
 
+    const addAggConfig = () => {
+      return mustBeFirst ? this.aggs.unshift(aggConfig) : this.aggs.push(aggConfig);
+    };
+
     if (addToAggConfigs) {
-      this.aggs.push(aggConfig);
+      addAggConfig();
     }
 
     return aggConfig as T;