基于Javascript的J4CF企业快速开发脚手架设计源码

package com.j4cf.common.util; import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert; import com.baomidou.mybatisplus.generator.config.rules.DbColumnType; import com.baomidou.mybatisplus.generator.config.rules.DbType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import org.apache.commons.lang.ObjectUtils; import org.apache.velocity.VelocityContext; import java.io.File; import java.text.SimpleDateFormat; import java.util.*; import static com.j4cf.common.util.StringUtil.lineToHump; /** * @Description: MB代码生成类 * @Author: LongRou * @CreateDate: 2018 2018/3/14 14:23 * @Version: 1.0 **/ public class MybatisPlusGeneratorUtil { // Controller模板路径 private static String controller_vm = "/template/Controller.vm"; // Service模板路径 private static String service_vm = "/template/Service.vm"; // ServiceImpl模板路径 private static String serviceImpl_vm = "/template/ServiceImpl.vm"; /** * 根据模板生成generatorConfig.xml文件 * @param jdbcDriver 驱动路径 * @param jdbcUrl 链接 * @param jdbcUsername 帐号 * @param jdbcPassword 密码 * @param module 项目模块 * @param database 数据库 * @param tablePrefix 表前缀 * @param packageName 包名 */ public static void generator( String jdbcDriver, String jdbcUrl, String jdbcUsername, String jdbcPassword, String module, String database, String tablePrefix, String packageName, Map<String, String> lastInsertIdTables) throws Exception{ String os = System.getProperty("os.name"); if (os.toLowerCase().startsWith("win")) { service_vm = MybatisPlusGeneratorUtil.class.getResource(service_vm).getPath().replaceFirst("/", ""); serviceImpl_vm = MybatisPlusGeneratorUtil.class.getResource(serviceImpl_vm).getPath().replaceFirst("/", ""); controller_vm = MybatisPlusGeneratorUtil.class.getResource(controller_vm).getPath().replaceFirst("/", ""); } else { service_vm = MybatisPlusGeneratorUtil.class.getResource(service_vm).getPath(); serviceImpl_vm = MybatisPlusGeneratorUtil.class.getResource(serviceImpl_vm).getPath(); controller_vm = MybatisPlusGeneratorUtil.class.getResource(controller_vm).getPath(); } String targetProject = module; String basePath = MybatisPlusGeneratorUtil.class.getResource("/").getPath().replace("/target/classes/", "").replace(targetProject, "").replaceFirst("/", ""); //String generatorConfigXml = MybatisGeneratorUtil.class.getResource("/").getPath().replace("/target/classes/", "") + "/src/main/resources/generatorConfig.xml"; targetProject = basePath + targetProject; String sql = "SELECT table_name FROM INFORMATION_SCHEMA.TABLES WHERE table_schema = '" + database + "' AND table_name LIKE '" + tablePrefix + "_%';"; System.out.println("========== 开始生成generatorConfig.xml文件 =========="); List<Map<String, Object>> tables = new ArrayList<>(); List<String> tablesName = new ArrayList<>(); try { VelocityContext context = new VelocityContext(); Map<String, Object> table; // 查询定制前缀项目的所有表 JdbcUtil jdbcUtil = new JdbcUtil(jdbcDriver, jdbcUrl, jdbcUsername, AESUtil.aesDecode(jdbcPassword)); List<Map> result = jdbcUtil.selectByParams(sql, null); for (Map map : result) { System.out.println(map.get("TABLE_NAME")); table = new HashMap<>(2); table.put("table_name", map.get("TABLE_NAME")); tablesName.add((String) table.get("table_name")); table.put("model_name", lineToHump(ObjectUtils.toString(map.get("TABLE_NAME")))); tables.add(table); } jdbcUtil.release(); // String targetProjectSqlMap = basePath + module + "/" + module + "-rest-service"; // context.put("tables", tables); // context.put("generator_javaModelGenerator_targetPackage", packageName + ".dao.model"); // context.put("generator_sqlMapGenerator_targetPackage", packageName + ".dao.mapper"); // context.put("generator_javaClientGenerator_targetPackage", packageName + ".dao.mapper"); // context.put("targetProject", targetProject); // context.put("targetProject_sqlMap", targetProjectSqlMap); // context.put("generator_jdbc_password", AESUtil.aesDecode(jdbcPassword)); // context.put("last_insert_id_tables", lastInsertIdTables); // VelocityUtil.generate(generatorConfig_vm, generatorConfigXml, context); // 删除旧代码 deleteDir(new File(targetProject + "/src/main/java/" + packageName.replaceAll("\\.", "/") + "/dao/entity")); deleteDir(new File(targetProject + "/src/main/java/" + packageName.replaceAll("\\.", "/") + "/dao/mapper")); //deleteDir(new File(targetProjectSqlMap + "/src/main/java/" + packageName.replaceAll("\\.", "/") + "/dao/mapper")); } catch (Exception e) { e.printStackTrace(); } System.out.println("========== 结束生成generatorConfig.xml文件 =========="); System.out.println("========== 开始运行MybatisGenerator =========="); // List<String> warnings = new ArrayList<>(); // File configFile = new File(generatorConfigXml); // ConfigurationParser cp = new ConfigurationParser(warnings); // Configuration config = cp.parseConfiguration(configFile); // DefaultShellCallback callback = new DefaultShellCallback(true); // MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings); // myBatisGenerator.generate(null); // for (String warning : warnings) { // System.out.println(warning); // } AutoGenerator mpg = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig(); gc.setOutputDir(targetProject+"/src/main/java/"); gc.setFileOverride(true); gc.setOpen(false); gc.setActiveRecord(true);// 开启 activeRecord 模式 gc.setEnableCache(false);// XML 二级缓存 gc.setBaseResultMap(true);// XML ResultMap gc.setBaseColumnList(false);// XML columList gc.setAuthor("LongRou"); // 自定义文件命名，注意 %s 会自动填充表实体属性！ // gc.setMapperName("%sDao"); // gc.setXmlName("%sDao"); // gc.setServiceName("MP%sService"); // gc.setServiceImplName("%sServiceDiy"); // gc.setControllerName("%sAction"); mpg.setGlobalConfig(gc); // 数据源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setDbType(DbType.MYSQL); dsc.setTypeConvert(new MySqlTypeConvert(){ // 自定义数据库表字段类型转换【可选】 @Override public DbColumnType processTypeConvert(String fieldType) { System.out.println("转换类型：" + fieldType); return super.processTypeConvert(fieldType); } }); dsc.setDriverName(jdbcDriver); dsc.setUsername(jdbcUsername); dsc.setPassword(AESUtil.aesDecode(jdbcPassword)); dsc.setUrl(jdbcUrl); mpg.setDataSource(dsc); // 策略配置 StrategyConfig strategy = new StrategyConfig(); // strategy.setCapitalMode(true);// 全局大写命名 ORACLE 注意 //strategy.setTablePrefix(new String[] { tablePrefix });// 此处可以修改为您的表前缀 strategy.setNaming(NamingStrategy.underline_to_