FreeMakerUtil.java

package cn.baiweigang.qtaf.toolkit.util;


import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.Map;

import freemarker.template.Configuration;
import freemarker.template.DefaultObjectWrapper;
import freemarker.template.Template;
import freemarker.template.TemplateException;

/**
 * 根据模板+数据 输出文件
 * @author @<a href='http://weibo.com/bwgang'>bwgang</a> (bwgang@163.com)<br/>
 */
public class FreeMakerUtil {
	//日志记录
	private static LogUtil log=LogUtil.getLogger(FreeMakerUtil.class);

	/**
	 * 根据模板+数据 输出文件
	 * @param templateFilePath 模板的绝对路径
	 * @param data 数据
	 * @param javaFilePath 输出的文件位置
	 * @return boolean
	 */
	public boolean CreateJavaFile(String templateFilePath,Map<String, Object> data,String javaFilePath) {
		
		if (!FileUtil.isExist(templateFilePath)) {
			log.info("模板文件不存在："+templateFilePath);
			return false;
		}
		if (null == data) {
			log.info("模板数据不能为null");
			return false;
		}
//		Map<String, String> clsMap= (Map<String, String>) data.get("clsInfo");
//		for (Entry<String, String> entry : clsMap.entrySet()) {
//			String key = entry.getKey().toString();
//			String value = entry.getValue().toString();
//			log.info(key+"=="+value);
//		}
//		log.info("FreeMaker--39--"+data.size());
//		PrintModelObject.P(data.get("clsInfo"));
		if (!FileUtil.createFile(javaFilePath)) {
			return false;
		}
		try {
			String tmpPath=FileUtil.getParentPath(templateFilePath);
			String tmpName=FileUtil.getFileName(templateFilePath);
			
			Configuration cfg =new Configuration();
			cfg.setDirectoryForTemplateLoading(new File(tmpPath));
			cfg.setObjectWrapper(new DefaultObjectWrapper());
			
			//读取模板
			Template template= cfg.getTemplate(tmpName);
			
			//输出到文件
			Writer out = new OutputStreamWriter(new FileOutputStream(javaFilePath),"UTF-8"); 
			template.process(data, out);//根据模板和数据输出java文件
			
			return true;
			
		} catch (IOException | TemplateException e) {
			//生成.java文件失败
			log.error(e.getMessage());
			return false;
		}
	}
}