Relay done but not working

model/channel.go

@@ -2,12 +2,13 @@ package model
 
 import (
 	_ "gorm.io/driver/sqlite"
+	"one-api/common"
 )
 
 type Channel struct {
 	Id           int    `json:"id"`
 	Type         int    `json:"type" gorm:"default:0"`
-	Key          string `json:"key"`
+	Key          string `json:"key" gorm:"not null"`
 	Status       int    `json:"status" gorm:"default:1"`
 	Name         string `json:"name" gorm:"index"`
 	Weight       int    `json:"weight"`
@@ -27,10 +28,26 @@ func SearchChannels(keyword string) (channels []*Channel, err error) {
 	return channels, err
 }
 
-func GetChannelById(id int) (*Channel, error) {
+func GetChannelById(id int, selectAll bool) (*Channel, error) {
 	channel := Channel{Id: id}
 	var err error = nil
-	err = DB.Omit("key").First(&channel, "id = ?", id).Error
+	if selectAll {
+		err = DB.First(&channel, "id = ?", id).Error
+	} else {
+		err = DB.Omit("key").First(&channel, "id = ?", id).Error
+	}
+	return &channel, err
+}
+
+func GetRandomChannel() (*Channel, error) {
+	// TODO: consider weight
+	channel := Channel{}
+	var err error = nil
+	if common.UsingSQLite {
+		err = DB.Where("status = ?", common.ChannelStatusEnabled).Order("RANDOM()").Limit(1).First(&channel).Error
+	} else {
+		err = DB.Where("status = ?", common.ChannelStatusEnabled).Order("RAND()").Limit(1).First(&channel).Error
+	}
 	return &channel, err
 }
 

model/main.go

@@ -45,6 +45,7 @@ func InitDB() (err error) {
 		})
 	} else {
 		// Use SQLite
+		common.UsingSQLite = true
 		db, err = gorm.Open(sqlite.Open(common.SQLitePath), &gorm.Config{
 			PrepareStmt: true, // precompile SQL
 		})

model/token.go

@@ -3,12 +3,14 @@ package model
 import (
 	"errors"
 	_ "gorm.io/driver/sqlite"
+	"one-api/common"
+	"strings"
 )
 
 type Token struct {
 	Id           int    `json:"id"`
 	UserId       int    `json:"user_id"`
-	Key          string `json:"key"`
+	Key          string `json:"key" gorm:"uniqueIndex"`
 	Status       int    `json:"status" gorm:"default:1"`
 	Name         string `json:"name" gorm:"index" `
 	CreatedTime  int64  `json:"created_time" gorm:"bigint"`
@@ -27,6 +29,29 @@ func SearchUserTokens(userId int, keyword string) (tokens []*Token, err error) {
 	return tokens, err
 }
 
+func ValidateUserToken(key string) (token *Token, err error) {
+	if key == "" {
+		return nil, errors.New("未提供 token")
+	}
+	key = strings.Replace(key, "Bearer ", "", 1)
+	token = &Token{}
+	err = DB.Where("key = ?", key).First(token).Error
+	if err == nil {
+		if token.Status != common.TokenStatusEnabled {
+			return nil, errors.New("该 token 已被禁用")
+		}
+		go func() {
+			token.AccessedTime = common.GetTimestamp()
+			err := token.Update()
+			if err != nil {
+				common.SysError("更新 token 访问时间失败：" + err.Error())
+			}
+		}()
+		return token, nil
+	}
+	return nil, err
+}
+
 func GetTokenByIds(id int, userId int) (*Token, error) {
 	if id == 0 || userId == 0 {
 		return nil, errors.New("id 或 userId 为空！")

model/user.go

@@ -3,7 +3,6 @@ package model
 import (
 	"errors"
 	"one-api/common"
-	"strings"
 )
 
 // User if you add sensitive fields, don't forget to clean them in setupLogin function.
@@ -149,18 +148,6 @@ func (user *User) FillUserByUsername() error {
 	return nil
 }
 
-func ValidateUserToken(token string) (user *User) {
-	if token == "" {
-		return nil
-	}
-	token = strings.Replace(token, "Bearer ", "", 1)
-	user = &User{}
-	if DB.Where("token = ?", token).First(user).RowsAffected == 1 {
-		return user
-	}
-	return nil
-}
-
 func IsEmailAlreadyTaken(email string) bool {
 	return DB.Where("email = ?", email).Find(&User{}).RowsAffected == 1
 }