【struts2+hibernate+spring项目实战】java读写实现代码生成器（ssh）

一、概述

开发阶段80%以上的模块都具有基本增删改查功能，反反复复的制作不仅效率低下，而且容易出错，使用代码生成器自动生成代码，只需要精品的开发出代码生成器，后期代码开发效率将直线上升。
代码生成器的思想就是通过执行程序，自动生成出Java源代码，核心原理是读写文件。

二、具体实现

2.1、数据初始化

这个方法主要是获取包名、实体类名称等信息。

//-1.数据初始化
private void dataInit() {
        String className = clazz.getSimpleName();                   //UserModel
        b = className.substring(0, className.length()-5);   //User
        String first = b.substring(0,1);                            //E
        l = first.toLowerCase();                        //e
        s = l+b.substring(1);                       //user
        String rootPkg = clazz.getPackage().getName();              //org.sihai.qualitycontrol.auth.user.vo
        pkg = rootPkg.substring(0,rootPkg.length()-3);      //org.sihai.qualitycontrol.auth.user
        dir = pkg.replace(".","/");                         //org/sihai/qualitycontrol/auth/user
    }

2.2、创建目录

生成具体的代码的所在目录

//0.创建目录
private void generatorDirectory() {
//business/ebi
// src+//org.sihai.qualitycontrol.auth.user+business/ebi .vo
        File f = new File("src/"+dir+"/business/ebi");
        f.mkdirs();
//business/ebo
        f = new File("src/"+dir+"/business/ebo");
        f.mkdirs();
//dao/dao
        f = new File("src/"+dir+"/dao/dao");
        f.mkdirs();
//dao/impl
        f = new File("src/"+dir+"/dao/impl");
        f.mkdirs();
//web
        f = new File("src/"+dir+"/web");
        f.mkdirs();
    }

2.3、创建查询实体类

//1.QueryModel
private void generatorQueryModel() throws Exception {
//1.创建文件
        File f = new File("src/"+dir+"/vo/"+b+"QueryModel.java");

//判断：如果该文件存在，终止操作
if(f.exists()){
return;
        }

        f.createNewFile();
//2.IO写入内容
        BufferedWriter bw = new BufferedWriter(new FileWriter(f));

bw.write("package "+pkg+".vo;");
bw.newLine();

bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseQueryModel;");
bw.newLine();

bw.newLine();

bw.write("public class "+b+"QueryModel extends "+b+"Model implements BaseQueryModel{");
bw.newLine();

bw.write(" // TODO 添加自定义查询条件");
bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close();
    }

2.4、创建实体类的hibernate的配置文件

//2.Hbm.xml
    private void generatorHbmXml() throws Exception {
        //1.创建文件
        File f = new File("src/"+dir+"/vo/"+b+"Model.hbm.xml");

        if(f.exists()){
            return;
        }

        f.createNewFile();
        //2.IO写入内容
        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."+b+"Model\" table=\"tbl_"+s+"\">");
        bw.newLine();

        bw.write(" <id name=\"uuid\">");
        bw.newLine();

        bw.write(" <generator class=\"native\" />");
        bw.newLine();

        bw.write(" </id>");
        bw.newLine();

        //hibernate的映射配置文件中要对原始模型类中的属性进行配置，反射获取所有字段
        Field[] fds = clazz.getDeclaredFields();
        for(Field fd:fds) {
            //如果字段的修饰符是private，生成
            if(fd.getModifiers() == Modifier.PRIVATE && !fd.getName().equals("uuid")){
                //如果是关联关系不生成,不是关联关系(Long,Integer,Double,String)
                if( fd.getType().equals(String.class)||
                    fd.getType().equals(Long.class)||
                    fd.getType().equals(Integer.class)||
                    fd.getType().equals(Double.class)
                    ){
                    bw.write(" <property name=\""+fd.getName()+"\"/>");
                    bw.newLine();
                }
            }
        }

        bw.write(" </class>");
        bw.newLine();

        bw.write("</hibernate-mapping>");
        bw.newLine();

        bw.flush();
        bw.close(); 
    }

2.5、创建dao接口

    //3.Dao
private void generatorDao() throws Exception {
        File f = new File("src/"+dir+"/dao/dao/"+b+"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."+b+"Model;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseDao;");
bw.newLine();

bw.newLine();

bw.write("public interface "+b+"Dao extends BaseDao<"+b+"Model> {");
bw.newLine();

bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close();     
    }

2.6、创建dao实现类

