New file |
| | |
| | | package com.odcc.cpzidc.test; |
| | | |
| | | import cn.smallbun.screw.core.Configuration; |
| | | import cn.smallbun.screw.core.engine.EngineConfig; |
| | | import cn.smallbun.screw.core.engine.EngineFileType; |
| | | import cn.smallbun.screw.core.engine.EngineTemplateType; |
| | | import cn.smallbun.screw.core.execute.DocumentationExecute; |
| | | import cn.smallbun.screw.core.process.ProcessConfig; |
| | | import com.zaxxer.hikari.HikariConfig; |
| | | import com.zaxxer.hikari.HikariDataSource; |
| | | import org.junit.Test; |
| | | import org.junit.runner.RunWith; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.boot.test.context.SpringBootTest; |
| | | import org.springframework.context.ApplicationContext; |
| | | import org.springframework.test.context.junit4.SpringRunner; |
| | | |
| | | import javax.sql.DataSource; |
| | | import java.util.ArrayList; |
| | | import java.util.Arrays; |
| | | import java.util.List; |
| | | |
| | | @RunWith(SpringRunner.class) |
| | | @SpringBootTest |
| | | public class ScrewApplicationTests { |
| | | |
| | | @Autowired |
| | | private ApplicationContext applicationContext; |
| | | |
| | | @Test |
| | | public void contextLoads() { |
| | | HikariConfig hikariConfig = new HikariConfig(); |
| | | hikariConfig.setDriverClassName("com.mysql.cj.jdbc.Driver"); |
| | | hikariConfig.setJdbcUrl("jdbc:mysql://rm-2ze76o0e1v97mb1689o.mysql.rds.aliyuncs.com:23306/db_cpz_idc?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8"); |
| | | hikariConfig.setUsername("laodai_20220711"); |
| | | hikariConfig.setPassword("odc#cwei288^49"); |
| | | //设置可以获取tables remarks信息 |
| | | hikariConfig.addDataSourceProperty("useInformationSchema", "true"); |
| | | hikariConfig.setMinimumIdle(2); |
| | | hikariConfig.setMaximumPoolSize(5); |
| | | DataSource dataSource = new HikariDataSource(hikariConfig); |
| | | |
| | | // 生成文件配置 |
| | | EngineConfig engineConfig = EngineConfig.builder() |
| | | // 生成文件路径,自己mac本地的地址,这里需要自己更换下路径 |
| | | .fileOutputDir("/Users/daizhiping/V") |
| | | // 打开目录 |
| | | .openOutputDir(false) |
| | | // 文件类型 |
| | | .fileType(EngineFileType.WORD) |
| | | // 生成模板实现 |
| | | .produceType(EngineTemplateType.freemarker).build(); |
| | | |
| | | // 生成文档配置(包含以下自定义版本号、描述等配置连接) |
| | | Configuration config = Configuration.builder() |
| | | .version("2.0.0") |
| | | .description("中国算力平台算力登记系统") |
| | | .dataSource(dataSource) |
| | | .engineConfig(engineConfig) |
| | | .produceConfig(getProcessConfig()) |
| | | .build(); |
| | | |
| | | // 执行生成 |
| | | new DocumentationExecute(config).execute(); |
| | | } |
| | | |
| | | |
| | | /** |
| | | * 配置想要生成的表+ 配置想要忽略的表 |
| | | * @return 生成表配置 |
| | | */ |
| | | public static ProcessConfig getProcessConfig(){ |
| | | // 忽略表名 |
| | | List<String> ignoreTableName = Arrays.asList("aa","test_group"); |
| | | // 忽略表前缀,如忽略a开头的数据库表 |
| | | List<String> ignorePrefix = Arrays.asList("gen_","qrtz_","sys_"); |
| | | // 忽略表后缀 |
| | | List<String> ignoreSuffix = Arrays.asList("_test","czb_"); |
| | | |
| | | return ProcessConfig.builder() |
| | | //根据名称指定表生成 |
| | | .designatedTableName(new ArrayList<>()) |
| | | //根据表前缀生成 |
| | | .designatedTablePrefix(new ArrayList<>()) |
| | | //根据表后缀生成 |
| | | .designatedTableSuffix(new ArrayList<>()) |
| | | //忽略表名 |
| | | .ignoreTableName(ignoreTableName) |
| | | //忽略表前缀 |
| | | .ignoreTablePrefix(ignorePrefix) |
| | | //忽略表后缀 |
| | | .ignoreTableSuffix(ignoreSuffix).build(); |
| | | } |
| | | } |