diff --git a/src/main/java/com/muyu/enums/DataType.java b/src/main/java/com/muyu/enums/DataType.java index 2d66436..198acb7 100644 --- a/src/main/java/com/muyu/enums/DataType.java +++ b/src/main/java/com/muyu/enums/DataType.java @@ -111,12 +111,32 @@ public enum DataType { break; } } - if (StringUtils.isNotEmpty(javaType)) { + if (StringUtils.isEmpty(javaType)) { log.info("SQL字段类型异常,sqlType ---> {}", sqlType); throw new RuntimeException(String.format("SQL字段类型异常,sqlType ---> {}", sqlType)); } return javaType; } + /** + * 获取枚举类型 + * @param sqlType sql数据类型 + * @return 枚举类型 + */ + public static DataType getDataType(String sqlType) { + DataType dataType = null; + for (DataType value : values()) { + if (sqlType != null && Arrays.asList(value.getSourceType()).contains(sqlType.toLowerCase())) { + dataType = value; + } + } + if (dataType == null) { + log.info("SQL字段类型异常,sqlType ---> {}", sqlType); + throw new RuntimeException(String.format("SQL字段类型异常,sqlType ---> {}", sqlType)); + } + return dataType; + + } + }