修复基于建表SQL生成代码 列名没有COMMENT注释报错的问题

This commit is contained in:
liu xm 2021-11-08 13:27:34 +08:00
parent b369b363b6
commit 2fc1ef519d

View File

@ -16,6 +16,7 @@ import org.apache.commons.collections4.keyvalue.DefaultKeyValue;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Objects;
import static com.alibaba.druid.sql.SQLUtils.normalize; import static com.alibaba.druid.sql.SQLUtils.normalize;
@ -97,7 +98,8 @@ public class ToolCodegenSQLParser {
columns.add(ToolSchemaColumnDO.builder() columns.add(ToolSchemaColumnDO.builder()
.columnName(normalize(definition.getColumnName())) .columnName(normalize(definition.getColumnName()))
.columnType(definition.getDataType().toString()) .columnType(definition.getDataType().toString())
.columnComment(normalize(definition.getComment().toString())) .columnComment(Objects.isNull(definition.getComment()) ? ""
: normalize(definition.getComment().toString()))
.nullable(!text.contains(" NOT NULL")) .nullable(!text.contains(" NOT NULL"))
.primaryKey(false) .primaryKey(false)
.autoIncrement(text.contains("AUTO_INCREMENT")) .autoIncrement(text.contains("AUTO_INCREMENT"))