main
黄海 9 months ago
parent a6b5aba86c
commit e2eb455a22

@ -232,31 +232,32 @@ public class GenerateCode {
String comment = jo.getStr("comment"); String comment = jo.getStr("comment");
//主键 //主键
String key = jo.getStr("key"); String key = jo.getStr("key");
//主键类型
String key_type = jo.getStr("key_type"); String key_type = jo.getStr("key_type");
//表名 //表名
String table = jo.getStr("table"); String table = jo.getStr("table");
//向 freemaker传递的数据结构体
Map<String, Object> dataModel = new HashMap<>(); Map<String, Object> dataModel = new HashMap<>();
dataModel.put("comment", comment); dataModel.put("comment", comment);//表描述
List<Record> params = new ArrayList<>(); List<Record> params = new ArrayList<>();//用于注释,显示@para 参数名 参数含义
//迭代 //迭代
for (Map.Entry<String, String> entry : fields.entrySet()) { for (Map.Entry<String, String> entry : fields.entrySet()) {
if (!entry.getKey().equals("create_time")) { if (!entry.getKey().equals("create_time")) {
Record record = new Record(); Record record = new Record();
record.set("key", entry.getKey()); record.set("key", entry.getKey());//字段
record.set("comment", fieldsComment.get(entry.getKey())); record.set("comment", fieldsComment.get(entry.getKey()));//描述
params.add(record); params.add(record);
} }
} }
dataModel.put("params", params); dataModel.put("params", params);//用于注释,显示@para 参数名 参数含义
dataModel.put("daoName", PropKit.get("daoName")); dataModel.put("daoName", PropKit.get("daoName"));//显示dao的名称比如ym
dataModel.put("beanNameWithoutT", beanNameWithoutT); dataModel.put("beanNameWithoutT", beanNameWithoutT);//实体Bean的名称用于构建方法名
dataModel.put("fullParameters", fullParameters); dataModel.put("fullParameters", fullParameters);//包括主键的参数列表,用于构建方法的参数
dataModel.put("parameters", parameters); dataModel.put("parameters", parameters);//不包括主键的参数列表,用于构建方法的参数
dataModel.put("key", key); dataModel.put("key", key);//主键名称
dataModel.put("key_type", key_type); dataModel.put("key_type", key_type);//主键类型
dataModel.put("table", table); dataModel.put("table", table);//表名
dataModel.put("fields", fields.entrySet());
//调用FreeMarker模板引擎 //调用FreeMarker模板引擎
Configuration cfg = new Configuration(new Version(2, 3, 33));// 创建一个 Configuration 实例 Configuration cfg = new Configuration(new Version(2, 3, 33));// 创建一个 Configuration 实例
@ -271,7 +272,7 @@ public class GenerateCode {
// 加载模板 // 加载模板
Template template = cfg.getTemplate("Controller.ftl"); Template template = cfg.getTemplate("Controller.ftl");
// 渲染模板到文件 // 渲染模板到文件
Writer out = new FileWriter(codePath +"/"+ table + "_Controller.txt", StandardCharsets.UTF_8); Writer out = new FileWriter(codePath + "/" + table + "_Controller.txt", StandardCharsets.UTF_8);
template.process(dataModel, out); template.process(dataModel, out);
out.close(); out.close();
System.out.println("生成Controller层代码完成。"); System.out.println("生成Controller层代码完成。");
@ -280,7 +281,7 @@ public class GenerateCode {
// 加载模板 // 加载模板
template = cfg.getTemplate("Model.ftl"); template = cfg.getTemplate("Model.ftl");
// 渲染模板到文件 // 渲染模板到文件
out = new FileWriter(codePath +"/"+ table + "_Model.txt", StandardCharsets.UTF_8); out = new FileWriter(codePath + "/" + table + "_Model.txt", StandardCharsets.UTF_8);
template.process(dataModel, out); template.process(dataModel, out);
out.close(); out.close();
System.out.println("生成Model层代码完成。"); System.out.println("生成Model层代码完成。");

@ -5,7 +5,7 @@
*/ */
public void add${beanNameWithoutT}(${fullParameters}){ public void add${beanNameWithoutT}(${fullParameters}){
Record record = new Record(); Record record = new Record();
<#list fields as entry> <#list params as entry>
<#if entry.key == "create_time"> <#if entry.key == "create_time">
record.set("create_time", DateTime.now()); record.set("create_time", DateTime.now());
<#else> <#else>
@ -31,7 +31,7 @@ public void del${beanNameWithoutT}ById(${key_type} ${key}){
</#list> </#list>
*/ */
public void update${beanNameWithoutT}ById(${key_type} ${key} ,${fullParameters}){ public void update${beanNameWithoutT}ById(${key_type} ${key} ,${fullParameters}){
String sql="update ${table} set <#list fields as entry> String sql="update ${table} set <#list params as entry>
<#if entry.key == "create_time"> <#if entry.key == "create_time">
record.set("create_time", DateTime.now()); record.set("create_time", DateTime.now());
<#else> <#else>

@ -5,7 +5,7 @@
*/ */
public void add${beanNameWithoutT}(${fullParameters}){ public void add${beanNameWithoutT}(${fullParameters}){
Record record = new Record(); Record record = new Record();
<#list fields as entry> <#list params as entry>
<#if entry.key == "create_time"> <#if entry.key == "create_time">
record.set("create_time", DateTime.now()); record.set("create_time", DateTime.now());
<#else> <#else>
@ -31,7 +31,7 @@ public void del${beanNameWithoutT}ById(${key_type} ${key}){
</#list> </#list>
*/ */
public void update${beanNameWithoutT}ById(${key_type} ${key} ,${fullParameters}){ public void update${beanNameWithoutT}ById(${key_type} ${key} ,${fullParameters}){
String sql="update ${table} set <#list fields as entry> String sql="update ${table} set <#list params as entry>
<#if entry.key == "create_time"> <#if entry.key == "create_time">
record.set("create_time", DateTime.now()); record.set("create_time", DateTime.now());
<#else> <#else>

Loading…
Cancel
Save