人人开源项目的代码生成器，可在线生成entity、xml、dao、service、vue、sql代码，减少70%以上的开发任务

package io.renren.utils; import io.renren.config.MongoManager; import io.renren.entity.ColumnEntity; import io.renren.entity.TableEntity; import io.renren.entity.mongo.MongoDefinition; import io.renren.entity.mongo.MongoGeneratorEntity; import org.apache.commons.configuration.Configuration; import org.apache.commons.configuration.ConfigurationException; import org.apache.commons.configuration.PropertiesConfiguration; import org.apache.commons.io.IOUtils; import org.apache.commons.lang.StringUtils; import org.apache.commons.lang.WordUtils; import org.apache.velocity.Template; import org.apache.velocity.VelocityContext; import org.apache.velocity.app.Velocity; import java.io.File; import java.io.IOException; import java.io.StringWriter; import java.util.*; import java.util.zip.ZipEntry; import java.util.zip.ZipOutputStream; /** * 代码生成器 工具类 * * @author chenshun * @email sunlightcs@gmail.com * @date 2016年12月19日 下午11:40:24 */ public class GenUtils { private static String currentTableName; public static List<String> getTemplates() { List<String> templates = new ArrayList<String>(); templates.add("template/Entity.java.vm"); templates.add("template/Dao.xml.vm"); templates.add("template/menu.sql.vm"); templates.add("template/Service.java.vm"); templates.add("template/ServiceImpl.java.vm"); templates.add("template/Controller.java.vm"); templates.add("template/Dao.java.vm"); templates.add("template/index.vue.vm"); templates.add("template/add-or-update.vue.vm"); if (MongoManager.isMongo()) { // mongo不需要mapper、sql 实体类需要替换 templates.remove(0); templates.remove(1); templates.remove(2); templates.add("template/MongoEntity.java.vm"); } return templates; } public static List<String> getMongoChildTemplates() { List<String> templates = new ArrayList<String>(); templates.add("template/MongoChildrenEntity.java.vm"); return templates; } /** * 生成代码 */ public static void generatorCode(Map<String, String> table, List<Map<String, String>> columns, ZipOutputStream zip) { //配置信息 Configuration config = getConfig(); boolean hasBigDecimal = false; boolean hasList = false; //表信息 TableEntity tableEntity = new TableEntity(); tableEntity.setTableName(table.get("tableName")); tableEntity.setComments(table.get("tableComment")); //表名转换成Java类名 String className = tableToJava(tableEntity.getTableName(), config.getStringArray("tablePrefix")); tableEntity.setClassName(className); tableEntity.setClassname(StringUtils.uncapitalize(className)); //列信息 List<ColumnEntity> columsList = new ArrayList<>(); for (Map<String, String> column : columns) { ColumnEntity columnEntity = new ColumnEntity(); columnEntity.setColumnName(column.get("columnName")); columnEntity.setDataType(column.get("dataType")); columnEntity.setComments(column.get("columnComment")); columnEntity.setExtra(column.get("extra")); //列名转换成Java属性名 String attrName = columnToJava(columnEntity.getColumnName()); columnEntity.setAttrName(attrName); columnEntity.setAttrname(StringUtils.uncapitalize(attrName)); //列的数据类型，转换成Java类型 String attrType = config.getString(columnEntity.getDataType(), columnToJava(columnEntity.getDataType())); columnEntity.setAttrType(attrType); if (!hasBigDecimal && attrType.equals("BigDecimal")) { hasBigDecimal = true; } if (!hasList && "array".equals(columnEntity.getExtra())) { hasList = true; } //是否主键 if ("PRI".equalsIgnoreCase(column.get("columnKey")) && tableEntity.getPk() == null) { tableEntity.setPk(columnEntity); } columsList.add(columnEntity); } tableEntity.setColumns(columsList); //没主键，则第一个字段为主键 if (tableEntity.getPk() == null) { tableEntity.setPk(tableEntity.getColumns().get(0)); } //设置velocity资源加载器 Properties prop = new Properties(); prop.put("file.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader"); Velocity.init(prop); String mainPath = config.getString("mainPath"); mainPath = StringUtils.isBlank(mainPath) ? "io.renren" : mainPath; //封装模板数据 Map<String, Object> map = new HashMap<>(); map.put("tableName", tableEntity.getTableName()); map.put("comments", tableEntity.getComments()); map.put("pk", tableEntity.getPk()); map.put("className", tableEntity.getClassName()); map.put("classname", tableEntity.getClassname()); map.put("pathName", tableEntity.getClassname().toLowerCase()); map.put("columns", tableEntity.getColumns()); map.put("hasBigDecimal", hasBigDecimal); map.put("hasList", hasList); map.put("mainPath", mainPath); map.put("package", config.getString("package")); map.put("moduleName", config.getString("moduleName")); 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, tableEntity.getClassName(), config.getString("package"), config.getString("moduleName")))); IOUtils.write(sw.toString(), zip, "UTF-8"); IOUtils.closeQuietly(sw); zip.closeEntry(); } catch (IOException e) { throw new RRException("渲染模板失败，表名：" + tableEntity.getTableName(), e); } } } /** * 生成mongo其他实体类的代码 */ public static void generatorMongoCode(String[] tableNames, ZipOutputStream zip) { for (String tableName : tableNames) { MongoDefinition info = MongoManager.getInfo(tableName); currentTableName = tableName; List<MongoGeneratorEntity> childrenInfo = info.getChildrenInfo(tableName); childrenInfo.remove(0); for (MongoGeneratorEntity mongoGeneratorEntity : childrenInfo) { generatorChildrenBeanCode(mongoGeneratorEntity, zip); } } } private static void generatorChildrenBeanCode(MongoGeneratorEntity mongoGeneratorEntity, ZipOutputStream zip) { //配置信息 Configuration config = getConfig(); boolean hasList = false; //表信息 TableEntity tableEntity = mongoGeneratorEntity.toTableEntity(); //表名转换成Java类名 String className = tableToJava(tableEntity.getTableName(), config.getStringArray("tablePrefix")); tableEntity.setClassName(className); tableEntity.setClassname(StringUtils.uncapitalize(className)); //列信息 List<ColumnEntity> columsList = new ArrayList<>(); for (Map<String, String> column : mongoGeneratorEntity.getColumns()) {