cn.easyproject.easyee.auto.generator.SHCodeGenerator Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of easyee-auto Show documentation
Show all versions of easyee-auto Show documentation
Code Auto Generator for EasyEE Open Source JavaEE Enterprise Basic Development Platform(https://github.com/ushelp/EasyEE).
The newest version!
package cn.easyproject.easyee.auto.generator;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.NameFileFilter;
import org.apache.commons.io.filefilter.TrueFileFilter;
import cn.easyproject.easyee.auto.EasyCriteria;
import cn.easyproject.easyee.auto.EasyCriterias;
import cn.easyproject.easyee.auto.EasyAddDialog;
import cn.easyproject.easyee.auto.EasyField;
import cn.easyproject.easyee.auto.EasyAutoException;
import cn.easyproject.easyee.auto.EasyEditDialog;
import cn.easyproject.easyee.auto.EasyPage;
import cn.easyproject.easyee.auto.EasyId;
import cn.easyproject.easyee.auto.EasyModule;
import cn.easyproject.easyee.auto.PageType;
import cn.easyproject.easyee.auto.PackageUtil;
import freemarker.core.ParseException;
import freemarker.template.Configuration;
import freemarker.template.MalformedTemplateNameException;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import freemarker.template.TemplateExceptionHandler;
import freemarker.template.TemplateNotFoundException;
/**
* EasyAuto Code Generator for SH-SpringBoot
* @author Ray
* @author [email protected]
* @author easyproject.cn
* @since 1.0.0
*/
public class SHCodeGenerator {
private final String TEMPLATE_DIR="/templates/sh";
/**
* 新模块所在包
*/
private String modulePackage = null;
/**
* 新模块所在目录
*/
private String javaModuleFolder = null;
/**
* webapp/content 目录
*/
private String webappFolder=null;
/**
* PageBean 完全限定名
*/
private String pageBeanImport="cn.easyproject.easyee.sh.util.PageBean";
/**
* BaseService 完全限定名
*/
private String baseServiceImport="cn.easyproject.easyee.sh.service.BaseService";
/**
* EasyCriteria 完全限定名
*/
private String easyCriteriaImport="cn.easyproject.easyee.sh.util.EasyCriteria";
/**
* StringUtils 完全限定名
*/
private String stringUtilsImport="cn.easyproject.easyee.sh.util.StringUtils";
/**
* StatusCode 完全限定名
*/
private String statusCodeImport="cn.easyproject.easyee.sh.base.util.StatusCode";
/**
* BaseController 完全限定名
*/
private String baseControllerImport="cn.easyproject.easyee.sh.base.controller.BaseController";
//TODO
/**
* 生成文件
* @param allClass
* @param modules
* @throws TemplateException
* @throws IOException
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
private void generatorFile(List allClass, String modules) throws TemplateException, IOException {
for (Class nowClass : allClass) {
/* Create a data-model */
Map root = new HashMap();
// 包
String pkgName = modulePackage;
// 类
String className = nowClass.getSimpleName();
// 模块
String module=modulePackage;
if(module.contains(".")){
module= module.substring(module.lastIndexOf(".")+1);
}
// ========== EasyId
// 主键ID
String oid=null;
// ========== EasyId
// ========== Criteria
// 条件属性数据列表
Set propertys = new LinkedHashSet();
// 需要额外导入的包列表
Set imports = new LinkedHashSet();
// ========== Criteria
// ========== EasyField
// 类显示名
String clsLabel=className;
// 表格列显示信息
Set autos = new LinkedHashSet();
// ========== EasyField
// ========== EasyPage
// 页面数据编辑方式(dialog,row,cell)
PageType pageType=PageType.NONE;
EasyPage ee=(EasyPage) nowClass.getAnnotation(EasyPage.class);
if(ee!=null){
pageType=ee.value();
}
// ========== EasyPage
Field[] fields = nowClass.getDeclaredFields();
for (int i = 0; i < fields.length; i++) {
Field f = fields[i];
String fieldName=f.getName();
if (!fieldName.equals("serialVersionUID")) {
// ========== EasyId
// 主键id属性查询
EasyId e=f.getAnnotation(EasyId.class);
if(e!=null){
oid=fieldName;
}
// ========== EasyId
// ========== EasyField
EasyField ea=f.getAnnotation(EasyField.class);
if(ea!=null){
String label=fieldName;
String field=fieldName;
String inputField=null;
// boolean show =ea.show();
if(!ea.labelKey().equals("")){
label=" clsCriterias=new ArrayList();
// 类上的条件注解列表
if(ca!=null){
EasyCriteria[] cArray=ca.value();
for (EasyCriteria criteria : cArray) {
clsCriterias.add(criteria);
}
}
// 类上的条件注解
if(ca2!=null){
clsCriterias.add(ca2);
}
// 类上存在条件信息
if(clsCriterias.size()>0){
for (EasyCriteria c : clsCriterias) {
String type=c.type().getSimpleName();
String field=c.filed();
String queryConditionName=null;
String like="=";
String label=null;
if(field.equals("")){
throw new EasyAutoException("When use @Criteria on Class, you must defined 'name'");
}
if(!c.type().getName().equals("java.lang.Object")){
type=c.type().getSimpleName();
// 如果数据类型不是java.lang，则添加import自动导入
if (!c.type().getName().startsWith("java.lang.")) {
imports.add(c.type().getName());
}
}
if(!c.queryConditionName().equals("")){
queryConditionName=c.queryConditionName();
}else{
queryConditionName=field;
}
label=field;
if(!c.labelKey().equals("")){
label=" allClass = PackageUtil.getClasses(entityPackage);
if (entityPackage.contains(".")) {
modulePackage = entityPackage.substring(0, entityPackage.lastIndexOf("."));
}
String m = (entityPackage.substring(0, entityPackage.lastIndexOf("."))).replace('.', '/');
String basePath = System.getProperty("user.dir");
javaModuleFolder = basePath + "/src/main/java/" + m;
webappFolder= basePath+"/src/main/webapp/WEB-INF/content";
generatorAll(allClass, modules);
} catch (TemplateNotFoundException e) {
e.printStackTrace();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (MalformedTemplateNameException e) {
e.printStackTrace();
} catch (ParseException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} catch (TemplateException e) {
e.printStackTrace();
}
}
/**
* 为指定类生成模块代码
*
* @param clses
*/
@SuppressWarnings("rawtypes")
public void generator(Class... clses){
this.generator(null, clses);
}
/**
* 为指定类生成模块代码
*
* @param modules
* @param clses
*/
@SuppressWarnings("rawtypes")
public void generator(EasyAutoModule[] modules,Class... clses){
try {
for (Class cls : clses) {
String entityPackage = cls.getPackage().getName();
if (entityPackage.contains(".")) {
modulePackage = entityPackage.substring(0, entityPackage.lastIndexOf("."));
}
String m = (entityPackage.substring(0, entityPackage.lastIndexOf("."))).replace('.', '/');
String basePath = System.getProperty("user.dir");
javaModuleFolder = basePath + "/src/main/java/" + m;
webappFolder= basePath+"/src/main/webapp/WEB-INF/content";
List allClass = new ArrayList();
allClass.add(cls);
generatorAll(allClass, modules);
}
} catch (TemplateNotFoundException e) {
e.printStackTrace();
} catch (MalformedTemplateNameException e) {
e.printStackTrace();
} catch (ParseException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} catch (TemplateException e) {
e.printStackTrace();
}
}
/**
* 根据类名搜索类的完全限定名
* @return 完全限定名
*/
private String searchPageBeanImport(final String simpleClassName){
String path="";
String basePath = System.getProperty("user.dir")+"/src/main/java/";
Collection files= FileUtils.listFiles(new File(basePath), new NameFileFilter(simpleClassName+".java"), TrueFileFilter.INSTANCE );
if(files.size()>0){
path=files.iterator().next().getAbsolutePath().replace("\\", "/");
path=path.substring(path.indexOf("src/main/java/")+("src/main/java/").length());
path=path.replace("/", ".").substring(0, path.lastIndexOf("."));
}
return path;
}
/**
* 生成所有
* @param allClass
* @param modules
* @throws TemplateException
* @throws IOException
*/
@SuppressWarnings("rawtypes")
private void generatorAll(List allClass, EasyAutoModule[] modules) throws TemplateException, IOException {
pageBeanImport=searchPageBeanImport("PageBean");
baseServiceImport=searchPageBeanImport("BaseService");
easyCriteriaImport=searchPageBeanImport("EasyCriteria");
stringUtilsImport=searchPageBeanImport("StringUtils");
statusCodeImport=searchPageBeanImport("StatusCode");
baseControllerImport=searchPageBeanImport("BaseController");
// TODO
// controller
String mds="#";
if(modules==null){
mds="#controller#criteria#dao#mapper#page#service#serviceImpl#";
}else{
for (int i = 0; i < modules.length; i++) {
mds+=modules[i].value+"#";
}
}
generatorFile(allClass, mds);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy