From c7c3b9d32675dfd56886bae8671da8ca6a1ebb76 Mon Sep 17 00:00:00 2001 From: ckt1031 <65409152+ckt1031@users.noreply.github.com> Date: Thu, 27 Jul 2023 11:07:08 +0800 Subject: [PATCH] fix: backstick for postgres --- model/cache.go | 4 ++-- model/channel.go | 2 +- model/redemption.go | 5 +++-- 3 files changed, 6 insertions(+), 5 deletions(-) diff --git a/model/cache.go b/model/cache.go index d734351d..0a318561 100644 --- a/model/cache.go +++ b/model/cache.go @@ -22,12 +22,12 @@ var ( func CacheGetTokenByKey(key string) (*Token, error) { var token Token if !common.RedisEnabled { - err := DB.Where("`key` = ?", key).First(&token).Error + err := DB.Where("key = ?", key).First(&token).Error return &token, err } tokenObjectString, err := common.RedisGet(fmt.Sprintf("token:%s", key)) if err != nil { - err := DB.Where("`key` = ?", key).First(&token).Error + err := DB.Where("key = ?", key).First(&token).Error if err != nil { return nil, err } diff --git a/model/channel.go b/model/channel.go index cb03ad11..d3053807 100644 --- a/model/channel.go +++ b/model/channel.go @@ -40,7 +40,7 @@ func GetAllChannels(startIdx int, num int, selectAll bool) ([]*Channel, error) { } func SearchChannels(keyword string) (channels []*Channel, err error) { - err = DB.Omit("key").Where("id = ? or name LIKE ? or `key` = ?", keyword, keyword+"%", keyword).Find(&channels).Error + err = DB.Omit("key").Where("id = ? or name LIKE ? or key = ?", keyword, keyword+"%", keyword).Find(&channels).Error return channels, err } diff --git a/model/redemption.go b/model/redemption.go index fafb2145..ef702345 100644 --- a/model/redemption.go +++ b/model/redemption.go @@ -3,8 +3,9 @@ package model import ( "errors" "fmt" - "gorm.io/gorm" "one-api/common" + + "gorm.io/gorm" ) type Redemption struct { @@ -51,7 +52,7 @@ func Redeem(key string, userId int) (quota int, err error) { redemption := &Redemption{} err = DB.Transaction(func(tx *gorm.DB) error { - err := tx.Set("gorm:query_option", "FOR UPDATE").Where("`key` = ?", key).First(redemption).Error + err := tx.Set("gorm:query_option", "FOR UPDATE").Where("key = ?", key).First(redemption).Error if err != nil { return errors.New("无效的兑换码") }