<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
	xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
	<modelVersion>4.0.0</modelVersion>
	<parent>
		<groupId>org.springframework.boot</groupId>
		<artifactId>spring-boot-starter-parent</artifactId>
		<version>3.3.3</version>
		<relativePath/> <!-- lookup parent from repository -->
	</parent>
	<groupId>com.easylinkin</groupId>
	<artifactId>my-screw</artifactId>
	<version>0.0.1-SNAPSHOT</version>
	<name>my-screw</name>
	<description>my-screw</description>
	<url/>
	<licenses>
		<license/>
	</licenses>
	<developers>
		<developer/>
	</developers>
	<scm>
		<connection/>
		<developerConnection/>
		<tag/>
		<url/>
	</scm>
	<properties>
		<java.version>17</java.version>
	</properties>
	<dependencies>
		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter</artifactId>
		</dependency>

		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-test</artifactId>
			<scope>test</scope>
		</dependency>

		<dependency>
			<groupId>org.springframework.boot</groupId>
			<artifactId>spring-boot-starter-web</artifactId>
		</dependency>
		<dependency>
			<groupId>cn.smallbun.screw</groupId>
			<artifactId>screw-core</artifactId>
			<version>1.0.5</version>
		</dependency>
		<dependency>
			<groupId>com.mysql</groupId>
			<artifactId>mysql-connector-j</artifactId>
		</dependency>
		<dependency>
			<groupId>com.zaxxer</groupId>
			<artifactId>HikariCP</artifactId>
		</dependency>
		<dependency>
			<groupId>org.projectlombok</groupId>
			<artifactId>lombok</artifactId>
		</dependency>
	</dependencies>

	<build>
		<plugins>
			<plugin>
				<groupId>org.springframework.boot</groupId>
				<artifactId>spring-boot-maven-plugin</artifactId>
			</plugin>
		</plugins>
	</build>

</project>

package com.easylinkin.myscrew;

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 lombok.extern.slf4j.Slf4j;
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;

@Slf4j
@SpringBootTest
class MyScrewApplicationTests {

	//@Test
	void contextLoads() {
	}

	/**
	 * 獲取數(shù)據(jù)源配置
	 */
	@Test
	void generate() {
		log.info("開(kāi)始生成數(shù)據(jù)庫(kù)文檔");
		String dbFilePath = "/Users/zwmac/work/ovu/數(shù)據(jù)庫(kù)";
		// 生成文件配置
		EngineConfig engineConfig = EngineConfig.builder()
				// 生成文件路徑，自己mac本地的地址，這里需要自己更換下路徑
				.fileOutputDir(dbFilePath)
				// 打開(kāi)目錄
				.openOutputDir(false)
				// 文件類型 HTML/WORD/MD 三種格式
				.fileType(EngineFileType.WORD)
				// 生成模板實(shí)現(xiàn)
				.produceType(EngineTemplateType.freemarker).build();

		//數(shù)據(jù)庫(kù)名稱
		String[] dbNames = {"ovuhome"};
		for (String dbName : dbNames) {
			HikariDataSource hikariDataSource = new HikariDataSource();
			//設(shè)置數(shù)據(jù)庫(kù)連接
			hikariDataSource.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/" + dbName + "?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=GMT%2B8&allowMultiQueries=true&allowPublicKeyRetrieval=true");
			hikariDataSource.setUsername("root");
			hikariDataSource.setPassword("12345678");
			// 生成文檔配置（包含以下自定義版本號(hào)、描述等配置連接）
			Configuration config = Configuration.builder()
					//文檔版本號(hào)
					.version("1.0.1")
					//文檔名稱
					.description("公寓數(shù)據(jù)庫(kù)設(shè)計(jì)文檔")
					//數(shù)據(jù)源
					.dataSource(hikariDataSource)
					//生成引擎配置
					.engineConfig(engineConfig)
					//高級(jí)配置
					//.produceConfig(getProcessConfig())
					.build();

			// 執(zhí)行生成
			new DocumentationExecute(config).execute();
		}
		log.info("生成數(shù)據(jù)庫(kù)文檔完成");
	}

	/**
	 * 配置想要生成的表+ 配置想要忽略的表
	 * @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()
				//根據(jù)名稱指定表生成 我需要生成所有表 這里暫時(shí)不設(shè)置
				.designatedTableName(new ArrayList<>())
				//根據(jù)表前綴生成 我需要生成所有表 這里暫時(shí)不設(shè)置
				.designatedTablePrefix(new ArrayList<>())
				//根據(jù)表后綴生成 我需要生成所有表 這里暫時(shí)不設(shè)置
				.designatedTableSuffix(new ArrayList<>())
				//忽略表名
				.ignoreTableName(ignoreTableName)
				//忽略表前綴
				.ignoreTablePrefix(ignorePrefix)
				//忽略表后綴
				.ignoreTableSuffix(ignoreSuffix).build();
	}

}