BJ_ZTEITS-Developers / parking-cloud-yudao

Commit b66d84535490824a27a75875218c5f678f177a78

Authored by chaocloud
1 parent 7991d685

代码自动生成工具，当表注释为空时，根据SQL导入失败问题修复

Inline Side-by-side
Showing 1 changed file with 8 additions and 1 deletions
yudao-admin-server/src/main/java/cn/iocoder/yudao/adminserver/modules/tool/service/codegen/impl/ToolCodegenSQLParser.java
  View file @b66d845
@@ -64,10 +64,17 @@ public class ToolCodegenSQLParser { @@ -64,10 +64,17 @@ public class ToolCodegenSQLParser {
64 private static ToolSchemaTableDO parseTable(SQLCreateTableStatement statement) { 64 private static ToolSchemaTableDO parseTable(SQLCreateTableStatement statement) {
65 return ToolSchemaTableDO.builder() 65 return ToolSchemaTableDO.builder()
66 .tableName(statement.getTableSource().getTableName(true)) 66 .tableName(statement.getTableSource().getTableName(true))
67 - .tableComment(((SQLCharExpr) statement.getComment()).getText()) 67 + .tableComment(getCommentText(statement))
68 .build(); 68 .build();
69 } 69 }
70 70
  71 + private static String getCommentText(SQLCreateTableStatement statement) {
  72 + if (statement == null || statement.getComment() == null) {
  73 + return "";
  74 + }
  75 + return ((SQLCharExpr) statement.getComment()).getText();
  76 + }
  77 +
71 private static List<ToolSchemaColumnDO> parseColumns(SQLCreateTableStatement statement) { 78 private static List<ToolSchemaColumnDO> parseColumns(SQLCreateTableStatement statement) {
72 List<ToolSchemaColumnDO> columns = new ArrayList<>(); 79 List<ToolSchemaColumnDO> columns = new ArrayList<>();
73 statement.getTableElementList().forEach(element -> parseColumn(columns, element)); 80 statement.getTableElementList().forEach(element -> parseColumn(columns, element));