From 657a200d8165ba92731cc13169ec1f1795f4784c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A6=BE=E5=87=A0=E6=B5=B7?= <zh56462271@qq.com>
Date: Sat, 1 Aug 2020 12:21:03 +0800
Subject: [PATCH] ...

---
 .../cn/celess/blog/service/PartnerSiteService.java  | 13 ++++++++++---
 .../service/serviceimpl/PartnerSiteServiceImpl.java |  4 +++-
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cn/celess/blog/service/PartnerSiteService.java b/src/main/java/cn/celess/blog/service/PartnerSiteService.java
index 61ba0c0..e76db4a 100644
--- a/src/main/java/cn/celess/blog/service/PartnerSiteService.java
+++ b/src/main/java/cn/celess/blog/service/PartnerSiteService.java
@@ -55,11 +55,18 @@ public interface PartnerSiteService {
     List<PartnerSite> findAll();
 
     /**
-     *  申请友链
-     * @param linkApplyReq  linkApplyReq
+     * 申请友链
+     *
+     * @param linkApplyReq linkApplyReq
      * @return linkApplyReq
      */
     PartnerSite apply(LinkApplyReq linkApplyReq);
 
-    Object reapply(String key);
+    /**
+     * 重写申请友链
+     *
+     * @param key key
+     * @return msg
+     */
+    String reapply(String key);
 }
diff --git a/src/main/java/cn/celess/blog/service/serviceimpl/PartnerSiteServiceImpl.java b/src/main/java/cn/celess/blog/service/serviceimpl/PartnerSiteServiceImpl.java
index 930f0d7..af49b48 100644
--- a/src/main/java/cn/celess/blog/service/serviceimpl/PartnerSiteServiceImpl.java
+++ b/src/main/java/cn/celess/blog/service/serviceimpl/PartnerSiteServiceImpl.java
@@ -176,6 +176,8 @@ public class PartnerSiteServiceImpl implements PartnerSiteService {
             boolean exists = partnerMapper.existsByUrl(linkApplyReq.getUrl());
             if (!exists) {
                 partnerMapper.insert(ps);
+            } else {
+                ps.setId(partnerMapper.findByUrl(linkApplyReq.getUrl()).getId());
             }
             SimpleMailMessage smm = new SimpleMailMessage();
             smm.setSubject("友链申请");
@@ -204,7 +206,7 @@ public class PartnerSiteServiceImpl implements PartnerSiteService {
 
     @SneakyThrows
     @Override
-    public Object reapply(String key) {
+    public String reapply(String key) {
         if (!redisUtil.hasKey(key)) {
             throw new MyException(ResponseEnum.DATA_EXPIRED);
         }