velocity代码生成器的使用

1.导入依赖

<dependency>
    <groupId>org.apache.velocity</groupId>
 <artifactId>velocity</artifactId>
 <version>1.7</version>
</dependency>

2.resource资源文件中

3.codegenerator.java

package cn.fifdistrict.basic;

import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;
import org.junit.Test;

import java.io.*;
import java.util.Properties;

public class CodeGenerator {

 //代码生成路径
 private static String mapperPath;
 private static String servicePath;
 private static String serviceImplPath;
 private static String controllerPath;
 private static String queryPath;
 private static String jspPath;
 private static String jsPath;

 static{
 //加载路径
 Properties properties = new Properties();
 try {
 properties.load(new InputStreamReader(CodeGenerator.class.getClassLoader().getResourceAsStream("generator.properties"),"utf-8"));
 mapperPath = properties.getProperty("gen.mapper.path");
 servicePath = properties.getProperty("gen.service.path");
 serviceImplPath = servicePath+"\\impl";
 controllerPath = properties.getProperty("gen.controller.path");
 queryPath = properties.getProperty("gen.query.path");
 jspPath = properties.getProperty("gen.jsp.path")+"\\domain";
 jsPath = properties.getProperty("gen.js.path");
 } catch (IOException e) {
 e.printStackTrace();
 }
 }


 //代码生成的先后顺序
 private static final String[] paths = {controllerPath,servicePath,
 serviceImplPath,mapperPath,queryPath,jspPath,jsPath};
 //模板先后顺序
 private static final String[] templates = {"DomainController.java.vm",
 "IDomainService.java.vm","DomainServiceImpl.java.vm",
 "DomainMapper.java.vm","DomainQuery.java.vm","index.jsp.vm",
 "domain.js.vm"};

 //要生成的那些实体类的相关代码
 private static final String[] domains = {"Operationlog"};

 //是否覆盖
 private static final boolean FLAG = true;

 @Test
 public void test() throws Exception{

 //模板上下文
 VelocityContext context = new VelocityContext();
 //遍历所有的domain
 for (String domain : domains) {
 //拿到domain类名的大小写
 String DomainName = domain;
 String domainName = domain.substring(0,1).toLowerCase()+
 domain.substring(1);
 //上下文中设置替换内容
 context.put("Domain",DomainName);
 context.put("domain",domainName);
 //遍历路径，拿到模板生成目标文件
 for (int i=0;i<paths.length;i++) {

 //初始化参数
 Properties properties=new Properties();
 //设置velocity资源加载方式为class
 properties.setProperty("resource.loader", "class");
 //设置velocity资源加载方式为file时的处理类
 properties.setProperty("class.resource.loader.class", "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader");
 //实例化一个VelocityEngine对象
 VelocityEngine velocityEngine=new VelocityEngine(properties);

 String templateName = templates[i];
 //生成代码
 //生成的文件名
 String fileName = templateName.substring(0, templateName.lastIndexOf(".vm"));
 String filePath = paths[i]+"\\"+fileName;
 filePath = filePath.replaceAll("domain", domainName).
 replaceAll("Domain", DomainName);
 File file = new File(filePath);

 System.out.println(filePath);

 //判断是否覆盖存在的文件
 if(file.exists()&&!FLAG){
 continue;
 }

 //获取父目录
 File parentFile = file.getParentFile();
 if(!parentFile.exists()){
 parentFile.mkdirs();
 }
 Writer writer = new FileWriter(file);
 //拿到模板，设置编码
 velocityEngine.mergeTemplate("template/"+templateName,"utf-8",context,writer);
 writer.close();

 }

 }

 }

}

rebel.xml

<?xml version="1.0" encoding="UTF-8"?>

<!--
 This is the JRebel configuration file. It maps the running application to your IDE workspace, enabling JRebel reloading for this project.
 Refer to https://manuals.zeroturnaround.com/jrebel/standalone/config.html for more information.
-->
<application generated-by="intellij" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://www.zeroturnaround.com" xsi:schemaLocation="http://www.zeroturnaround.com http://update.zeroturnaround.com/jrebel/rebel-2_1.xsd">

 <classpath>
 <dir name="D:/ruanjian/code/workspace/dms-parent/basic-generator/target/classes">
 </dir>
 </classpath>

</application>

rebel-remote.xml

<?xml version="1.0" encoding="UTF-8"?>
<rebel-remote xmlns="http://www.zeroturnaround.com/rebel/remote">
 <id>basic-generator</id>
</rebel-remote>

--generator.properties

#代码生成路径
#query生成的目录
gen.query.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-common\\src\\main\\java\\cn\\fifdistrict\\dms\\query
#mapper生成的目录
gen.mapper.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-mapper\\src\\main\\java\\cn\\fifdistrict\\dms\\mapper
#controller生成的目录
gen.controller.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-web\\src\\main\\java\\cn\\fifdistrict\\dms\\web\\controller
#service生成的目录
gen.service.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-service\\src\\main\\java\\cn\\fifdistrict\\dms\\service
#jsp生成的目录
gen.jsp.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-web\\src\\main\\webapp\\WEB-INF\\views
#js生成的目录
gen.js.path=D:\\ruanjian\\code\\workspace\\dms-parent\\dms-web\\src\\main\\webapp\\static\\js\\model

--template

package cn.fifdistrict.dms.web.controller;

import cn.fifdistrict.dms.domain.${Domain};
import cn.fifdistrict.dms.query.${Domain}Query;
import cn.fifdistrict.dms.service.I${Domain}Service;
import cn.fifdistrict.basic.util.AjaxResult;
import cn.fifdistrict.basic.util.PageList;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

import java.util.List;


@Controller
@RequestMapping("/${domain}")
public class ${Domain}Controller {

 @Autowired
 private I${Domain}Service ${domain}Service;

 /**
 * 保存
 * @return
 */
 @RequestMapping(value = "save",method = RequestMethod.POST)
 @ResponseBody
 public AjaxResult save(${Domain} ${domain}){
 try {
 if(${domain}.getId()==null){
 ${domain}Service.add(${domain});
 }else{
 ${domain}Service.update(${domain});
 }
 return new AjaxResult();
 }catch (Exception e){
 e.printStackTrace();
 return new AjaxResult("保存失败!"+e.getMessage());
 }
 }

 /**
 * 删除
 * @param id
 * @return
 */
 @RequestMapping(value = "/delete",method = RequestMethod.POST)
 @ResponseBody
 public AjaxResult delete(Long id){
 try {
 ${domain}Service.delete(id);
 return new AjaxResult();
 }catch (Exception e){
 e.printStackTrace();
 return new AjaxResult("删除失败!"+e.getMessage());
 }
 }

 /**
 * 根据id查询
 * @param id
 * @return
 */
 @RequestMapping(value = "/getOne",method = RequestMethod.GET)
 @ResponseBody
 public ${Domain} getOne(Long id){
 return ${domain}Service.get(id);
 }

 /**
 * 查询所有
 * @return
 */
 @RequestMapping(value = "/getAll",method = RequestMethod.GET)
 @ResponseBody
 public List<${Domain}> getAll(){
 return ${domain}Service.getAll();
 }

 /**
 * 条件分页查询
 * @param query
 * @return
 */
 @RequestMapping(value = "/query",method = RequestMethod.GET)
 @ResponseBody
 public PageList<${Domain}> query(${Domain}Query query){
 return ${domain}Service.queryPage(query);
 }
 
}