From c560f7ed0bac1b1c9cf1fa8ddfcc980e826ef851 Mon Sep 17 00:00:00 2001 From: SwallowGG <1558143046@qq.com> Date: Thu, 13 Jun 2024 09:53:34 +0800 Subject: [PATCH] =?UTF-8?q?=E4=B8=8D=E6=94=AF=E6=8C=81=E7=9A=84=E7=B1=BB?= =?UTF-8?q?=E5=9E=8B=E6=97=A0=E6=B3=95=E4=BD=BF=E7=94=A8AI?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../sqlserver/type/SqlServerColumnTypeEnum.java | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/type/SqlServerColumnTypeEnum.java b/chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/type/SqlServerColumnTypeEnum.java index 9117be84..7be1a04c 100644 --- a/chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/type/SqlServerColumnTypeEnum.java +++ b/chat2db-server/chat2db-plugins/chat2db-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 COLUMN_TYPE_MAP = Maps.newHashMap(); @@ -255,7 +260,9 @@ public enum SqlServerColumnTypeEnum implements ColumnBuilder { } return script.toString(); } - + if(OTHER.equals(columnType)){ + return column.getColumnType(); + } return columnType; }