代码生成器--4.23课程

1、新建一个Code的包，定义一个CodeBuilder类
代码如下

package com.zq.code;

import java.io.PrintWriter;
import java.util.Properties;

import org.apache.velocity.Template;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.VelocityEngine;

public class CodeBuilder {

	public static void main(String[] args) {
		//new一下他的模板引擎
		VelocityEngine ve = new VelocityEngine();
		//设置模板和输出的代码的文件的编码方式
		Properties p= new Properties();
		p.setProperty(VelocityEngine.ENCODING_DEFAULT, "UTF-8");
		p.setProperty(VelocityEngine.INPUT_ENCODING, "UTF-8");
		p.setProperty(VelocityEngine.OUTPUT_ENCODING, "UTF-8");
		ve.init(p);//引擎初始化
		
		//映入一个模板，通过模板路径
		Template serviceVm = ve.getTemplate("/WebContent/WEB-INF/vm/service.vm");
		
		//定义占位符变量，给个值
		String modelClass = "Menu";
		String modelName = "menu";
		//生成的代码放置的目录==项目目录
		String rootPath = "F:/企业/shop";
		
		//变量放在上下文对象
		VelocityContext ctx = new VelocityContext();
		ctx.put("modelClass", modelClass);
		ctx.put("modelName", modelName);
		
		//将占位符数据和模板合并，输出代码文件
		CodeBuilder.merge(serviceVm, ctx, rootPath + "/src/com/zq/service/" + modelClass + "Service.java");
		System.out.println("成功");
	}
	
	//定义一个新的方法
	private static void merge(Template template,VelocityContext ctx,String path){
		PrintWriter writer = null;
		try{
			writer = new PrintWriter(path);
			//合并数据和模板，输出文件
			template.merge(ctx, writer);
			writer.flush();
		}catch(Exception e){
			e.printStackTrace();
		}finally{
			writer.close();
		}
	}
}

2、新建一个vm文件夹，位于web-inf,在vm里新建一个service.vm文件
代码如下

package com.zq.service;

import java.util.List;
import com.github.pagehelper.PageInfo;
import com.zq.model.${modelClass};

public interface ${modelClass}Service {
	
	/**
	 * 带有分页的列表查询
	 * @param ${modelClass}
	 * @return
	 */
	public PageInfo<${modelClass}> list(Integer pageNum, Integer pageSize, ${modelClass} ${modelName});
	
	/**
	 * 没有分页的列表查询
	 * @param ${modelClass}
	 * @return
	 */
	public List<${modelClass}> list(${modelClass} ${modelName});
	public void create(${modelClass} ${modelName});
	public void update(${modelClass} ${modelName});
	public void delete(Integer id);
	public ${modelClass} findById(Integer id);
}

3、将UserService.java修改

package com.zq.service;

import java.util.List;

import com.github.pagehelper.PageInfo;
import com.zq.model.User;

public interface UserService {
	
	public List<User> list(User user);
	public void create(User user);
	public void update(User user);
	public void delete(Integer id);
	public void updateBatch(List<Integer> list);
	public User findById(Integer id);
	public void updateGreat();
	public PageInfo<User> list(Integer pageNum, Integer pageSize, User user);
}

4、右键Run as 选择java application运行