fix: support kick cron

This commit is contained in:
Vaala Cat
2023-06-19 19:54:47 +08:00
parent c278b3f516
commit 7bd7e940ba
5 changed files with 30 additions and 13 deletions

View File

@@ -9,7 +9,7 @@ import (
type User struct {
gorm.Model
TGID int64
TGID int64 `gorm:"unique"`
MCName string
Status int // 0: pending, 1: normal, 2: banned
}
@@ -38,15 +38,13 @@ func GetUserByMCName(mcName string) (user User, err error) {
return
}
func CreateUserIfNotExist(tgID int64, u *User) (err error) {
err = database.GetDB().Where(
&User{TGID: tgID},
).FirstOrCreate(&u).Error
func CreateUser(u *User) (err error) {
err = database.GetDB().Create(&u).Error
return
}
func (u *User) Delete(tgID int64) error {
return database.GetDB().Where(
&User{TGID: tgID},
).Delete(&u).Error
).Unscoped().Delete(&u).Error
}