39 lines
690 B
Go
39 lines
690 B
Go
package models
|
|
|
|
import "time"
|
|
|
|
type Limit struct {
|
|
ID int `gorm:"column:ID;primaryKey"`
|
|
Number int
|
|
ActiveAt string
|
|
Typ int
|
|
Num int
|
|
}
|
|
|
|
func getLimit(uid int, typ int) bool {
|
|
if Config.Lim == 0 {
|
|
return true
|
|
}
|
|
u := &Limit{}
|
|
err := db.Where("number = ? and typ = ? and active_at = ?", uid, typ, time.Now().Format("2006-01-02")).First(&u).Error
|
|
if err != nil {
|
|
if u.Num < Config.Lim {
|
|
db.Updates(&Limit{
|
|
Num: u.Num + 1,
|
|
}).Where("ID = ?", u.ID)
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
} else {
|
|
db.Begin()
|
|
u.ActiveAt = time.Now().Format("2006-01-02")
|
|
u.Typ = typ
|
|
u.Number = uid
|
|
u.Num = 1
|
|
db.Create(u)
|
|
db.Commit()
|
|
return true
|
|
}
|
|
}
|