堪稱資料庫界的 Swagger！一鍵生成資料庫檔案。。。

1、簡介

在企業級開發中、我們經常會有撰寫資料庫表結構檔案的時間付出，從業以來，待過幾家企業，關于資料庫表結構檔案狀態：要么沒有、要么有、但都是手寫、后期運維開發，需要手動進行維護到檔案中，很是繁瑣、如果忘記一次維護、就會給以后作業造成很多困擾、無形中制造了很多坑留給自己和后人，于是萌生了要自己寫一個插件工具的想法

2、特點

• 簡潔、輕量、設計良好
• 多資料庫支持
• 多種格式檔案
• 靈活擴展
• 支持自定義模板

3、資料庫支持

• MySQL
• MariaDB
• TIDB
• Oracle
• SqlServer
• PostgreSQL
• Cache DB（2016）
• H2 （開發中）
• DB2 （開發中）
• HSQL （開發中）
• SQLite（開發中）
• 瀚高（開發中）
• 達夢 （開發中）
• 虛谷 （開發中）
• 人大金倉（開發中）

4、檔案生成支持

• html
• word
• markdown

5、檔案截圖

html

word

markdwon

6、使用方式

普通方式

• 引入依賴

<dependency>
    <groupId>cn.smallbun.screw</groupId>
    <artifactId>screw-core</artifactId>
    <version>${lastVersion}</version>
 </dependency>

• 撰寫代碼

/**
 * 檔案生成
 */
void documentGeneration() {
   //資料源
   HikariConfig hikariConfig = new HikariConfig();
   hikariConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
   hikariConfig.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/database");
   hikariConfig.setUsername("root");
   hikariConfig.setPassword("password");
   //設定可以獲取tables remarks資訊
   hikariConfig.addDataSourceProperty("useInformationSchema", "true");
   hikariConfig.setMinimumIdle(2);
   hikariConfig.setMaximumPoolSize(5);
   DataSource dataSource = new HikariDataSource(hikariConfig);
   //生成配置
   EngineConfig engineConfig = EngineConfig.builder()
         //生成檔案路徑
         .fileOutputDir(fileOutputDir)
         //打開目錄
         .openOutputDir(true)
         //檔案型別
         .fileType(EngineFileType.HTML)
         //生成模板實作
         .produceType(EngineTemplateType.freemarker)
         //自定義檔案名稱
         .fileName("自定義檔案名稱").build();

   //忽略表
   ArrayList<String> ignoreTableName = new ArrayList<>();
   ignoreTableName.add("test_user");
   ignoreTableName.add("test_group");
   //忽略表前綴
   ArrayList<String> ignorePrefix = new ArrayList<>();
   ignorePrefix.add("test_");
   //忽略表后綴
   ArrayList<String> ignoreSuffix = new ArrayList<>();
   ignoreSuffix.add("_test");
   ProcessConfig processConfig = ProcessConfig.builder()
         //指定生成邏輯、當存在指定表、指定表前綴、指定表后綴時，將生成指定表，其余表不生成、并跳過忽略表配置
   //根據名稱指定表生成
   .designatedTableName(new ArrayList<>())
   //根據表前綴生成
   .designatedTablePrefix(new ArrayList<>())
   //根據表后綴生成
   .designatedTableSuffix(new ArrayList<>())
         //忽略表名
         .ignoreTableName(ignoreTableName)
         //忽略表前綴
         .ignoreTablePrefix(ignorePrefix)
         //忽略表后綴
         .ignoreTableSuffix(ignoreSuffix).build();
   //配置
   Configuration config = Configuration.builder()
         //版本
         .version("1.0.0")
         //描述
         .description("資料庫設計檔案生成")
         //資料源
         .dataSource(dataSource)
         //生成配置
         .engineConfig(engineConfig)
         //生成配置
         .produceConfig(processConfig)
         .build();
   //執行生成
   new DocumentationExecute(config).execute();
}

Maven 插件

<build>
    <plugins>
        <plugin>
            <groupId>cn.smallbun.screw</groupId>
            <artifactId>screw-maven-plugin</artifactId>
            <version>${lastVersion}</version>
            <dependencies>
                <!-- HikariCP -->
                <dependency>
                    <groupId>com.zaxxer</groupId>
                    <artifactId>HikariCP</artifactId>
                    <version>3.4.5</version>
                </dependency>
                <!--mysql driver-->
                <dependency>
                    <groupId>mysql</groupId>
                    <artifactId>mysql-connector-java</artifactId>
                    <version>8.0.20</version>
                </dependency>
            </dependencies>
            <configuration>
                <!--username-->
                <username>root</username>
                <!--password-->
                <password>password</password>
                <!--driver-->
                <driverClassName>com.mysql.cj.jdbc.Driver</driverClassName>
                <!--jdbc url-->
                <jdbcUrl>jdbc:mysql://127.0.0.1:3306/xxxx</jdbcUrl>
                <!--生成檔案型別-->
                <fileType>HTML</fileType>
                <!--打開檔案輸出目錄-->
                <openOutputDir>false</openOutputDir>
                <!--生成模板-->
                <produceType>freemarker</produceType>
                <!--檔案名稱 為空時:將采用[資料庫名稱-描述-版本號]作為檔案名稱-->
                <fileName>測驗檔案名稱</fileName>
                <!--描述-->
                <description>資料庫檔案生成</description>
                <!--版本-->
                <version>${project.version}</version>
                <!--標題-->
                <title>資料庫檔案</title>
            </configuration>
            <executions>
                <execution>
                    <phase>compile</phase>
                    <goals>
                        <goal>run</goal>
                    </goals>
                </execution>
            </executions>
        </plugin>
    </plugins>
