Skip to content

Commit 493f934

Browse files
committed
fix columnType error
1 parent 3961610 commit 493f934

File tree

23 files changed

+119
-78
lines changed

23 files changed

+119
-78
lines changed

chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/type/ClickHouseColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -67,7 +68,7 @@ public enum ClickHouseColumnTypeEnum implements ColumnBuilder {
6768
}
6869

6970
public static ClickHouseColumnTypeEnum getByType(String dataType) {
70-
return COLUMN_TYPE_MAP.get(dataType);
71+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
7172
}
7273

7374
public static List<ColumnType> getTypes() {

chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/type/DB2ColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -133,7 +134,7 @@ public enum DB2ColumnTypeEnum implements ColumnBuilder {
133134
private ColumnType columnType;
134135

135136
public static DB2ColumnTypeEnum getByType(String dataType) {
136-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
137+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
137138
}
138139

139140
private static Map<String, DB2ColumnTypeEnum> COLUMN_TYPE_MAP = Maps.newHashMap();

chat2db-server/chat2db-plugins/chat2db-dm/src/main/java/ai/chat2db/plugin/dm/type/DMColumnTypeEnum.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -123,11 +124,13 @@ public enum DMColumnTypeEnum implements ColumnBuilder {
123124

124125
VARCHAR2("VARCHAR2", true, false, true, false, false, false, true, true, false, true),
125126

127+
DATETIME("DATETIME", false, false, true, false, false, false, true, true, false, false),
126128
;
127129
private ColumnType columnType;
128130

129131
public static DMColumnTypeEnum getByType(String dataType) {
130-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
132+
String type = SqlUtils.removeDigits(dataType.toUpperCase());
133+
return COLUMN_TYPE_MAP.get(type);
131134
}
132135

133136
private static Map<String, DMColumnTypeEnum> COLUMN_TYPE_MAP = Maps.newHashMap();

chat2db-server/chat2db-plugins/chat2db-hive/src/main/java/ai/chat2db/plugin/hive/type/HiveColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -71,7 +72,7 @@ public enum HiveColumnTypeEnum implements ColumnBuilder {
7172
private ColumnType columnType;
7273

7374
public static HiveColumnTypeEnum getByType(String dataType) {
74-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
75+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
7576
}
7677

7778
public ColumnType getColumnType() {

chat2db-server/chat2db-plugins/chat2db-kingbase/src/main/java/ai/chat2db/plugin/kingbase/type/KingBaseColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -87,7 +88,7 @@ public enum KingBaseColumnTypeEnum implements ColumnBuilder {
8788
}
8889

8990
public static KingBaseColumnTypeEnum getByType(String dataType) {
90-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
91+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
9192
}
9293

9394
public static List<ColumnType> getTypes() {

chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/type/MysqlColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -116,7 +117,7 @@ public enum MysqlColumnTypeEnum implements ColumnBuilder {
116117
private ColumnType columnType;
117118

118119
public static MysqlColumnTypeEnum getByType(String dataType) {
119-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
120+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
120121
}
121122

122123
public ColumnType getColumnType() {

chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/value/MysqlValueProcessor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ public String getJdbcValue(JDBCDataValue dataValue) {
4242

4343

4444
@Override
45-
public String getJdbcValueString(JDBCDataValue dataValue) {
45+
public String getJdbcSqlValueString(JDBCDataValue dataValue) {
4646
Object value = dataValue.getObject();
4747
if (Objects.isNull(value)) {
4848
// mysql -> example: [date]->0000-00-00

chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/value/OracleValueProcessor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ public String getJdbcValue(JDBCDataValue dataValue) {
3636

3737

3838
@Override
39-
public String getJdbcValueString(JDBCDataValue dataValue) {
39+
public String getJdbcSqlValueString(JDBCDataValue dataValue) {
4040
if (OracleColumnTypeEnum.LONG_RAW.getColumnType().getTypeName().equalsIgnoreCase(dataValue.getType())) {
4141
return convertJDBCValueStrByType(dataValue);
4242
}

chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/type/PostgreSQLColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -76,7 +77,7 @@ public enum PostgreSQLColumnTypeEnum implements ColumnBuilder {
7677
}
7778

7879
public static PostgreSQLColumnTypeEnum getByType(String dataType) {
79-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
80+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
8081
}
8182

8283
public static List<ColumnType> getTypes() {

chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/type/SqliteColumnTypeEnum.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import ai.chat2db.spi.enums.EditStatus;
55
import ai.chat2db.spi.model.ColumnType;
66
import ai.chat2db.spi.model.TableColumn;
7+
import ai.chat2db.spi.util.SqlUtils;
78
import com.google.common.collect.Maps;
89
import org.apache.commons.lang3.StringUtils;
910

@@ -27,7 +28,7 @@ public enum SqliteColumnTypeEnum implements ColumnBuilder {
2728
private ColumnType columnType;
2829

2930
public static SqliteColumnTypeEnum getByType(String dataType) {
30-
return COLUMN_TYPE_MAP.get(dataType.toUpperCase());
31+
return COLUMN_TYPE_MAP.get(SqlUtils.removeDigits(dataType.toUpperCase()));
3132
}
3233

3334
public ColumnType getColumnType() {

0 commit comments

Comments
 (0)