From 7409df85d793101d7edea44dc22f813e06721c19 Mon Sep 17 00:00:00 2001
From: zhaojun1998 <root@zhaojun.im>
Date: Fri, 5 Mar 2021 22:59:08 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E5=A4=8D=E5=88=87=E6=8D=A2?=
 =?UTF-8?q?=E5=AD=98=E5=82=A8=E5=99=A8=20ID=20=E6=97=B6=EF=BC=8C=E4=B8=BA?=
 =?UTF-8?q?=E7=BA=A7=E8=81=94=E4=BF=AE=E6=94=B9=E6=88=96=E6=B8=85=E7=90=86?=
 =?UTF-8?q?=E7=BC=93=E5=AD=98=E3=80=81=E8=BF=87=E6=BB=A4=E5=99=A8=E3=80=81?=
 =?UTF-8?q?=E7=9F=AD=E9=93=BE=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../zfile/repository/FilterConfigRepository.java  | 15 +++++++++++++++
 .../repository/ShortLinkConfigRepository.java     | 15 +++++++++++++++
 .../zhaojun/zfile/service/DriveConfigService.java | 11 +++++++++++
 3 files changed, 41 insertions(+)

diff --git a/src/main/java/im/zhaojun/zfile/repository/FilterConfigRepository.java b/src/main/java/im/zhaojun/zfile/repository/FilterConfigRepository.java
index 6dd7774..e8fc7bc 100644
--- a/src/main/java/im/zhaojun/zfile/repository/FilterConfigRepository.java
+++ b/src/main/java/im/zhaojun/zfile/repository/FilterConfigRepository.java
@@ -2,6 +2,8 @@ package im.zhaojun.zfile.repository;
 
 import im.zhaojun.zfile.model.entity.FilterConfig;
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Modifying;
+import org.springframework.data.jpa.repository.Query;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
@@ -26,4 +28,17 @@ public interface FilterConfigRepository extends JpaRepository<FilterConfig, Inte
      */
     void deleteByDriveId(Integer driveId);
 
+
+    /**
+     * 更新驱动器 ID 对应的参数设置为新的驱动器 ID
+     *
+     * @param   updateId
+     *          驱动器原 ID
+     *
+     * @param   newId
+     *          驱动器新 ID
+     */
+    @Modifying
+    @Query(value="update FILTER_CONFIG set driveId = :newId where driveId = :updateId")
+    void updateDriveId(Integer updateId, Integer newId);
 }
\ No newline at end of file
diff --git a/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java b/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
index 32868e6..13bdfe6 100644
--- a/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
+++ b/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
@@ -5,6 +5,7 @@ import org.springframework.data.domain.Page;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
+import org.springframework.data.jpa.repository.Modifying;
 import org.springframework.data.jpa.repository.Query;
 import org.springframework.stereotype.Repository;
 
@@ -57,4 +58,18 @@ public interface ShortLinkConfigRepository extends JpaRepository<ShortLinkConfig
      *              短链 URL
      */
     ShortLinkConfig findByUrl(String url);
+
+
+    /**
+     * 更新驱动器 ID
+     *
+     * @param   updateSubPath
+     *          原路径部分名称
+     *
+     * @param   newSubPath
+     *          修改后路径部分名称
+     */
+    @Modifying
+    @Query(value="update SHORT_LINK set url = replace(url, :updateSubPath, :newSubPath)")
+    void updateUrlDriveId(String updateSubPath, String newSubPath);
 }
\ No newline at end of file
diff --git a/src/main/java/im/zhaojun/zfile/service/DriveConfigService.java b/src/main/java/im/zhaojun/zfile/service/DriveConfigService.java
index 60dff93..5786c97 100644
--- a/src/main/java/im/zhaojun/zfile/service/DriveConfigService.java
+++ b/src/main/java/im/zhaojun/zfile/service/DriveConfigService.java
@@ -1,5 +1,6 @@
 package im.zhaojun.zfile.service;
 
+import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSONObject;
 import im.zhaojun.zfile.cache.ZFileCache;
 import im.zhaojun.zfile.context.DriveContext;
@@ -13,6 +14,8 @@ import im.zhaojun.zfile.model.entity.DriveConfig;
 import im.zhaojun.zfile.model.entity.StorageConfig;
 import im.zhaojun.zfile.model.enums.StorageTypeEnum;
 import im.zhaojun.zfile.repository.DriverConfigRepository;
+import im.zhaojun.zfile.repository.FilterConfigRepository;
+import im.zhaojun.zfile.repository.ShortLinkConfigRepository;
 import im.zhaojun.zfile.repository.StorageConfigRepository;
 import im.zhaojun.zfile.service.base.AbstractBaseFileService;
 import lombok.extern.slf4j.Slf4j;
@@ -42,6 +45,12 @@ public class DriveConfigService {
     @Resource
     private StorageConfigRepository storageConfigRepository;
 
+    @Resource
+    private FilterConfigRepository filterConfigRepository;
+
+    @Resource
+    private ShortLinkConfigRepository shortLinkConfigRepository;
+
     @Resource
     private DriveContext driveContext;
 
@@ -248,6 +257,8 @@ public class DriveConfigService {
     public void updateId(Integer updateId, Integer newId) {
         driverConfigRepository.updateId(updateId, newId);
         storageConfigRepository.updateDriveId(updateId, newId);
+        filterConfigRepository.updateDriveId(updateId, newId);
+        shortLinkConfigRepository.updateUrlDriveId("/directlink/" + updateId, "/directlink/" + newId);
         driveContext.updateDriveId(updateId, newId);
     }