|
@@ -1,13 +1,81 @@
|
|
|
package com.shanghaichengdi.ghjgitem;
|
|
|
|
|
|
+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.HikariDataSource;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
|
|
|
|
+import java.util.ArrayList;
|
|
|
+import java.util.Arrays;
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
@SpringBootTest
|
|
|
class GhjgItemServerApplicationTests {
|
|
|
|
|
|
@Test
|
|
|
- void contextLoads() {
|
|
|
+ void generate() {
|
|
|
+
|
|
|
+ EngineConfig engineConfig = EngineConfig.builder()
|
|
|
+
|
|
|
+ .fileOutputDir("E:\\DataShareFolder\\tableStructure")
|
|
|
+
|
|
|
+ .openOutputDir(false)
|
|
|
+
|
|
|
+ .fileType(EngineFileType.WORD)
|
|
|
+
|
|
|
+ .produceType(EngineTemplateType.freemarker).build();
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ HikariDataSource hikariDataSource = new HikariDataSource();
|
|
|
+
|
|
|
+ hikariDataSource.setJdbcUrl("jdbc:oracle:thin:@//10.114.4.56/shhwdb");
|
|
|
+ hikariDataSource.setUsername("ghjg_channel");
|
|
|
+ hikariDataSource.setPassword("GHJG@123");
|
|
|
+
|
|
|
+ Configuration config = Configuration.builder()
|
|
|
+ .version("1.0.1")
|
|
|
+ .description("数据库设计文档")
|
|
|
+ .dataSource(hikariDataSource)
|
|
|
+ .engineConfig(engineConfig)
|
|
|
+ .produceConfig(getProcessConfig())
|
|
|
+ .build();
|
|
|
+
|
|
|
+
|
|
|
+ new DocumentationExecute(config).execute();
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * 配置想要生成的表+ 配置想要忽略的表
|
|
|
+ *
|
|
|
+ * @return 生成表配置
|
|
|
+ */
|
|
|
+ public static ProcessConfig getProcessConfig() {
|
|
|
+
|
|
|
+ List<String> ignoreTableName = Arrays.asList("testa_testa", "testb_testb");
|
|
|
+
|
|
|
+ List<String> ignorePrefix = Arrays.asList("testa", "testb");
|
|
|
+
|
|
|
+ List<String> ignoreSuffix = Arrays.asList("_testa", "_testb");
|
|
|
+ return ProcessConfig.builder()
|
|
|
+
|
|
|
+ .designatedTableName(new ArrayList<>())
|
|
|
+
|
|
|
+ .designatedTablePrefix(new ArrayList<>())
|
|
|
+
|
|
|
+ .designatedTableSuffix(new ArrayList<>())
|
|
|
+
|
|
|
+ .ignoreTableName(ignoreTableName)
|
|
|
+
|
|
|
+ .ignoreTablePrefix(ignorePrefix)
|
|
|
+
|
|
|
+ .ignoreTableSuffix(ignoreSuffix).build();
|
|
|
}
|
|
|
|
|
|
}
|