飞道的博客

数据库界的 Swagger,一键自动生成 Java 实体类和数据库文档!不讲武德!

508人阅读  评论(0)

    点击上方“芋道源码”,选择“设为星标

管她前浪,还是后浪?

能浪的浪,才是好浪!

每天 8:55 更新文章,每天掉亿点点头发...

源码精品专栏

 

摘要: 原创出处 http://www.iocoder.cn/Spring-Boot/DB-Doc-screw/ 「芋道源码」欢迎转载,保留摘要,谢谢!


1. 概述

闲来无事的周六,被 ???? 芳放了鸽子,只好蹲在被窝里研究会技术,安慰下自己受伤的心灵。

鸽子

相信胖友们都手写数据库表结构文档 ,一通无脑的 CV 大法之后,小几十页的  Word 文档就出炉了。

那么有没什么工具,可以帮我们偷懒高效的自动 生成数据库表结构文档呢?有,通过使用 screw 生成工具。

2. screw 简介

screw 是一个简洁好用的数据库表结构文档的生成工具 ,支持 MySQL、Oracle、PostgreSQL 等主流的关系数据库。

良心艿:screw 的仓库地址是 https://github.com/pingfangushi/screw,感兴趣的胖友,可以研究一波。

生成的文档有 HTML、Word、Markdown 三种格式 ,示例如下图所示:

格式
HTML
Word
Markdown

3. 快速入门

screw 有两种方式 来生成文档,通过 Java 代码 或者 Maven 插件

下面,我们来分别快速入门下。

3.1 使用 Java 代码的方式

创建 lab-70-db-doc-screw-01 示例项目,使用 screw 的 Java 代码 的方式,生成文档。

示例示例

友情提示:示例代码的完整地址,可见 https://github.com/YunaiV/SpringBoot-Labs/tree/master/lab-70-db-doc/lab-70-db-doc-screw-01。

3.1.1 引入依赖

pom.xml 文件中,引入 screw 的依赖 screw-core


   
  1. <?xml version= "1.0" encoding= "UTF-8"?>
  2. <project xmlns= "http://maven.apache.org/POM/4.0.0"
  3.          xmlns:xsi= "http://www.w3.org/2001/XMLSchema-instance"
  4.          xsi:schemaLocation= "http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  5.     <parent>
  6.         <artifactId>lab -70-db-doc</artifactId>
  7.         <groupId>cn.iocoder.springboot.labs</groupId>
  8.         <version> 1.0-SNAPSHOT</version>
  9.     </parent>
  10.     <modelVersion> 4.0 .0</modelVersion>
  11.     <artifactId>lab -70-db-doc-screw -01</artifactId>
  12.     <dependencies>
  13.         <!-- screw 库,简洁好用的数据库表结构文档生成器 -->
  14.         <dependency>
  15.             <groupId>cn.smallbun.screw</groupId>
  16.             <artifactId>screw-core</artifactId>
  17.             <version> 1.0 .5</version>
  18.         </dependency>
  19.         <!-- 数据库连接 -->
  20.         <dependency>
  21.             <groupId>com.zaxxer</groupId>
  22.             <artifactId>HikariCP</artifactId>
  23.             <version> 3.4 .5</version>
  24.         </dependency>
  25.         <dependency>
  26.             <groupId>mysql</groupId>
  27.             <artifactId>mysql-connector-java</artifactId>
  28.             <version> 8.0 .22</version>
  29.         </dependency>
  30.     </dependencies>
  31. </project>

依赖 HikariCPmysql-connector-java 是为了连接 MySQL 数据库。

3.1.2 ScrewMain

