feat: 添加查询余额的交互能力 (#70)

main.go

@@ -110,16 +110,19 @@ func ProcessRequest(rmsg public.ReceiveMsg) error {
  logger.Warning(fmt.Errorf("send message error: %v", err))
  }
  case "余额":
- rst, err := public.GetBalance()
- if err != nil {
- logger.Warning(fmt.Errorf("get balance error: %v", err))
- return err
+ cacheMsg := public.UserService.GetUserMode("system_balance")
+ if cacheMsg == "" {
+ rst, err := public.GetBalance()
+ if err != nil {
+ logger.Warning(fmt.Errorf("get balance error: %v", err))
+ return err
+ }
+ t1 := time.Unix(int64(rst.Grants.Data[0].EffectiveAt), 0)
+ t2 := time.Unix(int64(rst.Grants.Data[0].ExpiresAt), 0)
+ cacheMsg = fmt.Sprintf("💵 已用: 💲%v\n💵 剩余: 💲%v\n⏳ 有效时间: 从 %v 到 %v\n", fmt.Sprintf("%.2f", rst.TotalUsed), fmt.Sprintf("%.2f", rst.TotalAvailable), t1.Format("2006-01-02 15:04:05"), t2.Format("2006-01-02 15:04:05"))
  }
- t1 := time.Unix(int64(rst.Grants.Data[0].EffectiveAt), 0)
- t2 := time.Unix(int64(rst.Grants.Data[0].ExpiresAt), 0)
- msg := fmt.Sprintf("💵 已用: 💲%v\n💵 剩余: 💲%v\n⏳ 有效时间: 从 %v 到 %v\n", fmt.Sprintf("%.2f", rst.TotalUsed), fmt.Sprintf("%.2f", rst.TotalAvailable), t1.Format("2006-01-02 15:04:05"), t2.Format("2006-01-02 15:04:05"))
 
- _, err = rmsg.ReplyText(msg, rmsg.SenderStaffId)
+ _, err := rmsg.ReplyText(cacheMsg, rmsg.SenderStaffId)
  if err != nil {
  logger.Warning(fmt.Errorf("send message error: %v", err))
  }

public/gpt.go

@@ -10,7 +10,10 @@ import (
 )
 
 func InitAiCli() *resty.Client {
- return resty.New().SetTimeout(30*time.Second).SetHeader("Authorization", fmt.Sprintf("Bearer %s", config.LoadConfig().ApiKey)).SetProxy(config.LoadConfig().HttpProxy)
+ if config.LoadConfig().HttpProxy != "" {
+ return resty.New().SetTimeout(30*time.Second).SetHeader("Authorization", fmt.Sprintf("Bearer %s", config.LoadConfig().ApiKey)).SetProxy(config.LoadConfig().HttpProxy).SetRetryCount(3).SetRetryWaitTime(5 * time.Second)
+ }
+ return resty.New().SetTimeout(30*time.Second).SetHeader("Authorization", fmt.Sprintf("Bearer %s", config.LoadConfig().ApiKey)).SetRetryCount(3).SetRetryWaitTime(5 * time.Second)
 }
 
 type Billing struct {
@@ -42,5 +45,10 @@ func GetBalance() (Billing, error) {
  if err != nil {
  return data, err
  }
+ t1 := time.Unix(int64(data.Grants.Data[0].EffectiveAt), 0)
+ t2 := time.Unix(int64(data.Grants.Data[0].ExpiresAt), 0)
+ msg := fmt.Sprintf("💵 已用: 💲%v\n💵 剩余: 💲%v\n⏳ 有效时间: 从 %v 到 %v\n", fmt.Sprintf("%.2f", data.TotalUsed), fmt.Sprintf("%.2f", data.TotalAvailable), t1.Format("2006-01-02 15:04:05"), t2.Format("2006-01-02 15:04:05"))
+ // 放入缓存
+ UserService.SetUserMode("system_balance", msg)
  return data, nil
 }

public/public.go

@@ -12,6 +12,7 @@ var UserService service.UserServiceInterface
 func InitSvc() {
  config.LoadConfig()
  UserService = service.NewUserService()
+ _, _ = GetBalance()
 }
 
 func FirstCheck(rmsg ReceiveMsg) bool {