package com.mindant.generator.action;
|
|
import cn.hutool.db.Db;
|
import cn.hutool.db.Entity;
|
import cn.hutool.db.ds.simple.SimpleDataSource;
|
import com.mindant.generator.action.config.WebGeneratorConfig;
|
import com.mindant.generator.action.model.GenQo;
|
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* 代码生成器,可以生成实体,dao,service,controller,html,js
|
*
|
* @author stylefeng
|
* @Date 2017/5/21 12:38
|
*/
|
public class GunsCodeBatchGenerator {
|
|
public static void main(String[] args) throws Exception {
|
// 数据库名
|
String dbName = "xlzx";
|
// 数据库连接
|
String url = "jdbc:mysql://127.0.0.1:3306/"
|
+ dbName
|
+ "?autoReconnect=true&useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=CONVERT_TO_NULL&useSSL=false&serverTimezone=CTT&nullCatalogMeansCurrent=true";
|
// 用户名
|
String userName = "root";
|
// 密码
|
String password = "123456";
|
|
// 查询数据库表名、注释
|
String tableSql = "SELECT TABLE_NAME AS tableName,TABLE_COMMENT AS tableComment FROM information_schema.`TABLES` WHERE TABLE_SCHEMA = '" + dbName + "'" +
|
" AND TABLE_NAME NOT LIKE 'sys_%'" +
|
" ORDER BY TABLE_NAME ASC";
|
|
// 创建数据源
|
SimpleDataSource ds = new SimpleDataSource(url, userName, password);
|
|
// 使用数据源连接数据库
|
Db db = Db.use(ds);
|
|
List<Entity> list = db.query(tableSql);
|
|
// 查询表名、注释
|
LinkedHashMap<String, String> tableCommentMap = list.stream().collect(Collectors.toMap(e -> e.getStr("tableName"), e -> e.getStr("tableComment"), (e1, e2) -> e1, LinkedHashMap::new));
|
|
// 执行SQL查询
|
for (String k : tableCommentMap.keySet()) {
|
// 打印数据表名
|
System.out.println("\"" + k + "\",");
|
// System.out.println(k + "：" + tableCommentMap.get(k));
|
}
|
|
String[] tableNameArray = new String[]{
|
// "hr_org_approver",
|
// "hr_organization",
|
// "hr_position",
|
"t_version"
|
|
};
|
|
for (String tableName : tableNameArray) {
|
genTableModel(dbName, url, userName, password, tableName, tableCommentMap);
|
}
|
}
|
|
public static void genTableModel(String dbName, String url, String userName, String password, String tableName, Map<String, String> tableCommentMap) {
|
/*
|
* guns的生成器:
|
* guns的代码生成器可以生成controller,html页面,页面对应的js
|
*/
|
// GunsGeneratorConfig gunsGeneratorConfig = new GunsGeneratorConfig();
|
// gunsGeneratorConfig.doMpGeneration();
|
// gunsGeneratorConfig.doGunsGeneration();
|
|
// 代码生成的查询参数
|
GenQo genQo = new GenQo();
|
//数据库url
|
genQo.setUrl(url);
|
// 数据库用户名
|
genQo.setUserName(userName);
|
// 数据库密码
|
genQo.setPassword(password);
|
// 表名
|
genQo.setTableName(tableName);
|
// 作者
|
genQo.setAuthor("luodangjia");
|
// 数据库名
|
genQo.setDbName(dbName);
|
// 项目包
|
genQo.setProjectPackage("cn.stylefeng.guns");
|
genQo.setModuleName("business");
|
// 代码生成位置
|
genQo.setProjectPath("D:\\study\\PsychologicalCounseling\\common-buiness");
|
// 用户信息
|
genQo.setBizName(tableCommentMap.get(tableName));
|
genQo.setControllerSwitch(true);
|
genQo.setEntitySwitch(true);
|
genQo.setServiceSwitch(true);
|
genQo.setDaoSwitch(true);
|
genQo.setIgnoreTabelPrefix("t_");
|
|
WebGeneratorConfig webGeneratorConfig = new WebGeneratorConfig(genQo);
|
webGeneratorConfig.doMpGeneration();
|
webGeneratorConfig.doGunsGeneration();
|
}
|
|
}
|
