From 637a9d1ace167e82fece861b948d092852255939 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=BE=E6=96=87=E8=B1=AA?= <980287353@qq.com>
Date: Wed, 22 Feb 2023 00:24:26 +0800
Subject: [PATCH] =?UTF-8?q?perf=EF=BC=9A=E5=A2=9E=E5=8A=A0=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E5=90=8C=E6=AD=A5=E5=90=8E=E7=9A=84=E6=89=A7?=
 =?UTF-8?q?=E8=A1=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../database/service/DbDataInitializer.java      | 16 ++++++++++++++++
 .../database/service/DbMigrationInitializer.java |  3 +++
 2 files changed, 19 insertions(+)
 create mode 100644 springboot-database/src/main/java/com/tiesheng/database/service/DbDataInitializer.java

diff --git a/springboot-database/src/main/java/com/tiesheng/database/service/DbDataInitializer.java b/springboot-database/src/main/java/com/tiesheng/database/service/DbDataInitializer.java
new file mode 100644
index 0000000..d9bdbb4
--- /dev/null
+++ b/springboot-database/src/main/java/com/tiesheng/database/service/DbDataInitializer.java
@@ -0,0 +1,16 @@
+package com.tiesheng.database.service;
+
+import org.springframework.stereotype.Component;
+
+/**
+ * @author hao
+ */
+@Component
+public interface DbDataInitializer {
+
+    /**
+     * 初始化数据
+     */
+    void init();
+
+}
diff --git a/springboot-database/src/main/java/com/tiesheng/database/service/DbMigrationInitializer.java b/springboot-database/src/main/java/com/tiesheng/database/service/DbMigrationInitializer.java
index ab97a02..694184b 100644
--- a/springboot-database/src/main/java/com/tiesheng/database/service/DbMigrationInitializer.java
+++ b/springboot-database/src/main/java/com/tiesheng/database/service/DbMigrationInitializer.java
@@ -35,12 +35,15 @@ public class DbMigrationInitializer implements ServletContextInitializer {
     DataSource dataSource;
     @Autowired
     DbMigrationConfig dbMigrationConfig;
+    @Autowired
+    List<DbDataInitializer> dbDataInitializerList;
 
 
     @Override
     public void onStartup(ServletContext servletContext) {
         try {
             startDeal();
+            dbDataInitializerList.forEach(DbDataInitializer::init);
         } catch (Exception ignore) {
         }
     }