108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
|
package model
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
_ "gorm.io/driver/sqlite"
|
||
|
|
"one-api/common"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Redemption struct {
|
||
|
|
Id int `json:"id"`
|
||
|
|
UserId int `json:"user_id"`
|
||
|
|
Key string `json:"key" gorm:"uniqueIndex"`
|
||
|
|
Status int `json:"status" gorm:"default:1"`
|
||
|
|
Name string `json:"name" gorm:"index"`
|
||
|
|
Quota int `json:"quota" gorm:"default:100"`
|
||
|
|
CreatedTime int64 `json:"created_time" gorm:"bigint"`
|
||
|
|
RedeemedTime int64 `json:"redeemed_time" gorm:"bigint"`
|
||
|
|
Count int `json:"count" gorm:"-:all"` // only for api request
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetAllRedemptions(startIdx int, num int) ([]*Redemption, error) {
|
||
|
|
var redemptions []*Redemption
|
||
|
|
var err error
|
||
|
|
err = DB.Order("id desc").Limit(num).Offset(startIdx).Find(&redemptions).Error
|
||
|
|
return redemptions, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func SearchRedemptions(keyword string) (redemptions []*Redemption, err error) {
|
||
|
|
err = DB.Where("id = ? or name LIKE ?", keyword, keyword+"%").Find(&redemptions).Error
|
||
|
|
return redemptions, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetRedemptionById(id int) (*Redemption, error) {
|
||
|
|
if id == 0 {
|
||
|
|
return nil, errors.New("id 为空！")
|
||
|
|
}
|
||
|
|
redemption := Redemption{Id: id}
|
||
|
|
var err error = nil
|
||
|
|
err = DB.First(&redemption, "id = ?", id).Error
|
||
|
|
return &redemption, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func Redeem(key string, tokenId int) (quota int, err error) {
|
||
|
|
if key == "" {
|
||
|
|
return 0, errors.New("未提供兑换码")
|
||
|
|
}
|
||
|
|
if tokenId == 0 {
|
||
|
|
return 0, errors.New("未提供 token id")
|
||
|
|
}
|
||
|
|
redemption := &Redemption{}
|
||
|
|
err = DB.Where("key = ?", key).First(redemption).Error
|
||
|
|
if err != nil {
|
||
|
|
return 0, errors.New("无效的兑换码")
|
||
|
|
}
|
||
|
|
if redemption.Status != common.RedemptionCodeStatusEnabled {
|
||
|
|
return 0, errors.New("该兑换码已被使用")
|
||
|
|
}
|
||
|
|
err = TopUpToken(tokenId, redemption.Quota)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
go func() {
|
||
|
|
redemption.RedeemedTime = common.GetTimestamp()
|
||
|
|
redemption.Status = common.RedemptionCodeStatusUsed
|
||
|
|
err := redemption.SelectUpdate()
|
||
|
|
if err != nil {
|
||
|
|
common.SysError("更新兑换码状态失败：" + err.Error())
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
return redemption.Quota, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (redemption *Redemption) Insert() error {
|
||
|
|
var err error
|
||
|
|
err = DB.Create(redemption).Error
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (redemption *Redemption) SelectUpdate() error {
|
||
|
|
// This can update zero values
|
||
|
|
return DB.Model(redemption).Select("redeemed_time", "status").Updates(redemption).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update Make sure your token's fields is completed, because this will update non-zero values
|
||
|
|
func (redemption *Redemption) Update() error {
|
||
|
|
var err error
|
||
|
|
err = DB.Model(redemption).Select("name", "status", "redeemed_time").Updates(redemption).Error
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (redemption *Redemption) Delete() error {
|
||
|
|
var err error
|
||
|
|
err = DB.Delete(redemption).Error
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func DeleteRedemptionById(id int) (err error) {
|
||
|
|
if id == 0 {
|
||
|
|
return errors.New("id 为空！")
|
||
|
|
}
|
||
|
|
redemption := Redemption{Id: id}
|
||
|
|
err = DB.Where(redemption).First(&redemption).Error
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return redemption.Delete()
|
||
|
|
}
|