</build>

7、擴展模塊

pojo生成功能

• 功能簡介

pojo生成功能是基于screw延伸出的擴展模塊,目前處于初步開發的狀態，在日常的開發中,經過需求分析、建模之后,往往會先在資料庫中建表,其次在進行代碼的開發，

那么pojo生成功能在這個階段就可以幫助大家節省一些重復勞動了，使用pojo生成功能可以直接根據資料庫生成對應的java pojo物件，這樣后續的修改，開發都會很方便，

• 資料庫支持

[x] MySQL

• 使用方式

引入依賴

<dependency>
    <groupId>cn.smallbun.screw</groupId>
    <artifactId>screw-extension</artifactId>
    <version>${lastVersion}</version>
</dependency>

撰寫代碼

/**
 * pojo生成
 */
void pojoGeneration() {
    //資料源
    HikariConfig hikariConfig = new HikariConfig();
    hikariConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
    hikariConfig.setJdbcUrl("jdbc:mysql://127.0.0.1:3306/screw");
    hikariConfig.setUsername("screw");
    hikariConfig.setPassword("screw");
    //設定可以獲取tables remarks資訊
    hikariConfig.addDataSourceProperty("useInformationSchema", "true");
    hikariConfig.setMinimumIdle(2);
    hikariConfig.setMaximumPoolSize(5);
    DataSource dataSource = new HikariDataSource(hikariConfig);

    ProcessConfig processConfig = ProcessConfig.builder()
        //指定生成邏輯、當存在指定表、指定表前綴、指定表后綴時，將生成指定表，其余表不生成、并跳過忽略表配置
        //根據名稱指定表生成
        .designatedTableName(new ArrayList<>())
        //根據表前綴生成
        .designatedTablePrefix(new ArrayList<>())
        //根據表后綴生成
        .designatedTableSuffix(new ArrayList<>()).build();

    //設定生成pojo相關配置
    PojoConfiguration config = new PojoConfiguration();
    //設定檔案存放路徑
    config.setPath("/cn/smallbun/screw/");
    //設定包名
    config.setPackageName("cn.smallbun.screw");
    //設定是否使用lombok
    config.setUseLombok(false);
    //設定資料源
    config.setDataSource(dataSource);
    //設定命名策略
    config.setNameStrategy(new HumpNameStrategy());
    //設定表過濾邏輯
    config.setProcessConfig(processConfig);
    //執行生成
    new PojoExecute(config).execute();
}

8、常見問題

• 生成后檔案亂碼？

MySQL：URL加入?characterEncoding=UTF-8，

• Caused by: java.lang.NoSuchFieldError: VERSION_2_3_30？

檢查專案freemarker依賴，這是由于版本過低造成的，升級版本為2.3.30即可，

• java.lang.AbstractMethodError: oracle.jdbc.driver.T4CConnection.getSchema()Ljava/lang/String;

這是因為oracle驅動版本過低造成的，洗掉或屏蔽目前驅動版本，驅動添加升級為以下版本：

<dependency>
   <groupId>com.oracle.ojdbc</groupId>
   <artifactId>ojdbc8</artifactId>
   <version>19.3.0.0</version>
</dependency>
<dependency>
   <groupId>cn.easyproject</groupId>
   <artifactId>orai18n</artifactId>
   <version>12.1.0.2.0</version>
</dependency>

• MySQL資料庫表和列欄位有說明、生成檔案沒有說明？

URL鏈接加入useInformationSchema=true即可，

• java.lang.AbstractMethodError: com.mysql.jdbc.JDBC4Connection.getSchema()Ljava/lang/String;

這是因為mysql驅動版本過低造成的，升級mysql驅動版本為最新即可，

專案地址：https://gitee.com/leshalv/screw

近期熱文推薦：

1.1,000+ 道 Java面試題及答案整理(2022最新版)

2.勁爆！Java 協程要來了，，，

3.Spring Boot 2.x 教程，太全了！

4.別再寫滿屏的爆爆爆炸類了，試試裝飾器模式，這才是優雅的方式！！

5.《Java開發手冊（嵩山版）》最新發布，速速下載！

覺得不錯，別忘了隨手點贊+轉發哦！

標籤：Java

上一篇：Android C/C++層hook和java層hook原理以及比較

下一篇：java學習之反射機制