Преглед на файлове

完善代码生成器的 h2 脚本

YunaiV преди 3 години
родител
ревизия
0f5477a4df
променени са 1 файла, в които са добавени 14 реда и са изтрити 3 реда
  1. 14 3
      yudao-module-infra/yudao-module-infra-biz/src/main/resources/codegen/sql/h2.vm

+ 14 - 3
yudao-module-infra/yudao-module-infra-biz/src/main/resources/codegen/sql/h2.vm

@@ -1,19 +1,30 @@
 -- 将该建表 SQL 语句,添加到 yudao-module-${table.moduleName}-biz 模块的 test/resources/sql/create_tables.sql 文件里
 CREATE TABLE IF NOT EXISTS "${table.tableName}" (
 #foreach ($column in $columns)
+#if (${column.javaType} == 'Long')
+    #set ($dataType='bigint')
+#elseif (${column.javaType} == 'Integer')
+    #set ($dataType='int')
+#elseif (${column.javaType} == 'Boolean')
+    #set ($dataType='bit')
+#elseif (${column.javaType} == 'Date')
+    #set ($dataType='datetime')
+#else
+    #set ($dataType='varchar')
+#end
     #if (${column.primaryKey})##处理主键
-    "${column.javaField}"#if (${column.javaType} == 'String') ${column.dataType} NOT NULL#else ${column.dataType} NOT NULL GENERATED BY DEFAULT AS IDENTITY#end,
+    "${column.javaField}"#if (${column.javaType} == 'String') ${dataType} NOT NULL#else ${dataType} NOT NULL GENERATED BY DEFAULT AS IDENTITY#end,
     #else
     #if (${column.columnName} == 'create_time')
     "create_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
     #elseif (${column.columnName} == 'update_time')
     "update_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
     #elseif (${column.columnName} == 'creator' || ${column.columnName} == 'updater')
-    "${column.columnName}" ${column.dataType} DEFAULT '',
+    "${column.columnName}" ${dataType} DEFAULT '',
     #elseif (${column.columnName} == 'deleted')
     "deleted" bit NOT NULL DEFAULT FALSE,
     #else
-    "${column.columnName}" ${column.dataType}#if (${column.nullable} == false) NOT NULL#end,
+    "${column.columnName}" ${dataType}#if (${column.nullable} == false) NOT NULL#end,
     #end
     #end
 #end