Search in sources :

Example 1 with ConfigBuilder

use of com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder 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 ConfigBuilder

use of com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder in project matecloud by matevip.

the class SysCodeController method tableList.

/**
 * 数据库表信息
 * @param dataSourceId 数据源Id
 * @return Result
 */
@PreAuth
@Log(value = "数据库表信息", exception = "数据库表信息请求异常")
@ApiOperation(value = "数据库表信息", notes = "数据库表信息")
@PostMapping("/table-list")
public Result<List<TableInfo>> tableList(@RequestParam String dataSourceId) {
    SysDataSource sysDataSource = sysDataSourceService.getById(dataSourceId);
    GlobalConfig gc = new GlobalConfig();
    DataSourceConfig dsc = new DataSourceConfig();
    dsc.setDbType(DbType.getDbType(sysDataSource.getDbType()));
    dsc.setDriverName(sysDataSource.getDriverClass());
    dsc.setUrl(sysDataSource.getUrl());
    dsc.setUsername(sysDataSource.getUsername());
    dsc.setPassword(sysDataSource.getPassword());
    StrategyConfig strategyConfig = new StrategyConfig();
    TemplateConfig templateConfig = new TemplateConfig();
    ConfigBuilder config = new ConfigBuilder(new PackageConfig(), dsc, strategyConfig, templateConfig, gc);
    List<TableInfo> list = config.getTableInfoList();
    return Result.data(list);
}
Also used : ConfigBuilder(com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder) SysDataSource(vip.mate.code.entity.SysDataSource) TableInfo(com.baomidou.mybatisplus.generator.config.po.TableInfo) PreAuth(vip.mate.core.auth.annotation.PreAuth) PostMapping(org.springframework.web.bind.annotation.PostMapping) Log(vip.mate.core.log.annotation.Log) ApiOperation(io.swagger.annotations.ApiOperation)

Aggregations

ConfigBuilder (com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder)2 TableInfo (com.baomidou.mybatisplus.generator.config.po.TableInfo)1 ZipVelocityTemplateEngine (com.chao.cloud.common.extra.mybatis.generator.engine.ZipVelocityTemplateEngine)1 ApiOperation (io.swagger.annotations.ApiOperation)1 ZipOutputStream (java.util.zip.ZipOutputStream)1 PostMapping (org.springframework.web.bind.annotation.PostMapping)1 SysDataSource (vip.mate.code.entity.SysDataSource)1 PreAuth (vip.mate.core.auth.annotation.PreAuth)1 Log (vip.mate.core.log.annotation.Log)1