package org.yeshi.utils.generater;
|
|
import org.springframework.data.mongodb.core.mapping.Document;
|
import org.yeshi.utils.FileUtil;
|
import org.yeshi.utils.StringUtil;
|
import org.yeshi.utils.generater.entity.ClassInfo;
|
import org.yeshi.utils.generater.entity.admin.AdminGeneraterInfo;
|
import org.yeshi.utils.generater.mybatis.Table;
|
import org.yeshi.utils.generater.util.NameUtil;
|
import org.yeshi.utils.generater.vo.ExceptionVO;
|
import org.yeshi.utils.generater.vo.admin.controller.AdminControllerInfoVO;
|
import org.yeshi.utils.generater.vo.dao.MongoDBDaoVO;
|
import org.yeshi.utils.generater.vo.dao.MyBatisDBDaoVO;
|
import org.yeshi.utils.generater.vo.service.QueryVO;
|
import org.yeshi.utils.generater.vo.service.ServiceImplInfoVO;
|
import org.yeshi.utils.generater.vo.service.ServiceInfoVO;
|
import org.yeshi.utils.generater.vo.xmlconfig.GenertorConfig;
|
import org.yeshi.utils.mongo.MongodbBaseDao;
|
import org.yeshi.utils.mybatis.BaseMapper;
|
|
import java.io.File;
|
import java.io.IOException;
|
import java.lang.annotation.Annotation;
|
|
/**
|
* @author Administrator
|
* @title: GeneraterManager
|
* @description: 自动化代码生成管理器(采用解析xml配置文件的形式)
|
* @date 2021/10/13 14:48
|
*/
|
|
/**
|
* Title
|
*
|
* @Description //TODO
|
* @Author hxh
|
* @Date 18:11 2021/10/13
|
**/
|
public class GeneraterManagerV2 {
|
|
private static GeneraterManagerV2 instance;
|
|
private String classRootDir;
|
private GenertorConfig config;
|
|
public static GeneraterManagerV2 getInstance() {
|
if (instance == null)
|
instance = new GeneraterManagerV2();
|
return instance;
|
}
|
|
|
public void init(String classRootDir, GenertorConfig config) {
|
this.classRootDir = classRootDir;
|
this.config = config;
|
}
|
|
//获取文件夹的路径
|
private String getDirPath(String packageName) {
|
File f = new File(classRootDir);
|
if (!f.exists()) {
|
f.mkdirs();
|
}
|
f = new File(f.getAbsolutePath(), packageName.replace(".", "/"));
|
if (!f.exists()) {
|
f.mkdirs();
|
}
|
return f.getAbsolutePath();
|
}
|
|
/**
|
* @return void
|
* @author hxh
|
* @description 保存文件
|
* @date 10:06 2021/10/14
|
* @param: data
|
* @param: path
|
**/
|
private void saveFile(String data, String path) throws IOException {
|
if (!new File(path).exists())
|
new File(path).createNewFile();
|
FileUtil.saveAsFileByte(data.getBytes("UTF-8"), path);
|
}
|
|
public void createException(Class base) throws Exception {
|
if (config.getService().getException() == null) {
|
throw new Exception("尚未配置异常信息");
|
}
|
|
String path = getDirPath(config.getService().getException().getPkg());
|
try {
|
ExceptionVO vo = new ExceptionVO.Builder().setEntity(config.getEntity()).setPackageName(config.getService().getException().getPkg()).setBase(new ClassInfo(base.getSimpleName(), base.getName())).build();
|
String result = GeneraterUtil.createException(vo);
|
//保存
|
saveFile(result, path);
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
|
public void createDao() throws Exception {
|
|
Annotation[] as = config.getEntity().getAnnotations();
|
for (Annotation a : as) {
|
if (a instanceof Table) {
|
createMapper();
|
break;
|
} else if (a instanceof Document) {
|
createMongoDBDao();
|
break;
|
}
|
}
|
}
|
|
/**
|
* @return org.yeshi.utils.generater.vo.dao.MongoDBDaoVO
|
* @author hxh
|
* @description 自动生成Dao
|
* @date 18:24 2021/10/13
|
* @param: packageName
|
**/
|
public MongoDBDaoVO createMongoDBDao() throws Exception {
|
if (config.getDao() == null) {
|
throw new Exception("尚未配置Dao信息");
|
}
|
|
String daoName = NameUtil.getDefaultDaoName(config.getEntity());
|
String path = getDirPath(config.getDao().getPkg());
|
MongoDBDaoVO vo = new MongoDBDaoVO.Builder()
|
.setBaseDao(new ClassInfo(MongodbBaseDao.class.getSimpleName(), MongodbBaseDao.class.getName()))
|
.setDao(new ClassInfo(daoName, config.getDao().getPkg() + "." + daoName))
|
.setEntity(config.getEntity()).setDaoData(config.getDao()).build();
|
String result = GeneraterUtil.createMongoDBDao(vo);
|
//保存
|
saveFile(result, new File(path, daoName + ".java").getAbsolutePath());
|
return vo;
|
}
|
|
/**
|
* @return org.yeshi.utils.generater.vo.dao.MyBatisDBDaoVO
|
* @author hxh
|
* @description 自动生成Mapper
|
* @date 18:24 2021/10/13
|
* @param: packageName
|
* @param: mapperDir
|
**/
|
public MyBatisDBDaoVO createMapper() throws Exception {
|
if (config.getDao() == null) {
|
throw new Exception("尚未配置Dao信息");
|
}
|
|
if (StringUtil.isNullOrEmpty(config.getDao().getMapperDir())) {
|
throw new Exception("Mapper存放目录为空");
|
}
|
|
String mapperName = NameUtil.getDefaultMapperName(config.getEntity());
|
String path = getDirPath(config.getDao().getPkg());
|
MyBatisDBDaoVO vo = new MyBatisDBDaoVO.Builder()
|
.setBaseMapper(new ClassInfo(BaseMapper.class.getSimpleName(), BaseMapper.class.getName()))
|
.setMapper(new ClassInfo(mapperName, config.getDao().getPkg() + "." + mapperName))
|
.setEntity(config.getEntity())
|
.setDaoData(config.getDao())
|
.build();
|
|
String result = GeneraterUtil.createMybatisDBMapper(vo);
|
//保存
|
saveFile(result, new File(path, mapperName + ".java").getAbsolutePath());
|
if (!new File(config.getDao().getMapperDir()).exists())
|
new File(config.getDao().getMapperDir()).mkdirs();
|
result = GeneraterUtil.createMybatisDBMapperXML(vo);
|
saveFile(result, new File(config.getDao().getMapperDir(), mapperName + ".xml").getAbsolutePath());
|
return vo;
|
}
|
|
/**
|
* @return org.yeshi.utils.generater.vo.service.QueryVO
|
* @author hxh
|
* @description /创建查询Query
|
* @date 18:24 2021/10/13
|
* @param: packageName
|
**/
|
public QueryVO createQuery(String packageName) throws Exception {
|
//创建Query
|
String path = getDirPath(packageName);
|
QueryVO queryVO = new QueryVO.Builder().setEntity(config.getEntity()).setPackageName(packageName).build();
|
String result = GeneraterUtil.createQuery(queryVO);
|
saveFile(result, new File(path, queryVO.getClassName() + ".java").getAbsolutePath());
|
return queryVO;
|
}
|
|
/**
|
* @return org.yeshi.utils.generater.vo.service.ServiceInfoVO
|
* @author hxh
|
* @description 自动生成服务
|
* @date 18:23 2021/10/13
|
* @param: interPackageName
|
* @param: implPackageName
|
* @param: query
|
* @param: dao
|
**/
|
public ServiceInfoVO createService(ClassInfo dao) throws Exception {
|
|
if (config.getService() == null) {
|
throw new Exception("尚未配置service");
|
}
|
|
if (config.getDao() == null && dao == null) {
|
throw new Exception("请在xml中配置Dao，或者传入Dao信息");
|
}
|
|
if (dao == null) {
|
if (StringUtil.isNullOrEmpty(config.getDao().getMapperDir())) {
|
String daoName = NameUtil.getDefaultDaoName(config.getEntity());
|
dao = new ClassInfo(daoName, config.getDao().getPkg() + "." + daoName);
|
} else {
|
String daoName = NameUtil.getDefaultMapperName(config.getEntity());
|
dao = new ClassInfo(daoName, config.getDao().getPkg() + "." + daoName);
|
}
|
}
|
|
QueryVO queryVO = createQuery(config.getService().getQueryPackage());
|
|
ClassInfo query = new ClassInfo(queryVO.getClassName(), queryVO.getPackageName() + "." + queryVO.getClassName());
|
|
//创建service接口
|
String serviceName = NameUtil.getDefaultServiceName(config.getEntity());
|
String path = getDirPath(config.getService().getInterPackage());
|
ServiceInfoVO vo = new ServiceInfoVO.Builder()
|
.setPackageName(config.getService().getInterPackage())
|
.setEntity(config.getEntity())
|
.setQuery(query)
|
.setService(new ClassInfo(serviceName, config.getService().getInterPackage() + "." + serviceName))
|
.build();
|
String result = GeneraterUtil.createService(vo);
|
saveFile(result, new File(path, serviceName + ".java").getAbsolutePath());
|
|
//创建service实现
|
String serviceImplName = NameUtil.getDefaultServiceImplName(config.getEntity());
|
path = getDirPath(config.getService().getImplPackage());
|
ServiceImplInfoVO implVO = new ServiceImplInfoVO.Builder()
|
.setDao(dao)
|
.setPackageName(config.getService().getImplPackage())
|
.setEntity(config.getEntity())
|
.setQuery(query)
|
.setService(new ClassInfo(serviceName, config.getService().getInterPackage() + "." + serviceName))
|
.setDaoQuery(new ClassInfo("DaoQuery", dao.getClazz() + ".DaoQuery"))
|
.build();
|
result = GeneraterUtil.createServiceImpl(implVO);
|
saveFile(result, new File(path, serviceImplName + ".java").getAbsolutePath());
|
return vo;
|
}
|
|
|
/**
|
* @return void
|
* @author hxh
|
* @description 自动生成管理模块
|
* @date 18:23 2021/10/13
|
* @param: controllerPackage
|
* @param: htmlDir
|
* @param: query
|
* @param: service
|
**/
|
public void createAdmin(ClassInfo query, ClassInfo service) throws Exception {
|
if (config.getAdmin() == null) {
|
throw new Exception("尚未配置admin");
|
}
|
|
if (service == null) {
|
if (config.getService() == null) {
|
throw new Exception("请传入service或者在xml中配置service");
|
}
|
String serviceName = NameUtil.getDefaultServiceName(config.getEntity());
|
service = new ClassInfo(serviceName, config.getService().getInterPackage() + "." + serviceName);
|
}
|
|
if (query == null) {
|
if (config.getService() == null) {
|
throw new Exception("请传入service或者在xml中配置service");
|
}
|
String queryName = NameUtil.getDefaultQueryName(config.getEntity());
|
query = new ClassInfo(queryName, config.getService().getQueryPackage() + "." + queryName);
|
}
|
|
|
String path = getDirPath(config.getAdmin().getController().getPkg());
|
|
AdminGeneraterInfo generaterInfo = new AdminGeneraterInfo.Builder().setEntityClass(config.getEntity()).setAdminData(config.getAdmin()).build();
|
|
|
AdminControllerInfoVO vo = new AdminControllerInfoVO.Builder().setAdminInfo(generaterInfo).setPackageName(generaterInfo.getControllerData().getPkg()).setQueryVO(query).setService(service).build();
|
String controllerName = vo.getControllerName();
|
//生成controller
|
String result = GeneraterUtil.createAdminController(vo);
|
saveFile(result, new File(path, controllerName + ".java").getAbsolutePath());
|
|
|
if (!new File(generaterInfo.getPageDir()).exists()) {
|
new File(generaterInfo.getPageDir()).mkdirs();
|
}
|
|
String htmlNamePrefix = getHtmlNameFromEntityName(config.getEntity().getSimpleName());
|
|
if (generaterInfo.getShowDataList() != null && generaterInfo.getShowDataList().size() > 0) {
|
result = GeneraterUtil.createAdminPageForList(generaterInfo);
|
saveFile(result, new File(generaterInfo.getPageDir(), generaterInfo.getListFileName() != null ? generaterInfo.getListFileName() : htmlNamePrefix + "_list.html").getAbsolutePath());
|
}
|
|
if (config.getAdmin().getPages().getAdd() != null && generaterInfo.getAddFormRows() != null && generaterInfo.getAddFormRows().size() > 0) {
|
result = GeneraterUtil.createAdminPageForAdd(generaterInfo);
|
saveFile(result, new File(generaterInfo.getPageDir(), generaterInfo.getAddFileName() != null ? generaterInfo.getAddFileName() : htmlNamePrefix + "_add.html").getAbsolutePath());
|
}
|
|
if (config.getAdmin().getPages().getUpdate() != null && generaterInfo.getUpdateFormRows() != null && generaterInfo.getUpdateFormRows().size() > 0) {
|
result = GeneraterUtil.createAdminPageForUpdate(generaterInfo);
|
saveFile(result, new File(generaterInfo.getPageDir(), generaterInfo.getUpdateFileName() != null ? generaterInfo.getUpdateFileName() : htmlNamePrefix + "_update.html").getAbsolutePath());
|
}
|
}
|
|
|
/**
|
* @return void
|
* @author hxh
|
* @description 创建整个功能模块
|
* @date 18:19 2021/10/13
|
* @param: daoParams
|
* @param: serviceParams
|
* @param: adminParams
|
**/
|
public void createWholeFunction() throws Exception {
|
Annotation[] as = config.getEntity().getAnnotations();
|
int type = -1;
|
for (Annotation a : as) {
|
if (a instanceof Table) {
|
type = 1;
|
break;
|
} else if (a instanceof Document) {
|
type = 2;
|
break;
|
}
|
}
|
if (type == -1) {
|
throw new Exception("实体未配置mybatis或者mongodb");
|
}
|
|
createDao();
|
createService(null);
|
createAdmin(null, null);
|
}
|
|
/**
|
* @return java.lang.String
|
* @author hxh
|
* @description 通过entity的名称获取html文件的前缀名称
|
* @date 18:25 2021/10/13
|
* @param: entityName
|
**/
|
private static String getHtmlNameFromEntityName(String entityName) {
|
StringBuffer sb = new StringBuffer();
|
StringBuffer buffer = null;
|
for (int i = 0; i < entityName.length(); i++) {
|
if (entityName.charAt(i) >= 65 && entityName.charAt(i) <= 90) {
|
//大写字母
|
if (buffer == null)
|
buffer = new StringBuffer();
|
buffer.append(entityName.charAt(i));
|
} else {
|
if (buffer != null) {
|
sb.append("_" + buffer.toString().toLowerCase());
|
buffer = null;
|
}
|
sb.append(entityName.charAt(i));
|
}
|
}
|
|
if (buffer != null) {
|
sb.append("_" + buffer.toString().toLowerCase());
|
}
|
|
if (sb.charAt(0) == '_')
|
sb = sb.replace(0, 1, "");
|
return sb.toString();
|
}
|
|
|
public static void main(String[] args) {
|
System.out.print(getHtmlNameFromEntityName("TTestEEntity"));
|
}
|
|
|
}
|
