|
@@ -3,6 +3,9 @@ package com.zheng.cms.dao;
|
|
|
import com.zheng.common.util.MybatisGeneratorUtil;
|
|
|
import com.zheng.common.util.PropertiesFileUtil;
|
|
|
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
/**
|
|
|
* 代码生成类
|
|
|
* Created by ZhangShuzheng on 2017/1/10.
|
|
@@ -18,13 +21,18 @@ public class Generator {
|
|
|
private static String JDBC_URL = PropertiesFileUtil.getInstance("generator").get("generator.jdbc.url");
|
|
|
private static String JDBC_USERNAME = PropertiesFileUtil.getInstance("generator").get("generator.jdbc.username");
|
|
|
private static String JDBC_PASSWORD = PropertiesFileUtil.getInstance("generator").get("generator.jdbc.password");
|
|
|
+ // 需要insert后返回主键的表配置,key:表名,value:主键名
|
|
|
+ private static Map<String, String> LAST_INSERT_ID_TABLES = new HashMap<>();
|
|
|
+ static {
|
|
|
+
|
|
|
+ }
|
|
|
|
|
|
/**
|
|
|
* 自动代码生成
|
|
|
* @param args
|
|
|
*/
|
|
|
public static void main(String[] args) throws Exception {
|
|
|
- MybatisGeneratorUtil.generator(JDBC_DRIVER, JDBC_URL, JDBC_USERNAME, JDBC_PASSWORD, MODULE, DATABASE, TABLE_PREFIX, PACKAGE_NAME);
|
|
|
+ MybatisGeneratorUtil.generator(JDBC_DRIVER, JDBC_URL, JDBC_USERNAME, JDBC_PASSWORD, MODULE, DATABASE, TABLE_PREFIX, PACKAGE_NAME, LAST_INSERT_ID_TABLES);
|
|
|
}
|
|
|
|
|
|
}
|