🎨 🔥 统一限速器

plugin_wangyiyun/main.go

@@ -2,12 +2,10 @@
 package wangyiyun
 
 import (
-	"time"
-
 	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
 	"github.com/FloatTech/zbputils/web"
 	zero "github.com/wdvxdr1123/ZeroBot"
-	"github.com/wdvxdr1123/ZeroBot/extension/rate"
 	"github.com/wdvxdr1123/ZeroBot/message"
 	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 
@@ -21,15 +19,11 @@ const (
 )
 
 func init() {
-	limit := rate.NewManager(time.Minute, 60)
 	control.Register("wangyiyun", order.PrioWangYiYun, &control.Options{
 		DisableOnDefault: false,
 		Help:             "wangyiyun \n- 来份网易云热评",
-	}).OnFullMatch("来份网易云热评").SetBlock(true).
+	}).OnFullMatch("来份网易云热评").SetBlock(true).Limit(ctxext.LimitByUser).
 		Handle(func(ctx *zero.Ctx) {
-			if !limit.Load(ctx.Event.GroupID).Acquire() {
-				return
-			}
 			data, err := web.ReqWith(wangyiyunURL, "GET", wangyiyunReferer, ua)
 			if err != nil {
 				ctx.SendChain(message.Text("ERROR:", err))