From 65224685c83617fd8fdd4a9e72e3ce7c4c52c691 Mon Sep 17 00:00:00 2001 From: zhaojun1998 Date: Fri, 5 Mar 2021 22:23:41 +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 --- .../controller/home/ShortLinkController.java | 30 +++++++++++-------- 1 file changed, 17 insertions(+), 13 deletions(-) diff --git a/src/main/java/im/zhaojun/zfile/controller/home/ShortLinkController.java b/src/main/java/im/zhaojun/zfile/controller/home/ShortLinkController.java index 0db9ca2..a38be27 100644 --- a/src/main/java/im/zhaojun/zfile/controller/home/ShortLinkController.java +++ b/src/main/java/im/zhaojun/zfile/controller/home/ShortLinkController.java @@ -37,21 +37,25 @@ public class ShortLinkController { // 拼接直链地址. String fullPath = StringUtils.removeDuplicateSeparator("/directlink/" + driveId + path); - ShortLinkConfig shortLinkConfig; - String randomKey; - do { - // 获取短链 - randomKey = RandomUtil.randomString(6); - shortLinkConfig = shortLinkConfigService.findByKey(randomKey); - } while (shortLinkConfig != null); + ShortLinkConfig shortLinkConfig = shortLinkConfigService.findByUrl(fullPath); + + if (shortLinkConfig == null) { + + String randomKey; + do { + // 获取短链 + randomKey = RandomUtil.randomString(6); + shortLinkConfig = shortLinkConfigService.findByKey(randomKey); + } while (shortLinkConfig != null); + + shortLinkConfig = new ShortLinkConfig(); + shortLinkConfig.setKey(randomKey); + shortLinkConfig.setUrl(fullPath); + shortLinkConfigService.save(shortLinkConfig); + } - shortLinkConfig = new ShortLinkConfig(); - shortLinkConfig.setKey(randomKey); - shortLinkConfig.setUrl(fullPath); - shortLinkConfigService.save(shortLinkConfig); - - String shortUrl = StringUtils.removeDuplicateSeparator(domain + "/s/" + randomKey); + String shortUrl = StringUtils.removeDuplicateSeparator(domain + "/s/" + shortLinkConfig.getKey()); return ResultBean.successData(shortUrl); }