use of com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine in project chao-cloud by chaojunzi.
the class ZipAutoGenerator method execute.
/**
* 输出到流-zip
* @param out 输出流
* @throws IOException 生成模板文件抛出的异常
*/
public void execute(OutputStream out) throws Exception {
// 初始化配置
config = new ConfigBuilder(super.getPackageInfo(), super.getDataSource(), super.getStrategy(), super.getTemplate(), super.getGlobalConfig());
if (null != injectionConfig) {
injectionConfig.setConfig(config);
}
if (null == super.getTemplateEngine()) {
// 为了兼容之前逻辑,采用 Velocity 引擎 【 默认 】
super.setTemplateEngine(new ZipVelocityTemplateEngine());
}
// 模板引擎初始化执行文件输出
ZipVelocityTemplateEngine engine = (ZipVelocityTemplateEngine) super.getTemplateEngine().init(this.pretreatmentConfigBuilder(config));
// 使用zip输出流
try (ZipOutputStream zip = new ZipOutputStream(out)) {
// 输出
engine.batchOutput(zip);
}
log.info("==========================文件生成完成!!!==========================");
}
use of com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine in project chao-cloud by chaojunzi.
the class MybatisGeneratorConfig method autoGenerator.
@Bean
public //
ZipAutoGenerator autoGenerator(//
GlobalConfig gc, //
DataSourceConfig dsc, //
PackageConfig pc, //
InjectionConfig cfg, TemplateConfig tc, StrategyConfig sc) {
// 配置
ZipAutoGenerator mpg = new ZipAutoGenerator();
mpg.setGlobalConfig(gc);
mpg.setDataSource(dsc);
mpg.setPackageInfo(pc);
mpg.setCfg(cfg);
mpg.setTemplate(tc);
mpg.setStrategy(sc);
mpg.setTemplateEngine(new ZipVelocityTemplateEngine());
return mpg;
}
Aggregations