|
@@ -12,8 +12,6 @@ import java.util.Collections;
|
|
|
public class Generator {
|
|
|
|
|
|
|
|
|
-
|
|
|
-
|
|
|
public static void main(String[] args) {
|
|
|
String db_url = "jdbc:mysql://mysql.aimpdev.svc.k5.bigtree.zone:3306/aimp_model";
|
|
|
String db_username = "test_user";
|
|
@@ -21,18 +19,20 @@ public class Generator {
|
|
|
|
|
|
String javaDir = "/Users/alvin/bigtree/aimp-manager/aimp-manager-service/src/main/java";
|
|
|
String parentPackage = "com.bigtreefinance.aimp.manager";
|
|
|
+
|
|
|
//String javaDir = "/Users/alvin/bigtree/aimp-actuator/aimp-actuator-service/src/main/java";
|
|
|
//String parentPackage = "com.bigtreefinance.aimp.actuator";
|
|
|
- //String tableName = "aimp_list,aimp_model,aimp_model_log,aimp_model_version,aimp_node_config,aimp_operation_log,aimp_process,aimp_process_node,aimp_sample,";
|
|
|
- String tableName = "aimp_model_version";
|
|
|
+
|
|
|
+ String tableName = "aimp_list,aimp_model,aimp_model_log,aimp_model_version,aimp_node_config,aimp_operation_log,aimp_process,aimp_process_node,aimp_sample,";
|
|
|
+ //String tableName = "aimp_sample";
|
|
|
|
|
|
String author = "bigtree";
|
|
|
String tablePrefix = "";
|
|
|
|
|
|
-
|
|
|
generatorToFile(db_url, db_username, db_password, author, javaDir, parentPackage, tableName, tablePrefix);
|
|
|
}
|
|
|
|
|
|
+
|
|
|
private static void generatorToFile(String db_url, String db_username, String db_password, String author,
|
|
|
String javaDir, String parentPackage, String tableName, String tablePrefix) {
|
|
|
FastAutoGenerator.create(db_url, db_username, db_password)
|
|
@@ -59,9 +59,14 @@ public class Generator {
|
|
|
//实体类
|
|
|
builder.entityBuilder()
|
|
|
.enableLombok()
|
|
|
+ .logicDeleteColumnName("use_status")
|
|
|
.addTableFills(
|
|
|
new Column("create_time", FieldFill.INSERT),
|
|
|
- new Column("update_time", FieldFill.INSERT_UPDATE)
|
|
|
+ new Column("create_id", FieldFill.INSERT),
|
|
|
+ new Column("create_name", FieldFill.INSERT),
|
|
|
+ new Column("edit_id", FieldFill.INSERT_UPDATE),
|
|
|
+ new Column("edit_time", FieldFill.INSERT_UPDATE),
|
|
|
+ new Column("edit_name", FieldFill.INSERT_UPDATE)
|
|
|
);
|
|
|
// 设置需要生成的表名
|
|
|
for (String s : tableName.split(",")) {
|
|
@@ -73,6 +78,7 @@ public class Generator {
|
|
|
|
|
|
})
|
|
|
.templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板,默认的是Velocity引擎模板
|
|
|
+ .templateConfig(builder -> builder.entity("templates/entity1.java"))
|
|
|
//.templateConfig(builder -> builder.controller("templates/controller1.java")) // 设置为空则不会生成该文件
|
|
|
//.templateConfig(builder -> builder.service("templates/service1.java"))
|
|
|
//.templateConfig(builder -> builder.serviceImpl("templates/serviceImpl1.java"))
|