package com.hx.auto;
|
|
import java.io.BufferedWriter;
|
import java.io.File;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.io.OutputStreamWriter;
|
import java.io.Writer;
|
import java.lang.reflect.Field;
|
import java.util.HashMap;
|
import java.util.zip.ZipOutputStream;
|
|
import com.gitee.sunchenbin.mybatis.actable.annotation.Column;
|
import com.gitee.sunchenbin.mybatis.actable.annotation.Table;
|
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlTypeConstant;
|
import com.hx.auto.common.JdbcType;
|
import com.hx.auto.common.ReadEntityData;
|
import com.hx.auto.common.UrlData;
|
import com.hx.auto.util.CommonTool;
|
|
import com.hx.auto.util.GeneratorClassParentUtil;
|
import freemarker.template.Configuration;
|
import freemarker.template.DefaultObjectWrapper;
|
import freemarker.template.Template;
|
|
/**
|
* 自动生成mapper.xml
|
*
|
* @author chenjiahe 2019年09月08日23:57:47
|
*
|
*/
|
public class GeneratorMapperUtil {
|
|
private static String templateDir = "/ftl";//获取模板路径
|
private static String templateName = "Mapper.tpl";//action模板名称
|
|
/**生成mapper.xml
|
* @param cl 实体类
|
* @param urlData 生成配置信息
|
* @throws Exception
|
*/
|
public static void generatorMapper(Class<?> cl,UrlData urlData) throws Exception {
|
try {
|
// 反射start
|
// 类名
|
String entityName = cl.getSimpleName();
|
// 类名首字母小写
|
String initial = entityName.substring(0, 1).toLowerCase();
|
String entityNameSmall = initial + entityName.substring(1, entityName.length());
|
|
//获取实体类包名
|
String[] strs = cl.getName().split("\\.");
|
String packageName = "";
|
//去掉类名
|
for(int i=0;i<strs.length-1;i++) {
|
packageName += "."+strs[i];
|
}
|
packageName = packageName.replaceFirst(".", "");
|
|
//获取模板
|
Configuration configuration = new Configuration(Configuration.getVersion());
|
configuration.setClassForTemplateLoading(GeneratorMapperUtil.class, templateDir);
|
//configuration.setTemplateLoader(new ClassTemplateLoader(this.getClass(), "填你的resource下的路径，比如/ftl"));
|
Template template = configuration.getTemplate(templateName);
|
|
//实体类所有的字段
|
//用来存储数据
|
ReadEntityData readEntityData = new ReadEntityData();
|
//表名称
|
Table table = cl.getAnnotation(Table.class);
|
readEntityData.setTableName(table.name());
|
// 取得本类的全部属性
|
Field[] fields = cl.getDeclaredFields();
|
fields = GeneratorClassParentUtil.getPatentFields(fields,cl);
|
for (Field field:fields) {
|
// 判断方法中是否有指定注解类型的注解
|
boolean hasAnnotation = field.isAnnotationPresent(Column.class);
|
if (hasAnnotation) {
|
// 根据注解类型返回方法的指定类型注解
|
Column column = field.getAnnotation(Column.class);
|
|
//判断是不是
|
boolean isBol = false;
|
Integer isBlob = 0;
|
if(column.type().equals(MySqlTypeConstant.TEXT)||column.type().equals(MySqlTypeConstant.LONGTEXT)
|
||column.type().equals(MySqlTypeConstant.LONGBLOB)){
|
isBol = true;
|
isBlob = 1;
|
}
|
|
//类型
|
//String type = column.type().toUpperCase();
|
String type = JdbcType.jdbcTypeData(field.getType().getTypeName().toUpperCase(),isBol);
|
//主键
|
if (column.isKey()) {
|
readEntityData.setEntityIdName(field.getName());
|
readEntityData.setEntityIdData("#{"+field.getName()+"}");
|
if(!CommonTool.checkNotNull(column.name())) {
|
readEntityData.setTableIdName(field.getName());
|
}else {
|
readEntityData.setTableIdName(column.name());
|
}
|
readEntityData.setKeyType(type);
|
continue;
|
}
|
//存储数据
|
if(!CommonTool.checkNotNull(column.name())) {
|
readEntityData.fielData(field.getName(),field.getName(),type,isBlob,"#{"+field.getName()+"}");
|
}else {
|
readEntityData.fielData(field.getName(),column.name(),type,isBlob,"#{"+field.getName()+"}");
|
}
|
}
|
}
|
// 创建数据模型
|
HashMap<String, Object> root = new HashMap<String, Object>();
|
|
//action包名
|
if(!CommonTool.checkNotNull(urlData.getActionUrl())) {
|
System.err.println("没有生成action路径");
|
return;
|
}
|
if(!CommonTool.checkNotNull(urlData.getTotalUrl())) {
|
System.err.println("没有设置总包路径");
|
return;
|
}
|
root.put("TotalPackageName",urlData.getTotalUrl());
|
root.put("packageActionName",urlData.getActionUrl()[1]);
|
root.put("packageEntityName",packageName);
|
//实体类的类名
|
root.put("entityName", entityName);
|
//实体类的类名（首字母小写）
|
root.put("entityNameSmall", entityNameSmall);
|
//表字段 数据
|
root.put("fieldData",readEntityData);
|
|
root.put("sqlSentence","${sqlSentence}");
|
//root.put("sqlSentence","${sqlSentence}");
|
//root.put("sqlSentence","${sqlSentence}");
|
|
//dao的包名
|
if(!CommonTool.checkNotNull(urlData.getDaoUrl())) {
|
System.err.println("没有dao路径");
|
return;
|
}
|
root.put("DAOPackageName",urlData.getDaoUrl()[1]);
|
//实体类的包名
|
root.put("entityPackageName",packageName);
|
|
//生成文件路径
|
String targetFile = urlData.getMapperUrl()[1].replace(".", "/")+"/";
|
//生成文件名称
|
targetFile += entityName + "Mapper.xml";
|
//补全路径
|
targetFile = "./"+urlData.getMapperUrl()[0].replace(".", "/")+"/"+targetFile;
|
//获取是否已经生成过文件，拿取自定义的内容
|
String xmlStr = GeneratorReadXmlUtil.readMapperXml(targetFile);
|
if(!"false".equals(xmlStr)){
|
root.put("customData",GeneratorReadXmlUtil.readMapperXml(targetFile));
|
}else{
|
root.put("customData","");
|
}
|
|
System.out.println("mapperUrl:"+targetFile);
|
// 将模板和数据模型合并 输出到Console
|
Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(targetFile), "UTF-8"));
|
template.process(root, out);
|
|
|
|
out.flush();
|
out.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
