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 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); }