代码生成器的步骤

代码生成器可以实现快速开发

如果添加部门模型，复制IDepartmentService,修改类名,ctrl+f，进行人工替换

代码生成器就是先定义好模板，然后使用代码自动修改类名，进行名称的自动替换

 

关键:先备份一下项目

 

1.那些模型需要生成代码:先用一个临时模型Dept，进行测试

2.有那些模板要生成:添加类或者jsp,xml,js

3.上面的面板生成到那个文件夹

 

 

4.定义一下常量，规定文件夹的位置

  格式：全部都是以相对位置开始，前面没有/

        以/结尾

5.外循环：模型数组

6.内循环：模板或者文件夹

7.Velocity上下文

8.添加数据到Velocity上下文：大小写

9.获取模板

10.实例化File对象，打印输出绝对路径:步骤11之后

11.处理特殊的文件名称或者位置

12.实例化FileWriter

13.合并模板，关闭流

14.创建父目录

15.如果已经存在的文件，默认是不生成的，防止覆盖,千万不要用Employee来测试

 

 

package cn.itsource.pss;
import java.io.File;
import java.io.FileWriter;
import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;
import org.junit.Test;
public class CreateCodeTest {
// 15.如果已经存在的文件，默认是不生成的，防止覆盖,千万不要用Employee来测试
// 规则flag = false，不覆盖，
// 规则flag = true，覆盖，
private boolean flag = false;
// 1.定义路径固定值
// *********************规则：路径前面不要写/，后面结尾必须写/
private String SRC = "src/main/java/";
// 容易写错的地方：不能包路径，必须是文件路径
private String PACKAGE = "cn/itsource/pss/";
private String RESOURCES = "src/main/resources/";
private String TEST = "src/test/java/";
private String WEBAPP = "src/main/webapp/";
// 2.定义那些domain需要生成代码
// private String[] domains = { "Department" };
private String[] domains = { "Role", "Permission" };
// 3.定义模板的名称
private String[] templates = { "Service.java", "ServiceImpl.java", "Query.java", "Action.java", "ServiceTest.java",
"hbm.xml", "Context.xml", "domain.js", "list.jsp", "input.jsp" };
// 4.定义模板对应生成文件的位置
private String[] files = { SRC + PACKAGE + "service/", SRC + PACKAGE + "service/impl/", SRC + PACKAGE + "query/",
SRC + PACKAGE + "web/action/", TEST + PACKAGE + "service/", RESOURCES + PACKAGE + "domain/",
RESOURCES + "manager/", WEBAPP + "js/model/", WEBAPP + "WEB-INF/views/", WEBAPP + "WEB-INF/views/" };
@Test
public void create() throws Exception {
if (templates.length != files.length) {
throw new RuntimeException("模板!=模板对应生成文件");
}
// 实例化Velocity上下文对象
VelocityContext context = new VelocityContext();
// 5.外循环
for (int i = 0; i < domains.length; i++) {
// 准备数据,类似于struts放入值栈:String getText(){ return "xxxx"}
context.put("domainEntity", domains[i]);
// 获取domain模型首字母小写
String lowerDomainEntity = domains[i].substring(0, 1).toLowerCase() + domains[i].substring(1);
context.put("lowerDomainEntity", lowerDomainEntity);
// 6.内循环
for (int j = 0; j < templates.length; j++) {
// src/main/java/cn/itsource/pss/DeptService.java
File file = new File(files[j] + domains[i] + templates[j]);
if ("Service.java".equals(templates[j])) {
file = new File(files[j] + "I" + domains[i] + templates[j]);
} else if ("hbm.xml".equals(templates[j])) {
file = new File(files[j] + domains[i] + "." + templates[j]);
} else if ("domain.js".equals(templates[j])) {
file = new File(files[j] + lowerDomainEntity + ".js");
} else if ("list.jsp".equals(templates[j])) {
file = new File(files[j] + lowerDomainEntity + "/" + lowerDomainEntity + ".jsp");
} else if ("input.jsp".equals(templates[j])) {
file = new File(files[j] + lowerDomainEntity + "/" + lowerDomainEntity + "_input.jsp");
}
// 规则flag = false，不覆盖，
// 规则flag = true，覆盖，
// 如果不覆盖并且代码文件已经存在：不需要生成
if (!flag && file.exists()) {
// return;
// break;
continue; // 结束本次循环，进入下一次循环

}
// 打印必须在修改路径之后

System.out.println(file.getAbsolutePath());
// 生成没有的路径
File parentFile = file.getParentFile();
if (!parentFile.exists()) {
parentFile.mkdirs();
}
FileWriter writer = new FileWriter(file);
Template template = Velocity.getTemplate("template/" + templates[j], "UTF-8");
template.merge(context, writer);
// 关闭流

writer.close();
}
}
System.out.println("先刷新工程,没有错误，修改映射文件，运行测试，启动tomcat");
}
}