//4.Impl
private void generatorImpl()  throws Exception {
        File f = new File("src/"+dir+"/dao/impl/"+b+"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."+b+"Dao;");
bw.newLine();

bw.write("import "+pkg+".vo."+b+"Model;");
bw.newLine();

bw.write("import "+pkg+".vo."+b+"QueryModel;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseImpl;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseQueryModel;");
bw.newLine();

bw.newLine();

bw.write("public class "+b+"Impl extends BaseImpl<"+b+"Model> implements "+b+"Dao{");
bw.newLine();

bw.newLine();

bw.write(" public void doQbc(DetachedCriteria dc,BaseQueryModel qm){");
bw.newLine();

bw.write(" "+b+"QueryModel "+l+"qm = ("+b+"QueryModel)qm;");
bw.newLine();

bw.write(" // TODO 添加自定义查询条件");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close();     
    }

2.7、创建service的接口

//5.Ebi
private void generatorEbi()  throws Exception {
        File f = new File("src/"+dir+"/business/ebi/"+b+"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."+b+"Model;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseEbi;");
bw.newLine();

bw.newLine();

bw.write("@Transactional");
bw.newLine();

bw.write("public interface "+b+"Ebi extends BaseEbi<"+b+"Model>{");
bw.newLine();

bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close(); 
    }

2.8、创建service的实现类

