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

fix: 初始令牌

Browse Source
This commit is contained in:
CalciumIon 2024-09-17 23:07:16 +08:00
parent 131453dac8
commit 9032b5cfbf
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
controller/user.go
View File

@ -187,7 +187,7 @@ func Register(c *gin.Context) {
}) })
return return
} }
// 获取插入后的用户ID // 获取插入后的用户ID
var insertedUser model.User var insertedUser model.User
if err := model.DB.Where("username = ?", cleanUser.Username).First(&insertedUser).Error; err != nil { if err := model.DB.Where("username = ?", cleanUser.Username).First(&insertedUser).Error; err != nil {
@ -199,23 +199,15 @@ func Register(c *gin.Context) {
} }
// 生成默认令牌 // 生成默认令牌
if constant.GenerateDefaultToken { if constant.GenerateDefaultToken {
var insertedUser model.User
if err := model.DB.Where("username = ?", cleanUser.Username).First(&insertedUser).Error; err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,
"message": "用户注册失败或用户ID获取失败",
})
return
}
// 生成默认令牌 // 生成默认令牌
token := model.Token{ token := model.Token{
UserId: insertedUser.Id, // 使用插入后的用户ID UserId: insertedUser.Id, // 使用插入后的用户ID
Name: cleanUser.Username + "的初始令牌", Name: cleanUser.Username + "的初始令牌",
Key: common.GenerateKey(), Key: common.GenerateKey(),
CreatedTime: common.GetTimestamp(), CreatedTime: common.GetTimestamp(),
AccessedTime: common.GetTimestamp(), AccessedTime: common.GetTimestamp(),
ExpiredTime: -1, // 永不过期 ExpiredTime: -1, // 永不过期
RemainQuota: 500000, // 示例额度 RemainQuota: 500000, // 示例额度
UnlimitedQuota: true, UnlimitedQuota: true,
ModelLimitsEnabled: false, ModelLimitsEnabled: false,
} }