From 9de188ae7ed0b02a7666265db43b77465584b53e Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Thu, 9 May 2024 14:07:27 +0800
Subject: [PATCH] (improvement)(Headless) Support parsing description
 information and aggregation methods in the map phase. (#969)

---
 .../tencent/supersonic/headless/api/pojo/SchemaElement.java | 1 +
 .../headless/server/utils/DataSetSchemaBuilder.java         | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaElement.java b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaElement.java
index 398169d78..eeb8b4ee8 100644
--- a/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaElement.java
+++ b/headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/SchemaElement.java
@@ -32,6 +32,7 @@ public class SchemaElement implements Serializable {
     private String dataFormatType;
     private double order;
     private int isTag;
+    private String description;
 
     @Override
     public boolean equals(Object o) {
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/DataSetSchemaBuilder.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/DataSetSchemaBuilder.java
index 0e27d6581..c50deb226 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/DataSetSchemaBuilder.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/DataSetSchemaBuilder.java
@@ -76,7 +76,9 @@ public class DataSetSchemaBuilder {
                         .type(SchemaElementType.TAG)
                         .useCnt(metric.getUseCnt())
                         .alias(alias)
+                        .defaultAgg(metric.getDefaultAgg())
                         .isTag(metric.getIsTag())
+                        .description(metric.getDescription())
                         .build();
                 tags.add(tagToAdd);
             }
@@ -110,6 +112,7 @@ public class DataSetSchemaBuilder {
                         .alias(alias)
                         .schemaValueMaps(schemaValueMaps)
                         .isTag(dim.getIsTag())
+                        .description(dim.getDescription())
                         .build();
                 tags.add(tagToAdd);
             }
@@ -159,6 +162,7 @@ public class DataSetSchemaBuilder {
                     .alias(alias)
                     .schemaValueMaps(schemaValueMaps)
                     .isTag(dim.getIsTag())
+                    .description(dim.getDescription())
                     .build();
             dimensions.add(dimToAdd);
         }
@@ -191,6 +195,7 @@ public class DataSetSchemaBuilder {
                     .useCnt(dim.getUseCnt())
                     .alias(new ArrayList<>(Arrays.asList(dimValueAlias.toArray(new String[0]))))
                     .isTag(dim.getIsTag())
+                    .description(dim.getDescription())
                     .build();
             dimensionValues.add(dimValueToAdd);
         }
@@ -217,6 +222,7 @@ public class DataSetSchemaBuilder {
                     .defaultAgg(metric.getDefaultAgg())
                     .dataFormatType(metric.getDataFormatType())
                     .isTag(metric.getIsTag())
+                    .description(metric.getDescription())
                     .build();
             metrics.add(metricToAdd);