上一个随机图片未完成前拒绝请求

starim00 · Jun 4, 2021 · 352fd2e · 352fd2e
1 parent 08b2102
commit 352fd2e
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/msgext/msgext.go → api/msgext/msgext.go b/msgext/msgext.go → api/msgext/msgext.go
diff --git a/manager/manager.go b/manager/manager.go
@@ -4,7 +4,7 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/Yiwen-Chan/ZeroBot-Plugin/msgext"
+	"github.com/Yiwen-Chan/ZeroBot-Plugin/api/msgext"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
 )

diff --git a/randimg/setu_geter.go b/randimg/setu_geter.go
@@ -3,11 +3,12 @@ package randimg
 import (
 	"strings"
 
-	"github.com/Yiwen-Chan/ZeroBot-Plugin/msgext"
+	"github.com/Yiwen-Chan/ZeroBot-Plugin/api/msgext"
 	zero "github.com/wdvxdr1123/ZeroBot"
 )
 
 var RANDOM_API_URL = "https://api.pixivweb.com/anime18r.php?return=img"
+var BLOCK_REQUEST = false
 
 func init() { // 插件主体
 	zero.OnRegex(`^设置随机图片网址(.*)$`, zero.SuperUserPermission).SetBlock(true).SetPriority(20).
@@ -23,7 +24,13 @@ func init() { // 插件主体
 	// 随机图片
 	zero.OnFullMatchGroup([]string{"随机图片"}).SetBlock(true).SetPriority(24).
 		Handle(func(ctx *zero.Ctx) {
-			ctx.Send(msgext.ImageNoCache(RANDOM_API_URL))
+			if BLOCK_REQUEST {
+				ctx.Send("请稍后再试哦")
+			} else {
+				BLOCK_REQUEST = true
+				ctx.Send(msgext.ImageNoCache(RANDOM_API_URL))
+				BLOCK_REQUEST = false
+			}
 			return
 		})
 }