创建 ScrewMain 类,使用 screw 生成文档。代码如下:


   
  1. public class ScrewMain {
  2.     private static final String DB_URL =  "jdbc:mysql://400-infra.server.iocoder.cn:3306";
  3.     private static final String DB_NAME =  "mall_system";
  4.     private static final String DB_USERNAME =  "root";
  5.     private static final String DB_PASSWORD =  "3WLiVUBEwTbvAfsh";
  6.     private static final String FILE_OUTPUT_DIR =  "/Users/yunai/screw_test";
  7.     private static final EngineFileType FILE_OUTPUT_TYPE = EngineFileType.HTML;  // 可以设置 Word 或者 Markdown 格式
  8.     private static final String DOC_FILE_NAME =  "数据库文档";
  9.     private static final String DOC_VERSION =  "1.0.0";
  10.     private static final String DOC_DESCRIPTION =  "文档描述";
  11.     public static void main(String[] args) {
  12.          // 创建 screw 的配置
  13.         Configuration config = Configuration.builder()
  14.                 .version(DOC_VERSION)   // 版本
  15.                 .description(DOC_DESCRIPTION)  // 描述
  16.                 .dataSource(buildDataSource())  // 数据源
  17.                 .engineConfig(buildEngineConfig())  // 引擎配置
  18.                 .produceConfig(buildProcessConfig())  // 处理配置
  19.                 .build();
  20.          // 执行 screw,生成数据库文档
  21.          new DocumentationExecute(config).execute();
  22.     }
  23.      /**
  24.      * 创建数据源
  25.      */
  26.     private static DataSource buildDataSource() {
  27.          // 创建 HikariConfig 配置类
  28.         HikariConfig hikariConfig =  new HikariConfig();
  29.         hikariConfig.setDriverClassName( "com.mysql.cj.jdbc.Driver");
  30.         hikariConfig.setJdbcUrl(DB_URL +  "/" + DB_NAME);
  31.         hikariConfig.setUsername(DB_USERNAME);
  32.         hikariConfig.setPassword(DB_PASSWORD);
  33.         hikariConfig.addDataSourceProperty( "useInformationSchema""true");  // 设置可以获取 tables remarks 信息
  34.          // 创建数据源
  35.          return  new HikariDataSource(hikariConfig);
  36.     }
  37.      /**
  38.      * 创建 screw 的引擎配置
  39.      */
  40.     private static EngineConfig buildEngineConfig() {
  41.          return EngineConfig.builder()
  42.                 .fileOutputDir(FILE_OUTPUT_DIR)  // 生成文件路径
  43.                 .openOutputDir( false// 打开目录
  44.                 .fileType(FILE_OUTPUT_TYPE)  // 文件类型
  45.                 .produceType(EngineTemplateType.freemarker)  // 文件类型
  46.                 .fileName(DOC_FILE_NAME)  // 自定义文件名称
  47.                 .build();
  48.     }
  49.      /**
  50.      * 创建 screw 的处理配置,一般可忽略
  51.      * 指定生成逻辑、当存在指定表、指定表前缀、指定表后缀时,将生成指定表,其余表不生成、并跳过忽略表配置
  52.      */
  53.     private static ProcessConfig buildProcessConfig() {
  54.          return ProcessConfig.builder()
  55.                 .designatedTableName(Collections.<String>emptyList())   // 根据名称指定表生成
  56.                 .designatedTablePrefix(Collections.<String>emptyList())  //根据表前缀生成
  57.                 .designatedTableSuffix(Collections.<String>emptyList())  // 根据表后缀生成
  58.                 .ignoreTableName(Arrays.asList( "test_user""test_group"))  // 忽略表名
  59.                 .ignoreTablePrefix(Collections.singletonList( "test_"))  // 忽略表前缀
  60.                 .ignoreTableSuffix(Collections.singletonList( "_test"))  // 忽略表后缀
  61.                 .build();
  62.     }
  63. }

代码比较简单,胖友看看注释即可。

比较重要的变量,已经抽取成静态 变量,胖友看着修改哈。

3.1.3 简单测试

执行 ScrewMain 程序,进行文档的生成。这里我们生成的是 HTML 文档,效果如下图所示:

HTML 文档

3.2 使用 Maven 插件的方式

创建 lab-70-db-doc-screw-02 示例项目,使用 screw 的 Maven 插件 的方式,生成文档。

示例示例

友情提示:示例代码的完整地址,可见 https://github.com/YunaiV/SpringBoot-Labs/tree/master/lab-70-db-doc/lab-70-db-doc-screw-02。

3.2.1 引入插件

pom.xml 文件中,引入 screw 的插件 screw-maven-plugin


   
  1. <?xml version= "1.0" encoding= "UTF-8"?>
  2. <project xmlns= "http://maven.apache.org/POM/4.0.0"
  3.          xmlns:xsi= "http://www.w3.org/2001/XMLSchema-instance"
  4.          xsi:schemaLocation= "http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  5.     <parent>
  6.         <artifactId>lab -70-db-doc</artifactId>
  7.         <groupId>cn.iocoder.springboot.labs</groupId>
  8.         <version> 1.0-SNAPSHOT</version>
  9.     </parent>
  10.     <modelVersion> 4.0 .0</modelVersion>
  11.     <artifactId>lab -70-db-doc-screw -02</artifactId>
  12.     <build>
  13.         <plugins>
  14.             <plugin>
  15.                 <groupId>cn.smallbun.screw</groupId>
  16.                 <artifactId>screw-maven-plugin</artifactId>
  17.                 <version> 1.0 .5</version>
  18.                 <dependencies>
  19.                     <!-- 数据库连接 -->
  20.                     <dependency>
  21.                         <groupId>com.zaxxer</groupId>
  22.                         <artifactId>HikariCP</artifactId>
  23.                         <version> 3.4 .5</version>
  24.                     </dependency>
  25.                     <dependency>
  26.                         <groupId>mysql</groupId>
  27.                         <artifactId>mysql-connector-java</artifactId>
  28.                         <version> 8.0 .22</version>
  29.                     </dependency>
  30.                 </dependencies>
  31.                 <configuration>
  32.                     <!-- 数据库相关配置 -->
  33.                     <driverClassName>com.mysql.cj.jdbc.Driver</driverClassName>
  34.                     <jdbcUrl>jdbc:mysql: //400-infra.server.iocoder.cn:3306/mall_system</jdbcUrl>
  35.                     <username>root</username>
  36.                     <password> 3WLiVUBEwTbvAfsh</password>
  37.                     <!-- screw 配置 -->
  38.                     <fileType>HTML</fileType>
  39.                     <title>数据库文档</title> <!--标题-->
  40.                     <fileName>测试文档名称</fileName> <!--文档名称 为空时:将采用[数据库名称-描述-版本号]作为文档名称-->
  41.                     <description>数据库文档生成</description> <!--描述-->
  42.                     <version>${project.version}</version> <!--版本-->
  43.                     <openOutputDir> false</openOutputDir> <!--打开文件输出目录-->
  44.                     <produceType>freemarker</produceType> <!--生成模板-->
  45.                 </configuration>
  46.                 <executions>
  47.                     <execution>
  48.                         <phase>compile</phase>
  49.                         <goals>
  50.                             <goal>run</goal>
  51.                         </goals>
  52.                     </execution>
  53.                 </executions>
  54.             </plugin>
  55.         </plugins>
  56.     </build>
  57. </project>

Maven 插件的配置项 比较少,胖友按需修改下 <configuration/> 即可。

3.2.2 简单测试

执行 screw-maven-plugin 插件,会在 doc 目录下生成文档。如下图所示:

执行 Maven 插件

4. 生成实体类

screw 的实现原理,是基于数据库中的表结构 ,生成对应的文档。

那么,是否可以自动 生成 Java 实体类呢?答案是可以的,在 screw-extension 项目中,拓展 提供了该功能。

这样,日常开发中,在我们完成数据库的建表 之后,可以直接生成对应的 Java 实体类,避免枯燥的重复劳动。

下面,我们来快速入门下。创建 lab-70-db-doc-screw-03 示例项目,使用 screw 的 Java 代码 的方式,生成 Java 实体类。

示例示例

友情提示:示例代码的完整地址,可见 https://github.com/YunaiV/SpringBoot-Labs/tree/master/lab-70-db-doc/lab-70-db-doc-screw-03。

4.1 引入依赖

pom.xml 文件中,额外 引入 screw 的依赖 screw-extension


   
  1. <?xml version= "1.0" encoding= "UTF-8"?>
  2. <project xmlns= "http://maven.apache.org/POM/4.0.0"
  3.          xmlns:xsi= "http://www.w3.org/2001/XMLSchema-instance"
  4.          xsi:schemaLocation= "http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  5.     <parent>
  6.         <artifactId>lab -70-db-doc</artifactId>
  7.         <groupId>cn.iocoder.springboot.labs</groupId>
  8.         <version> 1.0-SNAPSHOT</version>
  9.     </parent>
  10.     <modelVersion> 4.0 .0</modelVersion>
  11.     <artifactId>lab -70-db-doc-screw -03</artifactId>
  12.     <dependencies>
  13.         <!-- screw 库,简洁好用的数据库表结构文档生成器 -->
  14.         <dependency>
  15.             <groupId>cn.smallbun.screw</groupId>
  16.             <artifactId>screw-core</artifactId>
  17.             <version> 1.0 .5</version>
  18.         </dependency>
  19.         <dependency>
  20.             <groupId>cn.smallbun.screw</groupId>
  21.             <artifactId>screw-extension</artifactId>
  22.             <version> 1.0 .5</version>
  23.         </dependency>
  24.         <!-- 数据库连接 -->
  25.         <dependency>
  26.             <groupId>com.zaxxer</groupId>
  27.             <artifactId>HikariCP</artifactId>
  28.             <version> 3.4 .5</version>
  29.         </dependency>
  30.         <dependency>
  31.             <groupId>mysql</groupId>
  32.             <artifactId>mysql-connector-java</artifactId>
  33.             <version> 8.0 .22</version>
  34.         </dependency>
  35.     </dependencies>
  36. </project>

友情提示:由于 screw-extension 项目处于初步开发阶段,暂时未将该依赖推到 Maven 中央仓库,所以需要自己克隆项目,进行编译打包到本地。

4.2 ScrewMain

创建 ScrewMain 类,使用 screw 生成 Java 实体类。代码如下:


   
  1. public class ScrewMain {
  2.     private static final String DB_URL =  "jdbc:mysql://400-infra.server.iocoder.cn:3306";
  3.     private static final String DB_NAME =  "mall_system";
  4.     private static final String DB_USERNAME =  "root";
  5.     private static final String DB_PASSWORD =  "3WLiVUBEwTbvAfsh";
  6.     private static final String FILE_OUTPUT_DIR =  "/Users/yunai/screw_test";
  7.     private static final String JAVA_CLASS_PACKAGE =  "cn.iocoder.dataobject";
  8.     public static void main(String[] args) {
  9.          // 创建 screw 的配置
  10.         PojoConfiguration config = PojoConfiguration.builder()
  11.                 .path(FILE_OUTPUT_DIR)  // 生成 POJO 相关的目录
  12.                 .packageName(JAVA_CLASS_PACKAGE)  // 包名
  13.                 .nameStrategy( new HumpNameStrategy())  // 包名策略
  14.                 .useLombok( false// 是否使用 Lombok
  15.                 .dataSource(buildDataSource())  // 数据源
  16.                 .processConfig(buildProcessConfig())  // 处理配置
  17.                 .build();
  18.          // 执行 screw,生成 POJO 实体类
  19.          new PojoExecute(config).execute();
  20.     }
  21.      /**
  22.      * 创建数据源
  23.      */
  24.     private static DataSource buildDataSource() {
  25.          // 创建 HikariConfig 配置类
  26.         HikariConfig hikariConfig =  new HikariConfig();
  27.         hikariConfig.setDriverClassName( "com.mysql.cj.jdbc.Driver");
  28.         hikariConfig.setJdbcUrl(DB_URL +  "/" + DB_NAME);
  29.         hikariConfig.setUsername(DB_USERNAME);
  30.         hikariConfig.setPassword(DB_PASSWORD);
  31.         hikariConfig.addDataSourceProperty( "useInformationSchema""true");  // 设置可以获取 tables remarks 信息
  32.          // 创建数据源
  33.          return  new HikariDataSource(hikariConfig);
  34.     }
  35.      /**
  36.      * 创建 screw 的处理配置,一般可忽略
  37.      * 指定生成逻辑、当存在指定表、指定表前缀、指定表后缀时,将生成指定表,其余表不生成、并跳过忽略表配置
  38.      */
  39.     private static ProcessConfig buildProcessConfig() {
  40.          return ProcessConfig.builder()
  41.                 .designatedTableName(Collections.<String>emptyList())   // 根据名称指定表生成
  42.                 .designatedTablePrefix(Collections.<String>emptyList())  //根据表前缀生成
  43.                 .designatedTableSuffix(Collections.<String>emptyList())  // 根据表后缀生成
  44.                 .ignoreTableName(Arrays.asList( "test_user""test_group"))  // 忽略表名
  45.                 .ignoreTablePrefix(Collections.singletonList( "test_"))  // 忽略表前缀
  46.                 .ignoreTableSuffix(Collections.singletonList( "_test"))  // 忽略表后缀
  47.                 .build();
  48.     }
  49. }

代码比较简单,胖友看看注释即可。不同于上面的快速入门,这里我们使用 PojoConfiguration 作为配置类,使用 PojoExecute 作为执行器。

比较重要的变量,已经抽取成静态 变量,胖友看着修改哈。

4.3 简单测试

执行 ScrewMain 程序,进行 Java 实体类的生成。效果如下图所示:

Java 实体类

生成的 Java 实体类的成员属性 还不太正确,需要等待作者进行下修复。

666. 彩蛋

至此,我们已经完成 screw 的学习,一起来简单总结下:

  • screw 支持生成数据库表结构文档 ,通过 Java 代码或者 Maven 插件的方式。

  • screw 支持生成 Java 实体类 ,通过 screw-extension 提供。

初略喵了下 screw 的代码,总体代码量在 5000+ 行,核心代码在 2000 行左右。项目分层干净,代码注释完成,胖友可以选择阅读了解下。这样,我们可以根据我们实际项目的需要,进行二次开发


End~继续抠脚。

我是艿艿,一个每天徘徊在煞笔牛啤 的死胖子。



欢迎加入我的知识星球,一起探讨架构,交流源码。加入方式,长按下方二维码噢

已在知识星球更新源码解析如下:

最近更新《芋道 SpringBoot 2.X 入门》系列,已经 20 余篇,覆盖了 MyBatis、Redis、MongoDB、ES、分库分表、读写分离、SpringMVC、Webflux、权限、WebSocket、Dubbo、RabbitMQ、RocketMQ、Kafka、性能测试等等内容。

提供近 3W 行代码的 SpringBoot 示例,以及超 4W 行代码的电商微服务项目。

获取方式:点“在看”,关注公众号并回复 666 领取,更多内容陆续奉上。

兄弟,一口,点个!????


转载:https://blog.csdn.net/github_38592071/article/details/111351136
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场