🎨 🔥 统一限速器

plugin_funny/laugh.go

@@ -3,10 +3,8 @@ package funny
 
 import (
 	"strings"
-	"time"
 
 	zero "github.com/wdvxdr1123/ZeroBot"
-	"github.com/wdvxdr1123/ZeroBot/extension/rate"
 	"github.com/wdvxdr1123/ZeroBot/message"
 
 	control "github.com/FloatTech/zbputils/control"
@@ -22,15 +20,11 @@ var (
 		Help: "讲个笑话\n" +
 			"- 讲个笑话[@xxx] | 讲个笑话[qq号]",
 	})
-	limit = rate.NewManager(time.Minute, 20)
-	db    = &sql.Sqlite{DBPath: dbfile}
+	db = &sql.Sqlite{DBPath: dbfile}
 )
 
 func init() {
-	engine.OnPrefix("讲个笑话").SetBlock(true).Handle(func(ctx *zero.Ctx) {
-		if !limit.Load(ctx.Event.GroupID).Acquire() {
-			return
-		}
+	engine.OnPrefix("讲个笑话").SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
 		// 获取名字
 		name := ctxext.NickName(ctx)
 		var j joke