xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-12 03:13:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 调整为定时每晚00:00

Browse Source 
feat: 添加Redis更新缓存操作
fate: expiration_date设置默认值0
This commit is contained in:
jinjianmingming
2024-07-18 22:58:04 +08:00
parent 1aba1e4ac3
commit 66d31102d3
6 changed files with 71 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
model/option.go
View File

@@ -1,9 +1,11 @@
package model
import (
"github.com/go-co-op/gocron"
"github.com/songquanpeng/one-api/common/config"
"github.com/songquanpeng/one-api/common/logger"
billingratio "github.com/songquanpeng/one-api/relay/billing/ratio"
"log"
"strconv"
"strings"
"time"
@@ -78,12 +80,19 @@ func InitOptionMap() {
loadOptionsFromDatabase()
}
func ScheduleCheckAndDowngrade(intervalHours int) {
for {
logger.SysLog("Check user group levels.")
checkAndDowngradeUsers()
time.Sleep(time.Duration(intervalHours) * time.Hour)
func ScheduleCheckAndDowngrade() {
s := gocron.NewScheduler(time.UTC)
// 设置每天0点执行
_, err := s.Every(1).Day().At("00:00").Do(checkAndDowngradeUsers)
if err != nil {
log.Fatalf("创建调度任务失败: %v", err)
}
// 开始调度
s.StartBlocking()
log.Printf("开始调度")
}
func loadOptionsFromDatabase() {

42
model/user.go
View File

@@ -49,7 +49,7 @@ type User struct {
Group string `json:"group" gorm:"type:varchar(32);default:'default'"`
AffCode string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"`
InviterId int `json:"inviter_id" gorm:"type:int;column:inviter_id;index"`
ExpirationDate int64 `json:"expiration_date" gorm:"column:expiration_date"` // Expiration date of the user's subscription or account.
ExpirationDate int64 `json:"expiration_date" gorm:"bigint;default:0;column:expiration_date"` // Expiration date of the user's subscription or account.
}
func GetMaxUserId() int {
@@ -214,7 +214,7 @@ func (user *User) ValidateAndFill() (err error) {
return errors.New("用户名或密码错误，或用户已被封禁")
}
// 校验用户是不是非default,如果是非default,判断到期时间如果过期了降级为default
if user.ExpirationDate > 0 {
if !(user.ExpirationDate > 0 && user.Username == "root") {
// 将时间戳转换为 time.Time 类型
expirationTime := time.Unix(user.ExpirationDate, 0)
// 获取当前时间
@@ -459,21 +459,35 @@ func GetUsernameById(id int) (username string) {
}
func checkAndDowngradeUsers() {
// 获取当前时间的 Unix 时间戳
currentTime := time.Now().Unix()
// 获取当前时间的前一天的日期部分
//yesterday := time.Now().AddDate(0, 0, -1).Format("2006-01-02")
// 构建更新条件并执行更新
result := DB.Model(&User{}).
Where("`Group` <> ?", "default").
Where("`username` <> ?", "root").
Where("`expiration_date` IS NOT NULL").
Where("`expiration_date` != ?", -1).
Where("`expiration_date` < ?", currentTime).
Update("Group", "default")
// 获取昨天的时间戳
yesterdayTimestamp := time.Now().AddDate(0, 0, -1).Unix()
// Construct the query
query := DB.Model(&User{}).
Where("`Group` != ?", "default"). // Use single quotes for string literal
Where("`username` != ?", "root").
Where("`expiration_date` > 0").
Where("`expiration_date` <= ?", yesterdayTimestamp)
var userList []int
query.Select("id").Find(&userList)
// 降级更新操作
query.Update("Group", "default")
// 处理错误
if result.Error != nil {
log.Printf("批量更新用户分组失败: %v", result.Error)
if query.Error != nil {
log.Printf("批量更新用户分组失败: %v", query.Error)
return
}
// 删除已过期用户的Redis缓存
if common.RedisEnabled {
for _, userId := range userList {
err := common.RedisSet(fmt.Sprintf("user_group:%d", userId), "default", time.Duration(UserId2GroupCacheSeconds)*time.Second)
if err != nil {
log.Printf("更新用户: %d,权益缓存失败, Error: %v", userId, query.Error)
}
}
}
}