🎨 🔥 统一限速器

plugin_shindan/shindan.go

@@ -2,12 +2,9 @@
 package shindan
 
 import (
-	"time"
-
 	"github.com/FloatTech/AnimeAPI/shindanmaker"
 	log "github.com/sirupsen/logrus"
 	zero "github.com/wdvxdr1123/ZeroBot"
-	"github.com/wdvxdr1123/ZeroBot/extension/rate"
 	"github.com/wdvxdr1123/ZeroBot/message"
 	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 
@@ -18,11 +15,6 @@ import (
 	"github.com/FloatTech/zbputils/txt2img"
 )
 
-var (
-	// 限制调用频率
-	limit = rate.NewManager(time.Minute*5, 5)
-)
-
 func init() {
 	engine := control.Register("shindan", order.PrioShinDan, &control.Options{
 		DisableOnDefault: false,
@@ -32,18 +24,14 @@ func init() {
 			"- 卖萌[@xxx]\n" +
 			"- 抽老婆[@xxx]",
 	})
-	engine.OnPrefix("异世界转生", number(587874)).SetBlock(true).Handle(handle)
-	engine.OnPrefix("今天是什么少女", number(162207)).SetBlock(true).Handle(handle)
-	engine.OnPrefix("卖萌", number(360578)).SetBlock(true).Handle(handle)
-	engine.OnPrefix("抽老婆", number(1075116)).SetBlock(true).Handle(handle)
+	engine.OnPrefix("异世界转生", number(587874)).SetBlock(true).Limit(ctxext.LimitByUser).Handle(handle)
+	engine.OnPrefix("今天是什么少女", number(162207)).SetBlock(true).Limit(ctxext.LimitByUser).Handle(handle)
+	engine.OnPrefix("卖萌", number(360578)).SetBlock(true).Limit(ctxext.LimitByUser).Handle(handle)
+	engine.OnPrefix("抽老婆", number(1075116)).SetBlock(true).Limit(ctxext.LimitByUser).Handle(handle)
 }
 
 // shindanmaker 处理函数
 func handle(ctx *zero.Ctx) {
-	if !limit.Load(ctx.Event.UserID).Acquire() {
-		ctx.SendChain(message.Text("请稍后重试0x0..."))
-		return
-	}
 	// 获取名字
 	name := ctxext.NickName(ctx)
 	// 调用接口