From a2fa8b3eeb1112961cf632f87da89d2f6f15f989 Mon Sep 17 00:00:00 2001
From: zhaojun <873019219@qq.com>
Date: Mon, 29 Aug 2022 14:38:47 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E5=A4=8D=E6=B2=A1=E6=9C=89?=
 =?UTF-8?q?=E8=87=AA=E5=8A=A8=E5=88=9B=E5=BB=BA=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E7=9B=AE=E5=BD=95=E7=9A=84=20bug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../im/zhaojun/zfile/ZfileApplication.java    | 22 ---------------
 .../common/config/MyBatisPlusConfig.java      | 27 +++++++++++++++++--
 2 files changed, 25 insertions(+), 24 deletions(-)

diff --git a/src/main/java/im/zhaojun/zfile/ZfileApplication.java b/src/main/java/im/zhaojun/zfile/ZfileApplication.java
index 679d65e..0f3cacf 100644
--- a/src/main/java/im/zhaojun/zfile/ZfileApplication.java
+++ b/src/main/java/im/zhaojun/zfile/ZfileApplication.java
@@ -1,16 +1,11 @@
 package im.zhaojun.zfile;
 
-import cn.hutool.core.io.FileUtil;
-import cn.hutool.core.util.StrUtil;
-import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.boot.web.servlet.ServletComponentScan;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.context.annotation.EnableAspectJAutoProxy;
 
-import javax.annotation.PostConstruct;
-
 @SpringBootApplication
 @EnableAspectJAutoProxy(exposeProxy = true, proxyTargetClass = true)
 @ServletComponentScan(basePackages = "im.zhaojun.zfile.*.filter")
@@ -22,22 +17,5 @@ public class ZfileApplication {
         SpringApplication.run(ZfileApplication.class, args);
     }
 
-    @Value("${spring.datasource.driver-class-name}")
-    private String datasourceDriveClassName;
-
-    @Value("${spring.datasource.url}")
-    private String datasourceUrl;
-
-
-    @PostConstruct
-    public void init() {
-        if (StrUtil.equals(datasourceDriveClassName, "org.sqlite.JDBC")) {
-            String path = datasourceUrl.replace("jdbc:sqlite:", "");
-            String folderPath = FileUtil.getParent(path, 1);
-            if (!FileUtil.exist(folderPath)) {
-                FileUtil.mkdir(folderPath);
-            }
-        }
-    }
 
 }
\ No newline at end of file
diff --git a/src/main/java/im/zhaojun/zfile/common/config/MyBatisPlusConfig.java b/src/main/java/im/zhaojun/zfile/common/config/MyBatisPlusConfig.java
index 22c71e6..ef225e3 100644
--- a/src/main/java/im/zhaojun/zfile/common/config/MyBatisPlusConfig.java
+++ b/src/main/java/im/zhaojun/zfile/common/config/MyBatisPlusConfig.java
@@ -1,11 +1,15 @@
 package im.zhaojun.zfile.common.config;
 
+import cn.hutool.core.io.FileUtil;
+import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.annotation.DbType;
 import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 
+import javax.annotation.PostConstruct;
 import javax.annotation.Resource;
 import javax.sql.DataSource;
 import java.sql.SQLException;
@@ -20,8 +24,27 @@ public class MyBatisPlusConfig {
 
     @Resource
     private DataSource dataSource;
-
-
+    
+    @Value("${spring.datasource.driver-class-name}")
+    private String datasourceDriveClassName;
+    
+    @Value("${spring.datasource.url}")
+    private String datasourceUrl;
+    
+    /**
+     * 如果是 sqlite 数据库，自动创建数据库文件所在目录
+     */
+    @PostConstruct
+    public void init() {
+        if (StrUtil.equals(datasourceDriveClassName, "org.sqlite.JDBC")) {
+            String path = datasourceUrl.replace("jdbc:sqlite:", "");
+            String folderPath = FileUtil.getParent(path, 1);
+            if (!FileUtil.exist(folderPath)) {
+                FileUtil.mkdir(folderPath);
+            }
+        }
+    }
+    
     /**
      * mybatis plus 分页插件配置
      */