模仿Activiti工作流自动建表机制，实现Springboot项目启动后自动创建多表关联的数据库与表的方案

文/朱季谦

熬夜写完，尚有不足，但仍在努力学习与总结中，而您的点赞与关注，是对我最大的鼓励！

在一些本地化项目开发当中，存在这样一种需求，即开发完成的项目，在第一次部署启动时，需能自行构建系统需要的数据库及其对应的数据库表。

若要解决这类需求，其实现在已有不少开源框架都能实现自动生成数据库表，如mybatis plus、spring JPA等，但您是否有想过，若要自行构建一套更为复杂的表结构时，这种开源框架是否也能满足呢，若满足不了话，又该如何才能实现呢？

我在前面写过一篇 Activiti工作流学习笔记(三)——自动生成28张数据库表的底层原理分析，里面分析过工作流Activiti自动构建28数据库表的底层原理。在我看来，学习开源框架的底层原理，其中一个原因是，须从中学到能为我所用的东西。故而，在分析理解完工作流自动构建28数据库表的底层原理之后，我决定也写一个基于Springboot框架的自行创建数据库与表的demo。我参考了工作流Activiti6.0版本的底层建表实现的逻辑，基于Springboot框架，实现项目在第一次启动时可自动构建各种复杂如多表关联等形式的数据库与表的。

整体实现思路并不复杂，大概是这样：先设计一套完整创建多表关联的数据库sql脚本，放到resource里，在springboot启动过程中，自动执行sql脚本。

首先，先一次性设计一套可行的多表关联数据库脚本，这里我主要参考使用Activiti自带的表做实现案例，因为它内部设计了众多表关联，就不额外设计了。

