From 4d42529c4d65e04ef4115675eec24a31f1946fc3 Mon Sep 17 00:00:00 2001
From: zhaojun1998 <root@zhaojun.im>
Date: Fri, 5 Mar 2021 22:24:28 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E7=9F=AD=E9=93=BE=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E9=80=BB=E8=BE=91=E4=BC=98=E5=8C=96,=20=E5=90=8C?=
 =?UTF-8?q?=E9=93=BE=E6=8E=A5=E4=B8=8D=E9=87=8D=E5=A4=8D=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../zfile/repository/ShortLinkConfigRepository.java    | 10 +++++++++-
 .../zhaojun/zfile/service/ShortLinkConfigService.java  |  4 ++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java b/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
index 3431d1d..32868e6 100644
--- a/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
+++ b/src/main/java/im/zhaojun/zfile/repository/ShortLinkConfigRepository.java
@@ -20,7 +20,7 @@ public interface ShortLinkConfigRepository extends JpaRepository<ShortLinkConfig
      * 获取驱动器下的所有规则
      *
      * @param       key
-     *              驱动器 ID
+     *              短链 Key
      */
     ShortLinkConfig findByKey(String key);
 
@@ -49,4 +49,12 @@ public interface ShortLinkConfigRepository extends JpaRepository<ShortLinkConfig
     //                 " and (:dateTo is null or date_format(create_date, '%Y-%m-%d') <= date_format(:dateTo, '%Y-%m-%d')) ) "
     // )
     Page<ShortLinkConfig> findByPage(String key, String url, Date dateFrom, Date dateTo, Pageable pageable);
+
+    /**
+     * 获取驱动器下的所有规则
+     *
+     * @param       url
+     *              短链 URL
+     */
+    ShortLinkConfig findByUrl(String url);
 }
\ No newline at end of file
diff --git a/src/main/java/im/zhaojun/zfile/service/ShortLinkConfigService.java b/src/main/java/im/zhaojun/zfile/service/ShortLinkConfigService.java
index 22afc89..b908432 100644
--- a/src/main/java/im/zhaojun/zfile/service/ShortLinkConfigService.java
+++ b/src/main/java/im/zhaojun/zfile/service/ShortLinkConfigService.java
@@ -27,6 +27,10 @@ public class ShortLinkConfigService {
         return shortLinkConfigRepository.findByKey(key);
     }
 
+    public ShortLinkConfig findByUrl(String url) {
+        return shortLinkConfigRepository.findByUrl(url);
+    }
+
     public void save(ShortLinkConfig shortLinkConfig) {
         shortLinkConfig.setCreateDate(new Date());
         shortLinkConfigRepository.save(shortLinkConfig);