Kelvin
/
api2gpt-mid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bug fix
continuous-integration/drone/push Build is passing Details

Browse Source
main
kelvin 3 years ago
parent 7f295644d2
commit ac22add9e8
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File

9
service.go
Unescape View File

@ -49,7 +49,7 @@ func queryBlance(key string) (float64, error) {
userInfoStr, err := Redis.Get(context.Background(), "user:"+key).Result() userInfoStr, err := Redis.Get(context.Background(), "user:"+key).Result()
var userInfo UserInfo var userInfo UserInfo
err = json.Unmarshal([]byte(userInfoStr), &userInfo) err = json.Unmarshal([]byte(userInfoStr), &userInfo)
balance, err := Redis.IncrByFloat(context.Background(), "user:"+userInfo.SID+":balance", 0).Result() balance, err := Redis.IncrByFloat(context.Background(), "user:"+userInfo.UID+":balance", 0).Result()
if err != nil { if err != nil {
return 0, errors.New("余额计算失败") return 0, errors.New("余额计算失败")
} }
@ -83,12 +83,13 @@ func checkBlance(key string, model string) (ServerInfo, error) {
return serverInfo, errors.New("模型信息解析失败") return serverInfo, errors.New("模型信息解析失败")
} }
//计算余额-先扣除指定金额 //计算余额-先扣除指定金额
balance, err := Redis.IncrByFloat(context.Background(), "user:"+userInfo.SID+":balance", -(float64(modelInfo.ModelPrepayment) * modelInfo.ModelPrice)).Result() balance, err := Redis.IncrByFloat(context.Background(), "user:"+userInfo.UID+":balance", -(float64(modelInfo.ModelPrepayment) * modelInfo.ModelPrice)).Result()
if err != nil { if err != nil {
return serverInfo, errors.New("余额计算失败") return serverInfo, errors.New("余额计算失败")
} }
log.Printf("用户余额 %f key: %v", balance, key)
if balance < 0 { if balance < 0 {
Redis.IncrByFloat(context.Background(), "user:"+userInfo.SID+":balance", float64(modelInfo.ModelPrepayment)*modelInfo.ModelPrice).Result() Redis.IncrByFloat(context.Background(), "user:"+userInfo.UID+":balance", float64(modelInfo.ModelPrepayment)*modelInfo.ModelPrice).Result()
return serverInfo, errors.New("用户余额不足") return serverInfo, errors.New("用户余额不足")
} }
@ -117,7 +118,7 @@ func consumption(key string, model string, prompt_tokens int, completion_tokens
if err != nil { if err != nil {
return "", errors.New("模型信息解析失败") return "", errors.New("模型信息解析失败")
} }
Redis.IncrByFloat(context.Background(), "user:"+userInfo.SID+":balance", float64(modelInfo.ModelPrepayment)*modelInfo.ModelPrice-(float64(total_tokens)*modelInfo.ModelPrice)).Result() Redis.IncrByFloat(context.Background(), "user:"+userInfo.UID+":balance", float64(modelInfo.ModelPrepayment)*modelInfo.ModelPrice-(float64(total_tokens)*modelInfo.ModelPrice)).Result()
// 余额消费日志请求 // 余额消费日志请求
result, err := balanceConsumption(key, model, prompt_tokens, completion_tokens, total_tokens, msg_id) result, err := balanceConsumption(key, model, prompt_tokens, completion_tokens, total_tokens, msg_id)

Write Preview
Loading…
Cancel
Save