jd/xdd-plus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add 增加查询次数限制等各种命令限制

Browse Source
This commit is contained in:
ZJY 2021-09-27 16:22:24 +08:00
parent 0ced679258
commit 072fe83e20
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/SendLimit.go
View File

@ -17,7 +17,6 @@ func getLimit(uid int, typ int) bool {
if Config.Lim == 0 { if Config.Lim == 0 {
return true return true
} }
logs.Info("开始限制")
u := &Limit{} u := &Limit{}
err := db.Where("number = ? and typ = ? and active_at = ?", uid, typ, time.Now().Format("2006-01-02")).First(&u).Error err := db.Where("number = ? and typ = ? and active_at = ?", uid, typ, time.Now().Format("2006-01-02")).First(&u).Error
if err == nil { if err == nil {
@ -31,7 +30,6 @@ func getLimit(uid int, typ int) bool {
return false return false
} }
} else { } else {
logs.Info("开始进入创建")
begin := db.Begin() begin := db.Begin()
begin.Create(&Limit{ begin.Create(&Limit{
ActiveAt: time.Now().Format("2006-01-02"), ActiveAt: time.Now().Format("2006-01-02"),

2
models/command.go
View File

@ -296,7 +296,7 @@ var codeSignals = []CodeSignal{
sender.Reply(ck.Query()) sender.Reply(ck.Query())
}) })
} else { } else {
sender.Reply("超过今日查询次数") sender.Reply(fmt.Sprintf("鉴于东哥对接口限流，为了不影响大家的任务正常运行，即日起每日限流%d次，已超过今日限制", Config.Lim))
} }
return nil return nil
}, },