springboot+shiro+mybatis+redis+通用mapper+aop+CodeGenerator

package com.example.demo; import com.example.demo.core.configurer.ProjectConstant; import com.google.common.base.CaseFormat; import freemarker.template.TemplateExceptionHandler; import org.apache.commons.lang3.StringUtils; import org.mybatis.generator.api.MyBatisGenerator; import org.mybatis.generator.config.*; import org.mybatis.generator.internal.DefaultShellCallback; import java.io.File; import java.io.FileWriter; import java.io.IOException; import java.text.SimpleDateFormat; import java.util.*; /** * @author jjp * @Description: 代码生成器，根据数据表名称生成对应的Model、Mapper、Service、Controller简化开发。 * @date 2018/4/23 20:28 */ public class CodeGenerator { // JDBC配置，请修改为你项目的实际配置 private static final String JDBC_URL = "jdbc:mysql://localhost:3306/demo"; private static final String JDBC_USERNAME = "root"; private static final String JDBC_PASSWORD = "root"; private static final String JDBC_DIVER_CLASS_NAME = "com.mysql.jdbc.Driver"; // 模板位置 private static final String TEMPLATE_FILE_PATH = "src/test/java/resources/template/generator"; private static final String JAVA_PATH = "src/main/java"; // java文件路径 private static final String RESOURCES_PATH = "src/main/resources";// 资源文件路径 // 生成的Service存放路径 private static final String PACKAGE_PATH_SERVICE = packageConvertPath(ProjectConstant.SERVICE_PACKAGE); // 生成的Service实现存放路径 private static final String PACKAGE_PATH_SERVICE_IMPL = packageConvertPath(ProjectConstant.SERVICE_IMPL_PACKAGE); // 生成的Controller存放路径 private static final String PACKAGE_PATH_CONTROLLER = packageConvertPath(ProjectConstant.CONTROLLER_PACKAGE); // @author private static final String AUTHOR = "jjp"; // @date private static final String DATE = new SimpleDateFormat("yyyy/MM/dd HH:mm").format(new Date()); /** * genCode("输入表名"); * @param args */ public static void main(String[] args) { genCode("sys_permission_init"); } /** * 通过数据表名称生成代码，Model 名称通过解析数据表名称获得，下划线转大驼峰的形式。 如输入表名称 "t_user_detail" 将生成 * TUserDetail、TUserDetailMapper、TUserDetailService ... * * @param tableNames 数据表名称... */ public static void genCode(String... tableNames) { for (String tableName : tableNames) { genCode(tableName); } } /** * 通过数据表名称生成代码 如输入表名称 "user_info" * 将生成 UserInfo、UserInfoMapper、UserInfoService ... * * @param tableName 数据表名称 */ public static void genCode(String tableName) { genModelAndMapper(tableName); genService(tableName); genController(tableName); } public static void genModelAndMapper(String tableName) { Context context = getContext(); JDBCConnectionConfiguration jdbcConnectionConfiguration = getJDBCConnectionConfiguration(); context.setJdbcConnectionConfiguration(jdbcConnectionConfiguration); PluginConfiguration pluginConfiguration = getPluginConfiguration(); context.addPluginConfiguration(pluginConfiguration); JavaModelGeneratorConfiguration javaModelGeneratorConfiguration = getJavaModelGeneratorConfiguration(); context.setJavaModelGeneratorConfiguration(javaModelGeneratorConfiguration); SqlMapGeneratorConfiguration sqlMapGeneratorConfiguration = getSqlMapGeneratorConfiguration(); context.setSqlMapGeneratorConfiguration(sqlMapGeneratorConfiguration); JavaClientGeneratorConfiguration javaClientGeneratorConfiguration = getJavaClientGeneratorConfiguration(); context.setJavaClientGeneratorConfiguration(javaClientGeneratorConfiguration); TableConfiguration tableConfiguration = new TableConfiguration(context); tableConfiguration.setTableName(tableName); tableConfiguration.setDomainObjectName(null); context.addTableConfiguration(tableConfiguration); List<String> warnings; MyBatisGenerator generator; try { Configuration config = new Configuration(); config.addContext(context); config.validate(); boolean overwrite = true; DefaultShellCallback callback = new DefaultShellCallback(overwrite); warnings = new ArrayList<>(); generator = new MyBatisGenerator(config, callback, warnings); generator.generate(null); } catch (Exception e) { throw new RuntimeException("生成Model和Mapper失败", e); } if (generator.getGeneratedJavaFiles().isEmpty() || generator.getGeneratedXmlFiles().isEmpty()) { throw new RuntimeException("生成Model和Mapper失败：" + warnings); } String modelName = tableNameConvertUpperCamel(tableName); System.out.println(modelName + ".java 生成成功"); System.out.println(modelName + "Mapper.java 生成成功"); System.out.println(modelName + "Mapper.xml 生成成功"); } public static void genService(String tableName) { try { freemarker.template.Configuration cfg = getConfiguration(); //模板所需要的参数 Map<String, Object> data = new HashMap<>(); data.put("date", DATE); data.put("author", AUTHOR); String modelNameUpperCamel = tableNameConvertUpperCamel(tableName); data.put("modelNameUpperCamel", modelNameUpperCamel); data.put("modelNameLowerCamel", CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, modelNameUpperCamel)); data.put("basePackage", ProjectConstant.BASE_PACKAGE); data.put("basePackageService", ProjectConstant.SERVICE_PACKAGE); data.put("basePackageServiceImpl", ProjectConstant.SERVICE_IMPL_PACKAGE); data.put("basePackageModel", ProjectConstant.MODEL_PACKAGE); data.put("basePackageDao", ProjectConstant.MAPPER_PACKAGE); File file = new File(JAVA_PATH + PACKAGE_PATH_SERVICE + modelNameUpperCamel + "Service.java"); if (!file.getParentFile().exists()) { file.getParentFile().mkdirs(); } cfg.getTemplate("service.ftl").process(data, new FileWriter(file)); System.out.println(modelNameUpperCamel + "Service.java 生成成功"); File file1 = new File(JAVA_PATH + PACKAGE_PATH_SERVICE_IMPL + modelNameUpperCamel + "ServiceImpl.java"); if (!file1.getParentFile().exists()) { file1.getParentFile().mkdirs(); } cfg.getTemplate("service-impl.ftl").process(data, new FileWriter(file1)); System.out.println(modelNameUpperCamel + "ServiceImpl.java 生成成功"); } catch (Exception e) { throw new RuntimeException("生成Service失败", e); } } public static void genController(String tableName) { try { freemarker.template.Configuration cfg = getConfiguration(); Map<String, Object> data = new HashMap<>(); data.put("date", DATE); data.put("author", AUTHOR); String modelNameUpperCamel = tableNameConvertUpperCamel(tableName); data.put("baseRequestMapping", CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, modelNameUpperCamel)); data.put("modelNameUpperCamel", modelNameUpperCamel); data.put("modelNameLowerCamel", CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_CAMEL, modelNameUpperCamel)); data.put("basePackage", ProjectConstant.BASE_PACKAGE);