不支持的类型无法使用AI

CodePhiliaX · Jun 13, 2024 · c560f7e · c560f7e
1 parent 3784118
commit c560f7e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/...2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/type/SqlServerColumnTypeEnum.java b/...2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/type/SqlServerColumnTypeEnum.java
@@ -98,11 +98,16 @@ public enum SqlServerColumnTypeEnum implements ColumnBuilder {
     XML("XML", false, false, true, false, false, false, true, true),
 
 
+    OTHER("OTHER", false, false, true, false, false, false, true, true),
     ;
     private ColumnType columnType;
 
     public static SqlServerColumnTypeEnum getByType(String dataType) {
-        return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
+        SqlServerColumnTypeEnum typeEnum = COLUMN_TYPE_MAP.get(dataType.toUpperCase());
+        if (typeEnum == null) {
+            return OTHER;
+        }
+        return typeEnum;
     }
 
     private static Map<String, SqlServerColumnTypeEnum> COLUMN_TYPE_MAP = Maps.newHashMap();
@@ -255,7 +260,9 @@ private String buildDataType(TableColumn column, SqlServerColumnTypeEnum type) {
             }
             return script.toString();
         }
-
+        if(OTHER.equals(columnType)){
+            return column.getColumnType();
+        }
 
         return columnType;
     }