package com.yqzx.generator.engine.base;
|
|
import cn.hutool.core.io.FileUtil;
|
//import com.sun.javafx.projectUtil;
|
import lombok.extern.slf4j.Slf4j;
|
import org.beetl.core.Configuration;
|
import org.beetl.core.GroupTemplate;
|
import org.beetl.core.Template;
|
import org.beetl.core.resource.ClasspathResourceLoader;
|
|
import java.io.File;
|
import java.io.FileNotFoundException;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.util.Properties;
|
|
/**
|
* @description: ADI项目模板生成 引擎
|
* @author: chaoyapeng
|
* @time: 2020/8/18 16:03
|
*/
|
@Slf4j
|
public abstract class AbstractTemplateEngine extends BaseTemplateEngine {
|
|
private GroupTemplate groupTemplate;
|
|
public AbstractTemplateEngine() {
|
initBeetlEngine();
|
}
|
|
protected void initBeetlEngine() {
|
Properties properties = new Properties();
|
properties.put("RESOURCE.root", "");
|
properties.put("DELIMITER_STATEMENT_START", "<%");
|
properties.put("DELIMITER_STATEMENT_END", "%>");
|
properties.put("HTML_TAG_FLAG", "##");
|
Configuration cfg = null;
|
try {
|
cfg = new Configuration(properties);
|
} catch (IOException e) {
|
log.error(e.getMessage());
|
}
|
ClasspathResourceLoader resourceLoader = new ClasspathResourceLoader();
|
groupTemplate = new GroupTemplate(resourceLoader, cfg);
|
}
|
|
protected void configTemplate(Template template) {
|
template.binding("context", super.contextConfig);
|
template.binding("controller", super.controllerConfig);
|
template.binding("service", super.serviceConfig);
|
template.binding("mapper", super.mapperConfig);
|
template.binding("model", super.modelConfig);
|
template.binding("query", super.queryConfig);
|
template.binding("tableInfo", super.tableInfo);
|
template.binding("columnStr", super.columnStr);
|
// 绑定查询参数列表
|
template.binding("queryTableInfo",super.queryTableInfo);
|
}
|
|
protected void generateFile(String template, String filePath) {
|
Template pageTemplate = groupTemplate.getTemplate(template);
|
configTemplate(pageTemplate);
|
filePath = filePath.replaceAll("/+|\\\\+", "\\\\");
|
//if (projectUtil.isWindows()) {
|
// filePath = filePath.replaceAll("/+|\\\\+", "\\\\");
|
//} else {
|
// filePath = filePath.replaceAll("/+|\\\\+", "/");
|
//}
|
// File file = new File(filePath);
|
File file = FileUtil.touch(filePath);
|
|
File parentFile = file.getParentFile();
|
if (!parentFile.exists()) {
|
parentFile.mkdirs();
|
}
|
FileOutputStream fileOutputStream = null;
|
try {
|
fileOutputStream = new FileOutputStream(file);
|
pageTemplate.renderTo(fileOutputStream);
|
} catch (FileNotFoundException e) {
|
log.error(e.getMessage());
|
} finally {
|
try {
|
fileOutputStream.close();
|
} catch (IOException e) {
|
log.error(e.getMessage());
|
}
|
}
|
}
|
|
public void start() {
|
//配置之间的相互依赖
|
super.initConfig();
|
|
//生成模板
|
if (super.contextConfig.getControllerSwitch()) {
|
generateController();
|
}
|
if (super.contextConfig.getServiceSwitch()) {
|
generateService();
|
}
|
if (super.contextConfig.getMapperSwitch()) {
|
generateMapper();
|
}
|
if (super.contextConfig.getModelSwitch()) {
|
generateModel();
|
}
|
if (super.contextConfig.getQuerySwitch()) {
|
generateQuery();
|
}
|
}
|
|
|
/**
|
* 生成Controller
|
*/
|
protected abstract void generateController();
|
|
/**
|
* 生成Service
|
*/
|
protected abstract void generateService();
|
|
/**
|
* 生成Mapper
|
*/
|
protected abstract void generateMapper();
|
|
/**
|
* 生成Model
|
*/
|
protected abstract void generateModel();
|
|
/**
|
* 生成Query
|
*/
|
protected abstract void generateQuery();
|
|
}
|