sql脚本的语句就是平常的create建表语句，类似如下：

  1 create table ACT_PROCDEF_INFO (2    ID_ varchar(64) not null,3     PROC_DEF_ID_ varchar(64) not null,4     REV_ integer,5     INFO_JSON_ID_ varchar(64),6     primary key (ID_)7 ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE utf8_bin;

增加外部主键、索引——

  1 create index ACT_IDX_INFO_PROCDEF on ACT_PROCDEF_INFO(PROC_DEF_ID_);2 3 alter table ACT_PROCDEF_INFO4     add constraint ACT_FK_INFO_JSON_BA5     foreign key (INFO_JSON_ID_)6     references ACT_GE_BYTEARRAY (ID_);7 8 alter table ACT_PROCDEF_INFO9     add constraint ACT_FK_INFO_PROCDEF10     foreign key (PROC_DEF_ID_)11     references ACT_RE_PROCDEF (ID_);12 13 alter table ACT_PROCDEF_INFO14     add constraint ACT_UNIQ_INFO_PROCDEF15     unique (PROC_DEF_ID_);

整体就是设计一套符合符合需求场景的sql语句，保存在.sql的脚本文件里，最后统一存放在resource目录下，类似如下：

image-20210315132805036

接下来，就是实现CommandLineRunner的接口，重写其run()的bean回调方法，在run方法里开发能自动建库与建表逻辑的功能。

目前，我已将开发的demo上传到了我的github，感兴趣的童鞋，可自行下载，目前能直接下下来在本地环境运行，可根据自己的实际需求针对性参考使用。

首先，在解决这类需求时，第一个先要解决的地方是，Springboot启动后如何实现只执行一次建表方法。

这里需要用到一个CommandLineRunner接口，这是Springboot自带的，实现该接口的类，其重写的run方法，会在Springboot启动完成后自动执行，该接口源码如下：

  1 @FunctionalInterface2 public interface CommandLineRunner {3 4    /**5     *用于运行bean的回调6     */7    void run(String... args) throws Exception;8 9 }

扩展一下，在Springboot中，可以定义多个实现CommandLineRunner接口类，并且可以对这些实现类中进行排序，只需要增加@Order，其重写的run方法就可以按照顺序执行，代码案例验证：

  1 @Component2 @Order(value=1)3 public class WatchStartCommandSqlRunnerImpl implements CommandLineRunner {4 5     @Override6     public void run(String... args) throws Exception {7         System.out.println("第一个Command执行");8     }9 10 11 @Component12 @Order(value = 2)13 public class WatchStartCommandSqlRunnerImpl2 implements CommandLineRunner {14     @Override15     public void run(String... args) throws Exception {16         System.out.println("第二个Command执行");17     }18 }19

控制台打印的信息如下：

  1 第一个Command执行2 第二个Command执行

根据以上的验证，因此，我们可以通过实现CommandLineRunner的接口，重写其run()的bean回调方法，用于在Springboot启动后实现只执行一次建表方法。实现项目启动建表的功能，可能还需实现判断是否已经有相应数据库，若无，则应先新建一个数据库，同时，得考虑还没有对应数据库的情况，因此，我们通过jdbc第一次连接MySQL时，应连接一个原有自带存在的库。每个MySql安装成功后，都会有一个mysql库，在第一次建立jdbc连接时，可以先连接它。

image-20210315080736373

代码如下：

Class.forName("com.mysql.jdbc.Driver");
String url="jdbc:mysql://127.0.0.1:3306/mysql?useUnicode=true&characterEncoding=UTF-8&ueSSL=false&serverTimezone=GMT%2B8";
Connection conn= DriverManager.getConnection(url,"root","root");

建立与MySql软件连接后，先创建一个Statement对象，该对象是jdbc中可用于执行静态 SQL 语句并返回它所生成结果的对象，这里可以使用它来执行查找库与创建库的作用。

  1  //创建Statement对象2  Statement statment=conn.createStatement();3  /**4  使用statment的查询方法executeQuery("show databases like \"fte\"")5  检查MySql是否有fte这个数据库6  **/7  ResultSet resultSet=statment.executeQuery("show databases like \"fte\"");8  //若resultSet.next()为true，证明已存在；9  //若false，证明还没有该库,则执行statment.executeUpdate("create database fte")创建库10  if(resultSet.next()){11      log.info("数据库已经存在");12   }else {13   log.info("数据库未存在，先创建fte数据库");14   if(statment.executeUpdate("create database fte")==1){15      log.info("新建数据库成功");16      }17    }

在数据库fte自动创建完成后，就可以在该fte库里去做建表的操作了。

我将建表的相关方法都封装到SqlSessionFactory类里，相关建表方法同样需要用到jdbc的Connection连接到数据库，因此，需要把已连接的Connection引用变量当做参数传给SqlSessionFactory的初始构造函数：

  1    public void createTable(Connection conn,Statement stat) throws SQLException {2         try {3 4             String url="jdbc:mysql://127.0.0.1:3306/fte?useUnicode=true&characterEncoding=UTF-8&ueSSL=false&serverTimezone=GMT%2B8";5             conn=DriverManager.getConnection(url,"root","root");6             SqlSessionFactory sqlSessionFactory=new SqlSessionFactory(conn);7             sqlSessionFactory.schemaOperationsBuild("create");8         } catch (SQLException e) {9             e.printStackTrace();10         }finally {11             stat.close();12             conn.close();13         }14     }

初始化new SqlSessionFactory(conn)后，就可以在该对象里使用已进行连接操作的Connection对象了。

  1 public class SqlSessionFactory{2     private Connection connection ;3     public SqlSessionFactory(Connection connection) {4         this.connection = connection;5     }6 ......7 }

这里传参可以有两种情况，即“create”代表创建表结构的功能，“drop”代表删除表结构的功能：

  1 sqlSessionFactory.schemaOperationsBuild("create");

进入到这个方法里，会先做一个判断——

  1 public void schemaOperationsBuild(String type) {2     switch (type){3         case "drop":4             this.dbSchemaDrop();break;5         case "create":6             this.dbSchemaCreate();break;7     }8 }

若是this.dbSchemaCreate()，执行建表操作：

  1 /**2  * 新增数据库表3  */4 public void dbSchemaCreate() {5 6     if (!this.isTablePresent()) {7         log.info("开始执行create操作");8         this.executeResource("create", "act");9         log.info("执行create完成");10     }11 }

this.executeResource("create", "act")代表创建表名为act的数据库表——

  1 public void executeResource(String operation, String component) {2     this.executeSchemaResource(operation, component, this.getDbResource(operation, operation, component), false);3 }

其中 this.getDbResource(operation, operation, component)是获取sql脚本的路径，进入到方法里，可见——

  1 public String getDbResource(String directory, String operation, String component) {2     return "static/db/" + directory + "/mysql." + operation + "." + component + ".sql";3 }

接下来，读取路径下的sql脚本，生成输入流字节流：

  1 public void executeSchemaResource(String operation, String component, String resourceName, boolean isOptional) {2     InputStream inputStream = null;3 4     try {5         //读取sql脚本数据6         inputStream = IoUtil.getResourceAsStream(resourceName);7         if (inputStream == null) {8             if (!isOptional) {9                 log.error("resource '" + resourceName + "' is not available");10                 return;11             }12         } else {13             this.executeSchemaResource(operation, component, resourceName, inputStream);14         }15     } finally {16         IoUtil.closeSilently(inputStream);17     }18 19 }

最后，整个执行sql脚本的核心实现在this.executeSchemaResource(operation, component, resourceName, inputStream)方法里——

  1 /**2  * 执行sql脚本3  * @param operation4  * @param component5  * @param resourceName6  * @param inputStream7  */8 private void executeSchemaResource(String operation, String component, String resourceName, InputStream inputStream) {9     //sql语句拼接字符串10     String sqlStatement = null;11     Object exceptionSqlStatement = null;12 13     try {14         /**15          * 1.jdbc连接mysql数据库16          */17         Connection connection = this.connection;18 19         Exception exception = null;20         /**21          * 2、分行读取"static/db/create/mysql.create.act.sql"里的sql脚本数据22          */23         byte[] bytes = IoUtil.readInputStream(inputStream, resourceName);24         /**25          * 3.将sql文件里数据分行转换成字符串，换行的地方，用转义符“\n”来代替26          */27         String ddlStatements = new String(bytes);28         /**29          * 4.以字符流形式读取字符串数据30          */31         BufferedReader reader = new BufferedReader(new StringReader(ddlStatements));32         /**33          * 5.根据字符串中的转义符“\n”分行读取34          */35         String line = IoUtil.readNextTrimmedLine(reader);36         /**37          * 6.循环读取的每一行38          */39         for(boolean inOraclePlsqlBlock = false; line != null; line = IoUtil.readNextTrimmedLine(reader)) {40             /**41              * 7.若下一行line还有数据，证明还没有全部读取，仍可执行读取42              */43             if (line.length() > 0) {44                 /**45                  8.在没有拼接够一个完整建表语句时，！line.endsWith(";")会为true，46                  即一直循环进行拼接，当遇到";"就跳出该if语句47                 **/48                if ((!line.endsWith(";") || inOraclePlsqlBlock) && (!line.startsWith("/") || !inOraclePlsqlBlock)) {49                     sqlStatement = this.addSqlStatementPiece(sqlStatement, line);50                 } else {51                    /**52                     9.循环拼接中若遇到符号";"，就意味着，已经拼接形成一个完整的sql建表语句，例如53                     create table ACT_GE_PROPERTY (54                     NAME_ varchar(64),55                     VALUE_ varchar(300),56                     REV_ integer,57                     primary key (NAME_)58                     ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE utf8_bin59                     这样，就可以先通过代码来将该建表语句执行到数据库中，实现如下：60                     **/61                     if (inOraclePlsqlBlock) {62                         inOraclePlsqlBlock = false;63                     } else {64                         sqlStatement = this.addSqlStatementPiece(sqlStatement, line.substring(0, line.length() - 1));65                     }66                    /**67                     * 10.将建表语句字符串包装成Statement对象68                     */69                     Statement jdbcStatement = connection.createStatement();70 71                     try {72                         /**73                          * 11.最后，执行建表语句到数据库中74                          */75                         log.info("SQL: {}", sqlStatement);76                         jdbcStatement.execute(sqlStatement);77                         jdbcStatement.close();78                     } catch (Exception var27) {79                         log.error("problem during schema {}, statement {}", new Object[]{operation, sqlStatement, var27});80                     } finally {81                         /**82                          * 12.到这一步，意味着上一条sql建表语句已经执行结束，83                          * 若没有出现错误话，这时已经证明第一个数据库表结构已经创建完成，84                          * 可以开始拼接下一条建表语句，85                          */86                         sqlStatement = null;87                     }88                 }89             }90         }91 92         if (exception != null) {93             throw exception;94         } 97     } catch (Exception var29) {98         log.error("couldn't " + operation + " db schema: " + exceptionSqlStatement, var29);99     }
100 }

复制代码

这部分代码主要功能是，先用字节流形式读取sql脚本里的数据，转换成字符串，其中有换行的地方用转义符“/n”来代替。接着把字符串转换成字符流BufferedReader形式读取，按照“/n”符合来划分每一行的读取，循环将读取的每行字符串进行拼接，当循环到某一行遇到“；”时，就意味着已经拼接成一个完整的create建表语句，类似这样形式——

  1 create table ACT_PROCDEF_INFO (2    ID_ varchar(64) not null,3     PROC_DEF_ID_ varchar(64) not null,4     REV_ integer,5     INFO_JSON_ID_ varchar(64),6     primary key (ID_)7 ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE utf8_bin;

这时，就可以先将拼接好的create建表字符串，通过 jdbcStatement.execute(sqlStatement)语句来执行入库了。当执行成功时，该ACT_PROCDEF_INFO表就意味着已经创建成功，接着以BufferedReader字符流形式继续读取下一行，进行下一个数据库表结构的构建。

整个过程大概就是这个逻辑，可以在此基础上，针对更为复杂的建表结构sql语句进行设计，在项目启动时，自行执行相应的sql语句，来进行建表。

该demo代码已经上传git，可直接下载运行：GitHub - z924931408/Springboot-AutoCreateMySqlTable: 模仿工作流引擎Activity自动建表机制实现Springboot在启动时自动生成数据库与表demo

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/336828.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

Qt之有趣的数字钟

Qt之有趣的数字钟

一.效果 基于网络代码修改，支持时、分、秒；支持滑动、翻页和旋转。 二.实现 #include <QtCore> #include <QPainter> #include <QAction> #include <QWidget> #include <QMainWindow> #include <QTimer> #include <QKeyEvent> #…
阅读更多...
【一竞技DOTA2】MinD_ContRoL加盟Tundra

【一竞技DOTA2】MinD_ContRoL加盟Tundra

1、近日Tundra战队正式官宣MinD_ContRoL加入队伍。他此前效力于Nigma战队,在离队不久后他便做出了加入Tundra战队的选择,借此他也成为了Tundra Esports战队已官宣了的阵容中的第四名选手。 2、Bleed战队正式官宣Mikoto和poloson加入战队。 Mikoto曾效力于东南亚赛区的Talon战队…
阅读更多...
深度解析高防产品---高防CDN

深度解析高防产品---高防CDN

高防CDN是一种基于云计算技术的网络安全防御系统&#xff0c;通过在全球范围内部署多个节点&#xff0c;实现对网站内容的加速和保护。其主要作用和功能包括安全防护、加速访问、跨运营商、跨地域的全网覆盖、提高网站的稳定性以及节约成本。高防CDN可以有效地解决不同地区的网…
阅读更多...
阿里云实时计算企业级状态存储引擎 Gemini 技术解读

阿里云实时计算企业级状态存储引擎 Gemini 技术解读

本文整理自阿里云 Flink 存储引擎团队李晋忠&#xff0c;兰兆千&#xff0c;梅源关于阿里云实时计算企业级状态存储引擎 Gemini 的研究&#xff0c;内容主要分为以下五部分&#xff1a; 流计算状态访问的痛点企业级状态存储引擎GeminiGemini 性能评测&线上表现结语参考 一、…
阅读更多...
AArch64 Exception Model学习

AArch64 Exception Model学习

提示 该博客主要为个人学习&#xff0c;通过阅读官网手册整理而来&#xff08;个人觉得阅读官网的英文文档非常有助于理解各个IP特性&#xff09;。若有不对之处请参考参考文档&#xff0c;以官网文档为准。 1 Privilege and Exception Levels 1.1 为什么要划分权限&#xf…
阅读更多...
# 大模型实战作业02

# 大模型实战作业02

大模型实战作业02 知识库助手搭建 注 有些问题在回答的时候可能出现乱码的情况 可能的原因 模型内部提示词知识库 这部分可以做适当的优化&#xff0c;对于无法回答或回答质量不佳的回答返回特定话术提升用户体验
阅读更多...
(一)Spring Cloud 直击微服务作用、架构应用、hystrix降级

(一)Spring Cloud 直击微服务作用、架构应用、hystrix降级

直击微服务作用 微服务架构: 遇到了什么问题? 将单体架构拆分成微服务架构后,如果保证多个服务(项目)正常运行? 哪个技术可以解决这个问题? 微服务技术 服务治理: 服务管理,维护服务与服务之间的关系 这个技术如何使用? netflix/网…
阅读更多...
c语言：输入成绩，统计不及格人数｜练习题

c语言：输入成绩，统计不及格人数｜练习题

一、题目 输入学生成绩&#xff0c;统计不及格的学生人数 二、代码截图【带注释】 三、源代码【带注释】 #include <stdio.h> //题目&#xff1a;输入成绩&#xff0c;统计不及格人数 //思考分析 //1、由于学生人数是未知数&#xff0c;所以可以在输入时&#xff0c;以0…
阅读更多...
vue2源码解析之第一步（对数据进行劫持）

vue2源码解析之第一步（对数据进行劫持）

###环境搭建 第一步 创建项目&#xff1a; npm init -y 第二步 安装对应的插件&#xff1a; npm i rollup rollup-plugin-babel babel/core babel/preset-env --save-dev 第三步 全局下创建rollup配置文件 rollup.config.js import babel from rollup-plug…
阅读更多...
【开源】基于JAVA、微信小程序的音乐平台

【开源】基于JAVA、微信小程序的音乐平台

目录 一、摘要1.1 项目介绍1.2 项目录屏 二、功能模块三、系统展示 四、核心代码4.1 查询单首音乐4.2 新增音乐4.3 新增音乐订单4.4 查询音乐订单4.5 新增音乐收藏 五、免责说明 一、摘要 1.1 项目介绍 基于微信小程序JAVAVueSpringBootMySQL的音乐平台&#xff0c;包含了音乐…
阅读更多...
用通俗易懂的方式讲解：大模型 RAG 在 LangChain 中的应用实战

用通俗易懂的方式讲解：大模型 RAG 在 LangChain 中的应用实战

Retrieval-Augmented Generation&#xff08;RAG&#xff09;是一种强大的技术&#xff0c;能够提高大型语言模型&#xff08;LLM&#xff09;的性能&#xff0c;使其能够从外部知识源中检索信息以生成更准确、具有上下文的回答。 本文将详细介绍 RAG 在 LangChain 中的应用&a…
阅读更多...
外汇天眼：

外汇天眼：

随着年关将近&#xff0c;各种电信诈骗也层出不穷&#xff0c;令人防不胜防。 上个月&#xff0c;台北市警方就接到辖区内银行的诈骗通报&#xff0c;得知竟有民众被骗走新台币1亿元以上&#xff0c;这究竟是怎么回事呢&#xff1f; 根据警方掌握到的情报&#xff0c;受害者是…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023