xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-26 20:06:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update user.go

Browse Source
This commit is contained in:
Mary Smiley 2024-03-16 09:37:09 +08:00 committed by GitHub
parent b8ef5cd34c
commit 317aeefd8d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 45 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
model/user.go
View File

@ -22,6 +22,8 @@ type User struct {
Email string `json:"email" gorm:"index" validate:"max=50"`
GitHubId string `json:"github_id" gorm:"column:github_id;index"`
LinuxDoId string `json:"linuxdo_id" gorm:"column:linuxdo_id;index"`
LinuxTrustLevel int `json:"trust_level" gorm:"type:int;default:0"`
RefeashTimeStamp int64 `json:"refeash_time_stamp" gorm:"type:bigint;default:0"`
WeChatId string `json:"wechat_id" gorm:"column:wechat_id;index"`
TelegramId string `json:"telegram_id" gorm:"column:telegram_id;index"`
VerificationCode string `json:"verification_code" gorm:"-:all"` // this field is only for Email verification, don't save it to database!
@ -386,6 +388,12 @@ func GetUserQuota(id int) (quota int, err error) {
return quota, err
}
func GetUserRefreshTimeStamp(id int) (timeStamp int64, err error) {
err = DB.Model(&User{}).Where("id = ?", id).Select("refeash_time_stamp").Find(&timeStamp).Error
return timeStamp, err
}
func GetUserUsedQuota(id int) (quota int, err error) {
err = DB.Model(&User{}).Where("id = ?", id).Select("used_quota").Find(&quota).Error
return quota, err
@ -406,6 +414,43 @@ func GetUserGroup(id int) (group string, err error) {
return group, err
}
func RefreshUserQuotaAndSave(id int, TimeStamp int64) (err error) {
user, err := GetUserById(id, true)
if err != nil && user.LinuxDoId == "" {
return err
}
// 根据信任等级刷新用户额度
user.Quota = user.UsedQuota
switch user.LinuxTrustLevel {
case 1:
user.Quota += common.QuotaForLinuxDoLevel1
case 2:
user.Quota += common.QuotaForLinuxDoLevel2
case 3:
user.Quota += common.QuotaForLinuxDoLevel3
case 4:
user.Quota += common.QuotaForLinuxDoLevel4
}
// 更新用户刷新时间戳
user.RefeashTimeStamp = TimeStamp
RecordLog(id, LogTypeSystem, fmt.Sprintf("LinuxDO 用户信任等级 %d，额度自动刷新", user.LinuxTrustLevel))
// 将额度和时间戳写入数据库
err = DB.Model(&User{}).Where("id = ?", id).Updates(
map[string]interface{}{
"quota": user.Quota,
"refeash_time_stamp": user.RefeashTimeStamp,
},
).Error
return err
}
func IncreaseUserQuota(id int, quota int) (err error) {
if quota < 0 {
return errors.New("quota 不能为负数！")