Search in sources :

Example 1 with ZipVelocityTemplateEngine

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("==========================文件生成完成!!!==========================");
}
Also used : ZipOutputStream(java.util.zip.ZipOutputStream) ConfigBuilder(com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder) ZipVelocityTemplateEngine(com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine)

Example 2 with ZipVelocityTemplateEngine

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;
}
Also used : ZipAutoGenerator(com.chao.cloud.common.extra.mybatis.generator.ZipAutoGenerator) ZipVelocityTemplateEngine(com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine) Bean(org.springframework.context.annotation.Bean)

Aggregations

ZipVelocityTemplateEngine (com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine)2 ConfigBuilder (com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder)1 ZipAutoGenerator (com.chao.cloud.common.extra.mybatis.generator.ZipAutoGenerator)1 ZipOutputStream (java.util.zip.ZipOutputStream)1 Bean (org.springframework.context.annotation.Bean)1