public class GenUtils { public static List<String> getTemplates() { List<String> templates = new ArrayList<String>(); templates.add("templates/common/generator/domain.java.vm"); templates.add("templates/common/generator/Dao.java.vm"); //templates.add("templates/common/generator/Mapper.java.vm"); templates.add("templates/common/generator/Mapper.xml.vm"); templates.add("templates/common/generator/Service.java.vm"); templates.add("templates/common/generator/ServiceImpl.java.vm"); templates.add("templates/common/generator/Controller.java.vm"); templates.add("templates/common/generator/list.html.vm"); templates.add("templates/common/generator/add.html.vm"); templates.add("templates/common/generator/edit.html.vm"); templates.add("templates/common/generator/list.js.vm"); templates.add("templates/common/generator/add.js.vm"); templates.add("templates/common/generator/edit.js.vm"); //templates.add("templates/common/generator/menu.sql.vm"); return templates; } /** * 生成代码 */ public static void generatorCode(Map<String, String> table, List<Map<String, String>> columns, ZipOutputStream zip) { //配置信息 Configuration config = getConfig(); //表信息 TableDO tableDO = new TableDO(); tableDO.setTableName(table.get("tableName")); tableDO.setComments(table.get("tableComment")); //表名转换成Java类名 String className = tableToJava(tableDO.getTableName(), config.getString("tablePrefix"), config.getString("autoRemovePre")); tableDO.setClassName(className); tableDO.setClassname(StringUtils.uncapitalize(className)); //列信息 List<ColumnDO> columsList = new ArrayList<>(); for (Map<String, String> column : columns) { ColumnDO columnDO = new ColumnDO(); columnDO.setColumnName(column.get("columnName")); columnDO.setDataType(column.get("dataType")); columnDO.setComments(column.get("columnComment")); columnDO.setExtra(column.get("extra")); //列名转换成Java属性名 String attrName = columnToJava(columnDO.getColumnName()); columnDO.setAttrName(attrName); columnDO.setAttrname(StringUtils.uncapitalize(attrName)); //列的数据类型,转换成Java类型 String attrType = config.getString(columnDO.getDataType(), "unknowType"); columnDO.setAttrType(attrType); //是否主键 if ("PRI".equalsIgnoreCase(column.get("columnKey")) && tableDO.getPk() == null) { tableDO.setPk(columnDO); } columsList.add(columnDO); } tableDO.setColumns(columsList); //没主键,则第一个字段为主键 if (tableDO.getPk() == null) { tableDO.setPk(tableDO.getColumns().get(0)); } //设置velocity资源加载器 Properties prop = new Properties(); prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader"); Velocity.init(prop); //封装模板数据 Map<String, Object> map = new HashMap<>(16); map.put("tableName", tableDO.getTableName()); map.put("comments", tableDO.getComments()); map.put("pk", tableDO.getPk()); map.put("className", tableDO.getClassName()); map.put("classname", tableDO.getClassname()); map.put("pathName", config.getString("package").substring(config.getString("package").lastIndexOf(".") + 1)); map.put("columns", tableDO.getColumns()); map.put("package", config.getString("package")); map.put("author", config.getString("author")); map.put("email", config.getString("email")); map.put("datetime", DateUtils.format(new Date(), DateUtils.DATE_TIME_PATTERN)); VelocityContext context = new VelocityContext(map); //获取模板列表 List<String> templates = getTemplates(); for (String template : templates) { //渲染模板 StringWriter sw = new StringWriter(); Template tpl = Velocity.getTemplate(template, "UTF-8"); tpl.merge(context, sw); try { //添加到zip zip.putNextEntry(new ZipEntry(getFileName(template, tableDO.getClassname(), tableDO.getClassName(), config.getString("package").substring(config.getString("package").lastIndexOf(".") + 1)))); IOUtils.write(sw.toString(), zip, "UTF-8"); IOUtils.closeQuietly(sw); zip.closeEntry(); } catch (IOException e) { throw new BDException("渲染模板失败,表名:" + tableDO.getTableName(), e); } } } /** * 列名转换成Java属性名 */ public static String columnToJava(String columnName) { return WordUtils.capitalizeFully(columnName, new char[]{'_'}).replace("_", ""); } /** * 表名转换成Java类名 */ public static String tableToJava(String tableName, String tablePrefix, String autoRemovePre) { if (Constant.AUTO_REOMVE_PRE.equals(autoRemovePre)) { tableName = tableName.substring(tableName.indexOf("_") + 1); } if (StringUtils.isNotBlank(tablePrefix)) { tableName = tableName.replace(tablePrefix, ""); } return columnToJava(tableName); } /** * 获取配置信息 */ public static Configuration getConfig() { try { return new PropertiesConfiguration("generator.properties"); } catch (ConfigurationException e) { throw new BDException("获取配置文件失败,", e); } } /** * 获取文件名 */ public static String getFileName(String template, String classname, String className, String packageName) { String packagePath = "main" + File.separator + "java" + File.separator; //String modulesname=config.getString("packageName"); if (StringUtils.isNotBlank(packageName)) { packagePath += packageName.replace(".", File.separator) + File.separator; } if (template.contains("domain.java.vm")) { return packagePath + "domain" + File.separator + className + "DO.java"; } if (template.contains("Dao.java.vm")) { return packagePath + "dao" + File.separator + className + "Dao.java"; } // if(template.contains("Mapper.java.vm")){ // return packagePath + "dao" + File.separator + className + "Mapper.java"; // } if (template.contains("Service.java.vm")) { return packagePath + "service" + File.separator + className + "Service.java"; } if (template.contains("ServiceImpl.java.vm")) { return packagePath + "service" + File.separator + "impl" + File.separator + className + "ServiceImpl.java"; } if (template.contains("Controller.java.vm")) { return packagePath + "controller" + File.separator + className + "Controller.java"; } if (template.contains("Mapper.xml.vm")) { return "main" + File.separator + "resources" + File.separator + "mapper" + File.separator + packageName + File.separator + className + "Mapper.xml"; } if (template.contains("list.html.vm")) { return "main" + File.separator + "resources" + File.separator + "templates" + File.separator + packageName + File.separator + classname + File.separator + classname + ".html"; // + "modules" + File.separator + "generator" + File.separator + className.toLowerCase() + ".html"; } if (template.contains("add.html.vm")) { return "main" + File.separator + "resources" + File.separator + "templates" + File.separator + packageName + File.separator + classname + File.separator + "add.html"; } if (template.contains("edit.html.vm")) { return "main" + File.separator + "resources" + File.separator + "templates" + File.separator + packageName + File.separator + classname + File.separator + "edit.html"; } if (template.contains("list.js.vm")) { return "main" + File.separator + "resources" + File.separator + "static" + File.separator + "js" + File.separator + "appjs" + File.separator + packageName + File.separator + classname + File.separator + classname + ".js"; // + "modules" + File.separator + "generator" + File.separator + className.toLowerCase() + ".js"; } if (template.contains("add.js.vm")) { return "main" + File.separator + "resources" + File.separator + "static" + File.separator + "js" + File.separator + "appjs" + File.separator + packageName + File.separator + classname + File.separator + "add.js"; } if (template.contains("edit.js.vm")) { return "main" + File.separator + "resources" + File.separator + "static" + File.separator + "js" + File.separator + "appjs" + File.separator + packageName + File.separator + classname + File.separator + "edit.js"; } // if(template.contains("menu.sql.vm")){ // return className.toLowerCase() + "_menu.sql"; // } return null; }
generator.properties
#\u4EE3\u7801\u751F\u6210\u5668\uFF0C\u914D\u7F6E\u4FE1\u606F #\u5305\u540D package=com.bootdo.system #\u4F5C\u8005 author=chglee #Email email=1992lcg@163.com #\u81EA\u52A8\u53BB\u9664\u8868\u524D\u7F00\uFF0C\u9ED8\u8BA4\u662F true autoRemovePre=true #\u8868\u524D\u7F00(\u7C7B\u540D\u4E0D\u4F1A\u5305\u542B\u8868\u524D\u7F00) tablePrefix=sys_ #\u7C7B\u578B\u8F6C\u6362\uFF0C\u914D\u7F6E\u4FE1\u606F tinyint=Integer smallint=Integer mediumint=Integer int=Integer integer=Integer bigint=Long float=Float double=Double decimal=BigDecimal bit=Boolean char=String varchar=String tinytext=String text=String mediumtext=String longtext=String date=Date datetime=Date timestamp=Date