//6.Ebo
private void generatorEbo()  throws Exception {
        File f = new File("src/"+dir+"/business/ebo/"+b+"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."+b+"Ebi;");
bw.newLine();

bw.write("import "+pkg+".dao.dao."+b+"Dao;");
bw.newLine();

bw.write("import "+pkg+".vo."+b+"Model;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseQueryModel;");
bw.newLine();

bw.newLine();

bw.write("public class "+b+"Ebo implements "+b+"Ebi{");
bw.newLine();

bw.write(" private "+b+"Dao "+s+"Dao;");
bw.newLine();

bw.write(" public void set"+b+"Dao("+b+"Dao "+s+"Dao) {");
bw.newLine();

bw.write(" this."+s+"Dao = "+s+"Dao;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public void save("+b+"Model "+l+"m) {");
bw.newLine();

bw.write(" "+s+"Dao.save("+l+"m);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public void update("+b+"Model "+l+"m) {");
bw.newLine();

bw.write(" "+s+"Dao.update("+l+"m);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public void delete("+b+"Model "+l+"m) {");
bw.newLine();

bw.write(" "+s+"Dao.delete("+l+"m);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public "+b+"Model get(Serializable uuid) {");
bw.newLine();

bw.write(" return "+s+"Dao.get(uuid);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public List<"+b+"Model> getAll() {");
bw.newLine();

bw.write(" return "+s+"Dao.getAll();");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public List<"+b+"Model> getAll(BaseQueryModel qm, Integer pageNum,Integer pageCount) {");
bw.newLine();

bw.write(" return "+s+"Dao.getAll(qm,pageNum,pageCount);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" public Integer getCount(BaseQueryModel qm) {");
bw.newLine();

bw.write(" return "+s+"Dao.getCount(qm);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close();         
    }

2.9、创建action类

//7.Action
private void generatorAction() throws Exception {
        File f = new File("src/"+dir+"/web/"+b+"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."+b+"Ebi;");
bw.newLine();

bw.write("import "+pkg+".vo."+b+"Model;");
bw.newLine();

bw.write("import "+pkg+".vo."+b+"QueryModel;");
bw.newLine();

bw.write("import org.sihai.qualitycontrol.util.base.BaseAction;");
bw.newLine();

bw.newLine();

bw.write("public class "+b+"Action extends BaseAction{");
bw.newLine();

bw.write(" public "+b+"Model "+l+"m = new "+b+"Model();");
bw.newLine();

bw.write(" public "+b+"QueryModel "+l+"qm = new "+b+"QueryModel();");
bw.newLine();

bw.newLine();

bw.write(" private "+b+"Ebi "+s+"Ebi;");
bw.newLine();

bw.write(" public void set"+b+"Ebi("+b+"Ebi "+s+"Ebi) {");
bw.newLine();

bw.write(" this."+s+"Ebi = "+s+"Ebi;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" //列表");
bw.newLine();

bw.write(" public String list(){");
bw.newLine();

bw.write(" setDataTotal("+s+"Ebi.getCount("+l+"qm));");
bw.newLine();

bw.write(" List<"+b+"Model> "+s+"List = "+s+"Ebi.getAll("+l+"qm,pageNum,pageCount);");
bw.newLine();

bw.write(" put(\""+s+"List\", "+s+"List);");
bw.newLine();

bw.write(" return LIST;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" //到添加");
bw.newLine();

bw.write(" public String input(){");
bw.newLine();

bw.write(" if("+l+"m.getUuid()!=null){");
bw.newLine();

bw.write(" "+l+"m = "+s+"Ebi.get("+l+"m.getUuid());");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.write(" return INPUT;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" //添加");
bw.newLine();

bw.write(" public String save(){");
bw.newLine();

bw.write(" if("+l+"m.getUuid() == null){");
bw.newLine();

bw.write(" "+s+"Ebi.save("+l+"m);");
bw.newLine();

bw.write(" }else{");
bw.newLine();

bw.write(" "+s+"Ebi.update("+l+"m);");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.write(" return TO_LIST;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write(" //删除");
bw.newLine();

bw.write(" public String delete(){");
bw.newLine();

bw.write(" "+s+"Ebi.delete("+l+"m);");
bw.newLine();

bw.write(" return TO_LIST;");
bw.newLine();

bw.write(" }");
bw.newLine();

bw.newLine();

bw.write("}");
bw.newLine();

bw.flush();
bw.close();         
    }

2.10、创建spring的配置文件

//8.applicationContext.xml
private void generatorApplicationContextXml() throws Exception {
        File f = new File("resources/applicationContext-"+s+".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=\""+s+"Action\" class=\""+pkg+".web."+b+"Action\" scope=\"prototype\">");
bw.newLine();

bw.write(" <property name=\""+s+"Ebi\" ref=\""+s+"Ebi\"/>");
bw.newLine();

bw.write(" </bean>");
bw.newLine();

bw.write(" <!-- Ebi -->");
bw.newLine();

bw.write(" <bean id=\""+s+"Ebi\" class=\""+pkg+".business.ebo."+b+"Ebo\">");
bw.newLine();

bw.write(" <property name=\""+s+"Dao\" ref=\""+s+"Dao\"/>");
bw.newLine();

bw.write(" </bean>");
bw.newLine();

bw.write(" <!-- Dao -->");
bw.newLine();

bw.write(" <bean id=\""+s+"Dao\" class=\""+pkg+".dao.impl."+b+"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();         
    }

2.11、对struts.xml配置文件修改

private void modifyStrutsXml() throws Exception {
//1.读取原始的内容
//2.读取到特定位置（package）添加指定内容

//我们要读的文件与写的文件是同一个文件
/*
 RandomAccessFile类读写文件时
 读取，一共100，读70，写，写的内容会覆盖后30
 111
 222
 333
 444
 在333的后面写5
 111
 222
 333
 544
 在333的后面写5
 111
 222
 333
 555
 */
//方案一：
/*
 读取原始文件，将内容写入新文件
 写之前判断，读取的内容是否是特定内容，特定内容写之前，加入新的内容
 写完毕之后生成了新的文件，删除老的文件，使用新文件更名为老的文件
 */
//方案二：
//1.读取原始文件的文件大小,字节总数1000
        File f = new File("resources/struts.xml");
long len = f.length();
//2.创建一个字节数组，大小等于原始文件字节总数
byte[] buf = new byte[(int)len];
//3.将原始文件读入该byte数组
        InputStream is = new FileInputStream(f);
is.read(buf);
is.close();
//4.将buf转化为字符串
        String all = new String(buf);
//5.查找固定位置
int idx = all.lastIndexOf(" </package>");
//6.将要写入的内容插入该位置
        String info = " <!-- "+b+" -->\r\n <action name=\""+s+"_*\" class=\""+s+"Action\" method=\"{1}\">\r\n </action>\r\n\r\n";
//7.将info加入all的指定位置
        StringBuilder sbf = new StringBuilder(all);
        sbf.insert(idx, info);
//8.将sbf中的组合最终内容写入struts.xml
        FileOutputStream fos = new FileOutputStream(f);
        fos.write(sbf.toString().getBytes());
        fos.close();
    }

2.12、生成工具方法

private Class clazz;
private String b ;      //User
private String l ;      //e
private String s ;      //user
private String pkg ;    //org.sihai.qualitycontrol.auth.user
private String dir ;    //org/sihai/qualitycontrol/auth/user/vo
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();
//9.struts.xml(选作)
//modifyStrutsXml();
    }

三、测试

public static void main(String[] args) throws Exception {
new GeneratorUtil(ResModel.class);
}