Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.coderepositories.jcommons.tools.mbg.MBGTemplate Maven / Gradle / Ivy
package com.github.coderepositories.jcommons.tools.mbg;
import com.github.coderepositories.jcommons.core.X;
import com.github.coderepositories.jcommons.tools.mbg.config.*;
import com.github.coderepositories.jcommons.tools.mbg.config.custom.CustomConfiguration;
import com.github.coderepositories.jcommons.tools.mbg.config.custom.CustomContent;
import com.github.coderepositories.jcommons.tools.mbg.config.custom.CustomTableConfig;
import com.github.coderepositories.jcommons.tools.mbg.config.table.Table;
import com.github.coderepositories.jcommons.tools.mbg.dbinfo.DbInfos;
import com.github.coderepositories.jcommons.tools.mbg.dbinfo.TableMetadata;
import com.google.common.io.Files;
import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.annotations.XStreamAlias;
import org.apache.commons.dbutils.DbUtils;
import java.io.File;
import java.io.FileWriter;
import java.io.FilenameFilter;
import java.io.IOException;
import java.nio.charset.Charset;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.*;
/**
* Mybatis Generator Teamplate
*
* @author zhangguangyong
*
* 2016年2月2日 上午11:19:53
*/
@SuppressWarnings("all")
public class MBGTemplate {
/**
* 编译后输出目录 classes
*/
static final String CLASSES_DIR = ClassLoader.getSystemResource("").getFile();
/**
* 当前项目路径
*/
static final String PROJECT_DIR = new File(CLASSES_DIR).getParentFile().getParent();
/**
* Mybatis Generator Config File
*/
static final String GENERATOR_CONFIG_FILE = "generatorConfig.xml";
/**
* 自定义配置文件
*/
static final String CUSTOM_CONFIG_FILE = "customConfig.xml";
/**
* 自定义内容文件
*/
static final String CUSTOM_CONTENT_FILE = "CustomContent";
/**
* Entity.java 备份文件夹
*/
static final String JAVA_MODEL_FOLDER = "javaModel";
/**
* Mapper.java 备份文件夹
*/
static final String JAVA_CLIENT_FOLDER = "javaClient";
/**
* Mapper.xml 备份文件夹
*/
static final String SQL_MAP_FOLDER = "sqlMap";
/**
* Entity.java 文件后缀
*/
static final String JAVA_MODEL_FILE_SUFFIX = ".java";
/**
* Mapper.java 文件后缀
*/
static final String JAVA_CLIENT_FILE_SUFFIX = "Mapper.java";
/**
* Mapper.xml 文件后缀
*/
static final String SQL_MAP_FILE_SUFFIX = "Mapper.xml";
/**
* 自定义内容记录文件中，文件行的标识
*/
static final String CUSTOM_CONTENT_FILE_LINE_FLAG = "#";
/**
* 默认字符
*/
static final Charset DEFAULT_CHARSET = Charset.defaultCharset();
/**
* 换行符
*/
static final String LINE_SEPARATOR = System.lineSeparator();
public static void start() {
try {
// 加载配置信息
Config config = loadConfig(GENERATOR_CONFIG_FILE, CUSTOM_CONFIG_FILE);
// 保存自定义内容
File customContent = saveCustomContent(config);
// 合并配置信息
GeneratorConfiguration mergeConfig = mergeConfig(config);
// 生成代码
generateCode(mergeConfig, customContent);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 启动接口
*/
public static void start(File generatorConfig, CustomConfiguration customConfig) {
try {
// 加载配置信息
Config config = loadConfig(generatorConfig, customConfig);
// 保存自定义内容
File customContent = saveCustomContent(config);
// 合并配置信息
GeneratorConfiguration mergeConfig = mergeConfig(config);
// 生成代码
generateCode(mergeConfig, customContent);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 加载配置信息
*
* @param generatorConfig
* @param customConfig
* @return
* @throws SQLException
* @throws IOException
*/
public static Config loadConfig(String generatorConfig, String customConfig) throws SQLException, IOException {
File generatorConfigFile = new File(CLASSES_DIR, generatorConfig);
File customConfigFile = new File(CLASSES_DIR, customConfig);
GeneratorConfiguration generatorCoinfgBean = new GeneratorConfiguration();
CustomConfiguration customConfigBean = new CustomConfiguration();
readConfig(generatorConfigFile, generatorCoinfgBean);
readConfig(customConfigFile, customConfigBean);
Config config = new Config();
config.setGeneratorConfig(generatorCoinfgBean);
config.setCustomConfig(customConfigBean);
return config;
}
/**
* 加载配置信息
* @param generatorConfig
* @param customConfig
* @return
* @throws SQLException
* @throws IOException
*/
public static Config loadConfig(File generatorConfig, CustomConfiguration customConfig) throws SQLException, IOException {
GeneratorConfiguration generatorCoinfgBean = new GeneratorConfiguration();
readConfig(generatorConfig, generatorCoinfgBean);
Config config = new Config();
config.setGeneratorConfig(generatorCoinfgBean);
config.setCustomConfig(customConfig);
return config;
}
/**
* 保存自定义内容
*
* @param config
* @return
* @throws SQLException
* @throws IOException
*/
private static File saveCustomContent(Config config) throws SQLException, IOException {
String output = config.getCustomConfig().getCustomContent().getOutput();
String customContent = getCustomContent(config);
String generateFileName = generateFileName();
File file = new File(PROJECT_DIR + File.separator + output + File.separator + generateFileName, CUSTOM_CONTENT_FILE);
Files.createParentDirs(file);
Files.write(customContent, file, DEFAULT_CHARSET);
return file;
}
/**
* 合并配置文件
*
* @param config
* @return
* @throws SQLException
*/
private static GeneratorConfiguration mergeConfig(Config config) throws SQLException {
GeneratorConfiguration generatorConfig = config.getGeneratorConfig();
Context context = generatorConfig.getContext();
List tables = context.getTables();
if (X.isEmpty(tables)) {
tables = new ArrayList<>();
}
List customTables = getCustomTables(config);
if (X.notEmpty(customTables)) {
tables.addAll(customTables);
}
context.setTables(tables);
return generatorConfig;
}
/**
* 生成代码
*
* @param mbgConfig
* @param customContent
* @throws IOException
*/
private static void generateCode(GeneratorConfiguration mbgConfig, File customContent) throws IOException {
Context context = mbgConfig.getContext();
List tables = context.getTables();
// 备份自动生成的文件-----------------------------------
String parent = customContent.getParent();
// 文件名与文件的映射
Map fileNameWithFileMap = new LinkedHashMap<>();
// 自动生成的文件
List generatorFiles = findGeneratorFiles(mbgConfig);
// 备份文件
File backFile = null;
String fileName = null;
for (File file : generatorFiles) {
fileName = file.getName();
// 备份
if (fileName.endsWith(SQL_MAP_FILE_SUFFIX)) {
backFile = new File(parent + File.separator + SQL_MAP_FOLDER + File.separator + fileName);
} else if (fileName.endsWith(JAVA_CLIENT_FILE_SUFFIX)) {
backFile = new File(parent + File.separator + JAVA_CLIENT_FOLDER + File.separator + fileName);
} else {
backFile = new File(parent + File.separator + JAVA_MODEL_FOLDER + File.separator + fileName);
}
// 创建父目录，进行备份
Files.createParentDirs(backFile);
Files.write(Files.toString(file, DEFAULT_CHARSET), backFile, DEFAULT_CHARSET);
// 文件名与文件的映射，后面步骤会用到
fileNameWithFileMap.put(fileName, file);
}
// 每个table的对应的entity name
List entityNames = new ArrayList<>();
for (Table table : tables) {
entityNames.add(table.getDomainObjectName());
}
// 删除需要更新的文件,同时记录这些文件----------------------------
List updateFileNames = new ArrayList<>();
File updateFile = null;
Set fileNames = fileNameWithFileMap.keySet();
String entityFileName = null;
String mapperFileName = null;
String mapperXmlFileName = null;
for (String entityName : entityNames) {
entityFileName = entityName.concat(JAVA_MODEL_FILE_SUFFIX);
mapperFileName = entityName.concat(JAVA_CLIENT_FILE_SUFFIX);
mapperXmlFileName = entityName.concat(SQL_MAP_FILE_SUFFIX);
if (fileNames.contains(entityFileName)) {
updateFile = fileNameWithFileMap.get(entityFileName);
updateFileNames.add(updateFile.getAbsolutePath());
updateFile.delete();
}
if (fileNames.contains(mapperFileName)) {
updateFile = fileNameWithFileMap.get(mapperFileName);
updateFileNames.add(updateFile.getAbsolutePath());
updateFile.delete();
}
if (fileNames.contains(mapperXmlFileName)) {
updateFile = fileNameWithFileMap.get(mapperXmlFileName);
updateFileNames.add(updateFile.getAbsolutePath());
updateFile.delete();
}
}
// 生成文件 ----------------------------------
File mergedMbgConfigFile = new File(PROJECT_DIR, generateFileName() + ".xml");
// a.生成合并的 mbgConfig.xml
writeConfig(mergedMbgConfigFile, mbgConfig);
String mbgConfigContent = Files.toString(mergedMbgConfigFile, DEFAULT_CHARSET);
String mbgConfigDTD = getMBGConfigDTD();
Files.write(mbgConfigDTD.concat(mbgConfigContent), mergedMbgConfigFile, DEFAULT_CHARSET);
// b.执行maven命令调用MBG插件生成代码
File pomFile = new File(PROJECT_DIR, "pom.xml");
X.execCmd(Arrays.asList("mvn -f " + pomFile.getAbsolutePath() + " -Dmybatis.generator.configurationFile="
+ mergedMbgConfigFile.getAbsolutePath() + " mybatis-generator:generate"));
// c.删除合并的 mbgConfig.xml
mergedMbgConfigFile.delete();
// d.添加自定义内容
loadCustomContent(updateFileNames, customContent);
}
/**
* 获取自定义表
*
* @param config
* @return
* @throws SQLException
*/
private static List getCustomTables(Config config) throws SQLException {
GeneratorConfiguration generatorConfig = config.getGeneratorConfig();
CustomConfiguration customConfig = config.getCustomConfig();
// JDBC配置信息
JdbcConnection jdbc = generatorConfig.getContext().getJdbcConnection();
DbUtils.loadDriver(jdbc.getDriverClass());
String url = jdbc.getConnectionURL();
Connection conn = getConnection(url, jdbc.getUserId(), jdbc.getPassword());
// 从Connection地址截取数据库名称
String dbName = url.substring(url.lastIndexOf("/") + 1);
// 获取数据库所有表信息
List tables = DbInfos.getTables(conn, dbName);
// 自定义Table公共配置信息
CustomTableConfig tableConfig = customConfig.getTableConfig();
String propsInfo = tableConfig.getPropsInfo();
String domainObjectNameSuffix = tableConfig.getDomainObjectNameSuffix();
// 全部表
List allTableNames = new ArrayList<>();
for (TableMetadata tableMetadata : tables) {
allTableNames.add(tableMetadata.getTableName());
}
// 最终匹配的表
List machedTableNames = new ArrayList<>();
// 需要导入的表
List includes = tableConfig.getIncludes();
// 需要排除的表
List excludes = tableConfig.getExcludes();
// 优先级 include > exclude > database
if (X.notEmpty(includes)) {
machedTableNames.addAll(includes);
} else if (X.notEmpty(excludes)) {
for (String e : excludes) {
allTableNames.remove(e);
}
machedTableNames.addAll(allTableNames);
} else {
machedTableNames.addAll(allTableNames);
}
StringBuilder tableTags = new StringBuilder();
String className = null;
for (String tableName : machedTableNames) {
className = tableNameToClassName(tableName).concat(domainObjectNameSuffix);
tableTags.append(
"")
.append(System.lineSeparator());
}
Class cls = Context.class;
XStreamAlias alias = cls.getAnnotation(XStreamAlias.class);
String aliasValue = alias.value();
String tableXml = tableTags.toString();
tableXml = "<" + aliasValue + ">" + tableXml + "";
XStream xstream = new XStream();
xstream.autodetectAnnotations(true);
xstream.alias(aliasValue, cls);
Context context = (Context) xstream.fromXML(tableXml);
return context.getTables();
}
/**
* 获取自定义内容
*
* @param config
* @return
* @throws SQLException
* @throws IOException
*/
private static String getCustomContent(Config config) throws SQLException, IOException {
GeneratorConfiguration generatorConfig = config.getGeneratorConfig();
// 获取扫描目录 targetProject / targetPackage
List files = findGeneratorFiles(generatorConfig);
CustomConfiguration customConfig = config.getCustomConfig();
CustomContent customContent = customConfig.getCustomContent();
String startLimit = customContent.getStartLimit();
String endLimit = customContent.getEndLimit();
StringBuilder customCodeFileContent = new StringBuilder();
String customCodeContent = null;
for (File file : files) {
customCodeContent = findCustomCodeContent(file, startLimit, endLimit);
if (X.notEmpty(customCodeContent)) {
customCodeFileContent.append(CUSTOM_CONTENT_FILE_LINE_FLAG).append(file.getAbsolutePath())
.append(LINE_SEPARATOR).append(customCodeContent).append(LINE_SEPARATOR);
}
}
return customCodeFileContent.toString();
}
/**
* 获取自动生成的文件
*
* @param mbgConfig
* @return
*/
private static List findGeneratorFiles(GeneratorConfiguration mbgConfig) {
// 获取扫描目录 targetProject / targetPackage
Context context = mbgConfig.getContext();
JavaModelGenerator javaModelGenerator = context.getJavaModelGenerator();
String javaModelTargetProject = javaModelGenerator.getTargetProject();
String javaModelTargetPackage = javaModelGenerator.getTargetPackage();
File javaModelTargetPath = assembleTargetPath(javaModelTargetProject, javaModelTargetPackage);
SqlMapGenerator sqlMapGenerator = context.getSqlMapGenerator();
String sqlMapTargetProject = sqlMapGenerator.getTargetProject();
String sqlMapTargetPackage = sqlMapGenerator.getTargetPackage();
File sqlMapTargetPath = assembleTargetPath(sqlMapTargetProject, sqlMapTargetPackage);
JavaClientGenerator javaClientGenerator = context.getJavaClientGenerator();
String javaClientTargetProject = javaClientGenerator.getTargetProject();
String javaClientTargetPackage = javaClientGenerator.getTargetPackage();
File javaClientTargetPath = assembleTargetPath(javaClientTargetProject, javaClientTargetPackage);
File[] javaModelFiles = javaModelTargetPath.listFiles(new FilenameFilter() {
@Override
public boolean accept(File dir, String name) {
return name.endsWith(JAVA_MODEL_FILE_SUFFIX);
}
});
File[] sqlMapFiles = sqlMapTargetPath.listFiles(new FilenameFilter() {
@Override
public boolean accept(File dir, String name) {
return name.endsWith(SQL_MAP_FILE_SUFFIX);
}
});
File[] javaClientFiles = javaClientTargetPath.listFiles(new FilenameFilter() {
@Override
public boolean accept(File dir, String name) {
return name.endsWith(JAVA_CLIENT_FILE_SUFFIX);
}
});
List matchedFiles = new ArrayList<>();
if (X.notEmpty(javaModelFiles)) {
matchedFiles.addAll(Arrays.asList(javaModelFiles));
}
if (X.notEmpty(sqlMapFiles)) {
matchedFiles.addAll(Arrays.asList(sqlMapFiles));
}
if (X.notEmpty(javaClientFiles)) {
matchedFiles.addAll(Arrays.asList(javaClientFiles));
}
return matchedFiles;
}
/**
* 获取自定义内容内容
*
* @param file
* @return
* @throws IOException
*/
private static String findCustomCodeContent(File file, String startLimit, String endLimit) throws IOException {
List lines = Files.readLines(file, DEFAULT_CHARSET);
StringBuilder content = new StringBuilder();
boolean flag = false;
for (String line : lines) {
if (line.matches(startLimit)) {
flag = true;
}
if (flag) {
content.append(line).append(System.lineSeparator());
}
if (line.matches(endLimit)) {
flag = false;
}
}
return content.toString();
}
/**
* 加载自定义内容
*
* @param updateFileNames 需要更新的文件
* @param from 自定义内容内容文件
* @throws IOException
*/
private static void loadCustomContent(List updateFileNames, File from) throws IOException {
List lines = Files.readLines(from, DEFAULT_CHARSET);
if (lines.isEmpty()) {
// 没有自定义内容
System.out.println("没有自定义内容: " + from);
return;
}
File file = null;
StringBuilder customCodeContent = new StringBuilder();
for (String line : lines) {
// 文件路径行
if (line.matches("^" + CUSTOM_CONTENT_FILE_LINE_FLAG + ".*")) {
// 文件不为空，说明已经读取了一个文件的自定义内容
if (null != file) {
// 如果这个文件包含在这次更新的文件中，那么就把自定义内容添加到末尾
if (updateFileNames.contains(file.getName())) {
appendCustomContentTo(file, customCodeContent.toString());
}
// 清空自定义内容，重新接受下一个文件的自定义内容哦
customCodeContent.setLength(0);
}
file = new File(line.replace(CUSTOM_CONTENT_FILE_LINE_FLAG, ""));
continue;
}
customCodeContent.append(line).append(System.lineSeparator());
}
if (updateFileNames.contains(file.getName())) {
appendCustomContentTo(file, customCodeContent.toString());
}
}
/**
* 追加自定义内容到一个文件
*
* @param to 追加到这个文件
* @param customCodeContent 自定义内容内容
* @throws IOException
*/
private static void appendCustomContentTo(File to, String customCodeContent) throws IOException {
// 添加自定义内容后的内容
StringBuilder appendCustomCodeContent = new StringBuilder();
List lines = Files.readLines(to, DEFAULT_CHARSET);
String line = null;
int lineCount = lines.size();
for (int i = 0; i < lineCount; i++) {
line = lines.get(i);
// 文件的最后一行
if (i == (lineCount - 1)) {
appendCustomCodeContent.append(System.lineSeparator()).append(customCodeContent);
// 这里把文件末尾多余的空行移除
String content = appendCustomCodeContent.toString().replaceAll("(\r\n){1,}$", "");
appendCustomCodeContent.setLength(0);
appendCustomCodeContent.append(content).append(System.lineSeparator()).append(line);
break;
}
appendCustomCodeContent.append(line).append(System.lineSeparator());
}
Files.write(appendCustomCodeContent, to, DEFAULT_CHARSET);
}
/**
* 组装目标路径
*
* @param targetProject
* @param targetPackage
* @return
*/
private static File assembleTargetPath(String targetProject, String targetPackage) {
String targetPackagePath = targetPackage.replace(".", "/");
if ("MAVEN".equalsIgnoreCase(targetProject)) {
return new File(PROJECT_DIR, "src/main/java/" + targetPackagePath);
}
return new File(targetProject, targetPackagePath);
}
/**
* 数据库表名转换为Java类名
*
* @param tableName
* @return
*/
private static String tableNameToClassName(String tableName) {
return formatToHumpNamed(tableName, "_", true);
}
/**
* 格式化为驼峰命名
*
* @param str 需要格式化话的字符串
* @param separator 字符串分隔符
* @return
*/
public static String formatToHumpNamed(String str, String separator, boolean letterUpper) {
StringBuilder sb = new StringBuilder();
String[] items = str.split(separator);
for (int i = 0; i < items.length; i++) {
sb.append(firstLetterUpper(items[i].toLowerCase()));
}
String returnValue = sb.toString();
return letterUpper ? firstLetterUpper(returnValue) : firstLetterLower(returnValue);
}
/**
* 首字母大写
*
* @param str
* @return
*/
public static String firstLetterUpper(String str) {
return str.substring(0, 1).toUpperCase() + str.substring(1);
}
/**
* 首字母小写
*
* @param str
* @return
*/
public static String firstLetterLower(String str) {
return str.substring(0, 1).toLowerCase() + str.substring(1);
}
/**
* 生成一个文件名称
*
* @return
*/
private static String generateFileName() {
return X.dateFormat(new Date(), "yyyyMMddHHmmss");
}
/**
* 读取配置信息
*
* @param config
* @param obj
*/
private static void readConfig(File config, Object obj) {
XStream xstream = new XStream();
xstream.autodetectAnnotations(true);
xstream.alias("include", String.class);
xstream.alias("exclude", String.class);
Class objClass = obj.getClass();
XStreamAlias alias = objClass.getAnnotation(XStreamAlias.class);
String rootAlias = alias.value();
xstream.alias(rootAlias, objClass);
xstream.fromXML(config, obj);
}
/**
* 写入配置信息
*
* @param config
* @param obj
*/
private static void writeConfig(File config, Object obj) {
XStream xstream = new XStream();
xstream.autodetectAnnotations(true);
Class objClass = obj.getClass();
XStreamAlias alias = objClass.getAnnotation(XStreamAlias.class);
String rootAlias = alias.value();
xstream.alias(rootAlias, objClass);
FileWriter writer = null;
try {
writer = new FileWriter(config);
xstream.toXML(obj, writer);
} catch (Exception e) {
e.printStackTrace();
} finally {
if (X.notNull(writer)) {
try {
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 获取MybatisGeneratorConfig.xml的DTD定义
*
* @return
*/
public static String getMBGConfigDTD() {
StringBuilder dtd = new StringBuilder();
dtd.append("").append(System.lineSeparator())
.append("")
.append(System.lineSeparator());
return dtd.toString();
}
/**
* 获取数据库连接
*
* @param url
* @param user
* @param password
* @return
*/
private static Connection getConnection(String url, String user, String password) {
try {
return DriverManager.getConnection(url, user, password);
} catch (SQLException e) {
e.printStackTrace();
}
return null;
}
public static void main(String[] args) {
try {
// 启动
start();
} catch (Exception e) {
e.printStackTrace();
}
}
}
