package cn.itcast.invoice.util.generator;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import cn.itcast.invoice.invoice.storeoper.vo.StoreOperModel;
//生成器
public class GeneratorUtil {
private Class clazz;
private String all; // EmpModel
private String small; // emp
private String big; // Emp
private String little; // e
private String pkg; // cn.itcast.invoice.auth
private String dir; // cn\itcast\invoice\auth
public static void main(String[] args) throws Exception {
// 只要你提供Model我就可以给你生成你要的所有文件
//EmpModel,RoleModel,ResModel,MenuModel
//SupplierModel,GoodsTypeModel,GoodsModel
//OrderModel,OrderDetailModel
//StoreModel,StoreDetailModel,StoreOperModel
new GeneratorUtil(StoreOperModel.class);
System.out.println("请添加Action在struts.xml中的定义");
System.out.println("请添加关联关系,位置hbm.xml");
System.out.println("请添加自定义查询条件数据层实现,位置Impl");
System.out.println("请添加自定义查询条件,位置QueryModel");
System.out.println("运行完毕!");
}
public GeneratorUtil(Class clazz) throws Exception {
this.clazz = clazz;
// -1.数据初始化
dataInit();
// 0.生成目录(包)
generatorDirectory();
// 1.QueryModel
generatorQueryModel();
// 2.hbm.xml
generatorHbmXml();
// 3.Dao
generatorDao();
// 4.Impl
generatorImpl();
// 5.Ebi
generatorEbi();
// 6.Ebo
generatorEbo();
// 7.Action
generatorAction();
// 8.applicationContext-**.xml
generatorApplicationContextXml();
}
// 8.applicationContext-**.xml
private void generatorApplicationContextXml() throws Exception {
File f = new File("resources/applicationContext-"+small+".xml");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
bw.newLine();
bw.write("<beans xmlns=\"http://www.springframework.org/schema/beans\"");
bw.newLine();
bw.write(" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"");
bw.newLine();
bw.write(" xsi:schemaLocation=\"");
bw.newLine();
bw.write(" http://www.springframework.org/schema/beans ");
bw.newLine();
bw.write(" http://www.springframework.org/schema/beans/spring-beans.xsd");
bw.newLine();
bw.write(" \"> ");
bw.newLine();
bw.write(" <!-- Action -->");
bw.newLine();
bw.write(" <bean id=\""+small+"Action\" class=\""+pkg+".web."+big+"Action\" scope=\"prototype\">");
bw.newLine();
bw.write(" <property name=\""+small+"Ebi\" ref=\""+small+"Ebi\"/>");
bw.newLine();
bw.write(" </bean>");
bw.newLine();
bw.newLine();
bw.write(" <!-- Ebi -->");
bw.newLine();
bw.write(" <bean id=\""+small+"Ebi\" class=\""+pkg+".business.ebo."+big+"Ebo\">");
bw.newLine();
bw.write(" <property name=\""+small+"Dao\" ref=\""+small+"Dao\"/>");
bw.newLine();
bw.write(" </bean>");
bw.newLine();
bw.newLine();
bw.write(" <!-- Dao -->");
bw.newLine();
bw.write(" <bean id=\""+small+"Dao\" class=\""+pkg+".dao.impl."+big+"Impl\">");
bw.newLine();
bw.write(" <property name=\"sessionFactory\" ref=\"sessionFactory\"/>");
bw.newLine();
bw.write(" </bean>");
bw.newLine();
bw.write("</beans>");
bw.newLine();
bw.flush();
bw.close();
}
// 7.Action
private void generatorAction() throws Exception {
File f = new File("src/"+dir+"/web/"+big+"Action.java");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".web;");
bw.newLine();
bw.newLine();
bw.write("import java.util.List;");
bw.newLine();
bw.newLine();
bw.write("import "+pkg+".business.ebi."+big+"Ebi;");
bw.newLine();
bw.write("import "+pkg+".vo."+big+"Model;");
bw.newLine();
bw.write("import "+pkg+".vo."+big+"QueryModel;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseAction;");
bw.newLine();
bw.newLine();
bw.write("public class "+big+"Action extends BaseAction{");
bw.newLine();
bw.write(" public "+big+"Model "+little+"m = new "+big+"Model();");
bw.newLine();
bw.write(" public "+big+"QueryModel "+little+"qm = new "+big+"QueryModel();");
bw.newLine();
bw.newLine();
bw.write(" private "+big+"Ebi "+small+"Ebi;");
bw.newLine();
bw.write(" public void set"+big+"Ebi("+big+"Ebi "+small+"Ebi) {");
bw.newLine();
bw.write(" this."+small+"Ebi = "+small+"Ebi;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public String list(){");
bw.newLine();
bw.write(" setDataTotal("+small+"Ebi.getCount("+little+"qm));");
bw.newLine();
bw.write(" List<"+big+"Model> "+small+"List = "+small+"Ebi.getAll("+little+"qm,pageNum,pageCount);");
bw.newLine();
bw.write(" put(\""+small+"List\", "+small+"List);");
bw.newLine();
bw.write(" return LIST;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public String input(){");
bw.newLine();
bw.write(" if("+little+"m.getUuid() != null){");
bw.newLine();
bw.write(" "+little+"m = "+small+"Ebi.get("+little+"m.getUuid());");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.write(" return INPUT;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public String save(){");
bw.newLine();
bw.write(" if("+little+"m.getUuid() == null){");
bw.newLine();
bw.write(" "+small+"Ebi.save("+little+"m);");
bw.newLine();
bw.write(" }else{");
bw.newLine();
bw.write(" "+small+"Ebi.update("+little+"m);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.write(" return TO_LIST;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public String delete(){");
bw.newLine();
bw.write(" "+small+"Ebi.delete("+little+"m);");
bw.newLine();
bw.write(" return TO_LIST;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 6.Ebo
private void generatorEbo() throws Exception {
File f = new File("src/"+dir+"/business/ebo/"+big+"Ebo.java");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".business.ebo;");
bw.newLine();
bw.newLine();
bw.write("import java.io.Serializable;");
bw.newLine();
bw.write("import java.util.List;");
bw.newLine();
bw.newLine();
bw.write("import "+pkg+".business.ebi."+big+"Ebi;");
bw.newLine();
bw.write("import "+pkg+".dao.dao."+big+"Dao;");
bw.newLine();
bw.write("import "+pkg+".vo."+big+"Model;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseQueryModel;");
bw.newLine();
bw.newLine();
bw.write("public class "+big+"Ebo implements "+big+"Ebi{");
bw.newLine();
bw.write(" private "+big+"Dao "+small+"Dao;");
bw.newLine();
bw.write(" public void set"+big+"Dao("+big+"Dao "+small+"Dao) {");
bw.newLine();
bw.write(" this."+small+"Dao = "+small+"Dao;");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public void save("+big+"Model "+little+"m) {");
bw.newLine();
bw.write(" "+small+"Dao.save("+little+"m);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public void update("+big+"Model "+little+"m) {");
bw.newLine();
bw.write(" "+small+"Dao.update("+little+"m);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public void delete("+big+"Model "+little+"m) {");
bw.newLine();
bw.write(" "+small+"Dao.delete("+little+"m);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public "+big+"Model get(Serializable uuid) {");
bw.newLine();
bw.write(" return "+small+"Dao.get(uuid);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public List<"+big+"Model> getAll() {");
bw.newLine();
bw.write(" return "+small+"Dao.getAll();");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public Integer getCount(BaseQueryModel qm) {");
bw.newLine();
bw.write(" return "+small+"Dao.getCount(qm);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write(" public List<"+big+"Model> getAll(BaseQueryModel qm, Integer pageNum,Integer pageCount) {");
bw.newLine();
bw.write(" return "+small+"Dao.getAll(qm,pageNum,pageCount);");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 5.Ebi
private void generatorEbi() throws Exception {
File f = new File("src/"+dir+"/business/ebi/"+big+"Ebi.java");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".business.ebi;");
bw.newLine();
bw.newLine();
bw.write("import org.springframework.transaction.annotation.Transactional;");
bw.newLine();
bw.newLine();
bw.write("import "+pkg+".vo."+big+"Model;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseEbi;");
bw.newLine();
bw.write("@Transactional");
bw.newLine();
bw.write("public interface "+big+"Ebi extends BaseEbi<"+big+"Model>{");
bw.newLine();
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 4.Impl
private void generatorImpl() throws Exception {
File f = new File("src/"+dir+"/dao/impl/"+big+"Impl.java");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".dao.impl;");
bw.newLine();
bw.newLine();
bw.write("import org.hibernate.criterion.DetachedCriteria;");
bw.newLine();
bw.write("import org.hibernate.criterion.Restrictions;");
bw.newLine();
bw.newLine();
bw.write("import "+pkg+".dao.dao."+big+"Dao;");
bw.newLine();
bw.write("import "+pkg+".vo."+big+"Model;");
bw.newLine();
bw.write("import "+pkg+".vo."+big+"QueryModel;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseDaoImpl;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseQueryModel; ");
bw.newLine();
bw.newLine();
bw.write("public class "+big+"Impl extends BaseDaoImpl<"+big+"Model> implements "+big+"Dao{");
bw.newLine();
bw.newLine();
bw.write(" public void doQbc(BaseQueryModel qm,DetachedCriteria dc){");
bw.newLine();
bw.write(" "+big+"QueryModel "+little+"qm = ("+big+"QueryModel)qm;");
bw.newLine();
bw.write(" //TODO 添加自定义查询条件");
bw.newLine();
bw.write(" }");
bw.newLine();
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 3.Dao
private void generatorDao() throws Exception {
File f = new File("src/"+dir+"/dao/dao/"+big+"Dao.java");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".dao.dao;");
bw.newLine();
bw.newLine();
bw.write("import "+pkg+".vo."+big+"Model;");
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseDao;");
bw.newLine();
bw.newLine();
bw.write("public interface "+big+"Dao extends BaseDao<"+big+"Model>{");
bw.newLine();
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 2.hbm.xml
private void generatorHbmXml() throws Exception {
File f = new File("src/"+dir+"/vo/"+big+"Model.hbm.xml");
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
bw.newLine();
bw.write("<!DOCTYPE hibernate-mapping PUBLIC");
bw.newLine();
bw.write(" '-//Hibernate/Hibernate Mapping DTD 3.0//EN'");
bw.newLine();
bw.write(" 'http://www.hibernate.org/dtd/hibernate-mapping-3.0.dtd'>");
bw.newLine();
bw.write("<hibernate-mapping>");
bw.newLine();
bw.write(" <class name=\""+pkg+".vo."+big+"Model\" table=\"tbl_"+small+"\">");
bw.newLine();
bw.write(" <id name=\"uuid\">");
bw.newLine();
bw.write(" <generator class=\"native\" />");
bw.newLine();
bw.write(" </id>");
bw.newLine();
//迭代生成所有的属性
//使用反射获取所有的字段
Field[] fds = clazz.getDeclaredFields();
for (Field fd :fds) {
//如果是私有的生成
if(Modifier.PRIVATE == fd.getModifiers() && !"uuid".equals(fd.getName())){
//String,Long,Double,Integer
if( fd.getType().equals(String.class) ||
fd.getType().equals(Long.class) ||
fd.getType().equals(Integer.class) ||
fd.getType().equals(Double.class)
){
if(!fd.getName().endsWith("View")){
bw.write(" <property name=\""+fd.getName()+"\"/>");
bw.newLine();
}
}
}
}
bw.newLine();
bw.write(" <!-- 关系 -->");
bw.newLine();
bw.newLine();
bw.write(" </class>");
bw.newLine();
bw.write("</hibernate-mapping>");
bw.newLine();
bw.flush();
bw.close();
}
// 1.QueryModel
private void generatorQueryModel() throws Exception {
//1.创建文件
//2.IO写数据
//3.关闭流
File f = new File("src/"+dir+"/vo/"+big+"QueryModel.java");
//如果该文件存在,则不做任何操作
if(f.exists()){
return;
}
f.createNewFile();
BufferedWriter bw = new BufferedWriter(new FileWriter(f));
bw.write("package "+pkg+".vo;");
bw.newLine();
bw.newLine();
bw.write("import cn.itcast.invoice.util.base.BaseQueryModel;");
bw.newLine();
bw.newLine();
bw.write("public class "+big+"QueryModel extends "+big+"Model implements BaseQueryModel{");
bw.newLine();
bw.write(" //TODO 添加查询范围字段 ");
bw.newLine();
bw.write("}");
bw.newLine();
bw.flush();
bw.close();
}
// 0.生成目录(包)
private void generatorDirectory() {
File f = new File("src/"+dir+"/web");
f.mkdirs();
f = new File("src/"+dir+"/business/ebi");
f.mkdirs();
f = new File("src/"+dir+"/business/ebo");
f.mkdirs();
f = new File("src/"+dir+"/dao/dao");
f.mkdirs();
f = new File("src/"+dir+"/dao/impl");
f.mkdirs();
}
// -1.数据初始化
private void dataInit() {
// EmpModel
all = clazz.getSimpleName();
// Emp
big = all.substring(0,all.length()-5);
// e
little = all.substring(0,1).toLowerCase();
// emp
small = little + big.substring(1);
// cn.itcast.invoice.auth.emp
String pkgTemp = clazz.getPackage().getName();
pkg = pkgTemp.substring(0,pkgTemp.length()-3);
// cn\itcast\invoice\auth\emp
dir = pkg.replace(".", "/");
}
}