merge

shanhexi · shanhexi · commit db84c2927306 · 2023-10-14T16:10:57.000+08:00
diff --git a/chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLMetaData.java b/chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLMetaData.java
@@ -291,6 +291,7 @@ public TableMeta getTableMeta(String databaseName, String schemaName, String tab
                 .columnTypes(PostgreSQLColumnTypeEnum.getTypes())
                 .charsets(PostgreSQLCharsetEnum.getCharsets())
                 .collations(PostgreSQLCollationEnum.getCollations())
+                .indexTypes(PostgreSQLIndexTypeEnum.getIndexTypes())
                 .build();
     }
 }
diff --git a/chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/type/PostgreSQLIndexTypeEnum.java b/chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/type/PostgreSQLIndexTypeEnum.java
@@ -1,12 +1,16 @@
 package ai.chat2db.plugin.postgresql.type;
 
 import ai.chat2db.spi.enums.EditStatus;
+import ai.chat2db.spi.model.IndexType;
 import ai.chat2db.spi.model.TableIndex;
 import ai.chat2db.spi.model.TableIndexColumn;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang3.BooleanUtils;
 import org.apache.commons.lang3.StringUtils;
 
+import java.util.Arrays;
+import java.util.List;
+
 public enum PostgreSQLIndexTypeEnum {
 
     PRIMARY("Primary", "PRIMARY KEY"),
@@ -21,10 +25,13 @@ public enum PostgreSQLIndexTypeEnum {
     private String name;
     private String keyword;
 
+    private IndexType indexType;
+
 
     PostgreSQLIndexTypeEnum(String name, String keyword) {
         this.name = name;
         this.keyword = keyword;
+        this.indexType =new IndexType(name);
     }
 
     public static PostgreSQLIndexTypeEnum getByType(String type) {
@@ -36,6 +43,14 @@ public static PostgreSQLIndexTypeEnum getByType(String type) {
         return null;
     }
 
+    public static List<IndexType> getIndexTypes() {
+        return Arrays.asList(PostgreSQLIndexTypeEnum.values()).stream().map(PostgreSQLIndexTypeEnum::getIndexType).collect(java.util.stream.Collectors.toList());
+    }
+
+    public IndexType getIndexType() {
+        return indexType;
+    }
+
     public String getName() {
         return name;
     }

Original file line number	Diff line number	Diff line change
`@@ -291,6 +291,7 @@ public TableMeta getTableMeta(String databaseName, String schemaName, String tab`
`291`	`291`	`.columnTypes(PostgreSQLColumnTypeEnum.getTypes())`
`292`	`292`	`.charsets(PostgreSQLCharsetEnum.getCharsets())`
`293`	`293`	`.collations(PostgreSQLCollationEnum.getCollations())`
	`294`	`+ .indexTypes(PostgreSQLIndexTypeEnum.getIndexTypes())`
`294`	`295`	`.build();`
`295`	`296`	`}`
`296`	`297`	`}`