Ви не можете вибрати більше 25 тем Теми мають розпочинатися з літери або цифри, можуть містити дефіси (-) і не повинні перевищувати 35 символів.
 
 
 

162 рядки
4.0 KiB

  1. package logger
  2. import (
  3. "context"
  4. "fmt"
  5. "io"
  6. "log"
  7. "os"
  8. "path/filepath"
  9. "sync"
  10. "time"
  11. "github.com/QuantumNous/new-api/common"
  12. "github.com/QuantumNous/new-api/setting/operation_setting"
  13. "github.com/bytedance/gopkg/util/gopool"
  14. "github.com/gin-gonic/gin"
  15. )
  16. const (
  17. loggerINFO = "INFO"
  18. loggerWarn = "WARN"
  19. loggerError = "ERR"
  20. loggerDebug = "DEBUG"
  21. )
  22. const maxLogCount = 1000000
  23. var logCount int
  24. var setupLogLock sync.Mutex
  25. var setupLogWorking bool
  26. func SetupLogger() {
  27. defer func() {
  28. setupLogWorking = false
  29. }()
  30. if *common.LogDir != "" {
  31. ok := setupLogLock.TryLock()
  32. if !ok {
  33. log.Println("setup log is already working")
  34. return
  35. }
  36. defer func() {
  37. setupLogLock.Unlock()
  38. }()
  39. logPath := filepath.Join(*common.LogDir, fmt.Sprintf("oneapi-%s.log", time.Now().Format("20060102150405")))
  40. fd, err := os.OpenFile(logPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
  41. if err != nil {
  42. log.Fatal("failed to open log file")
  43. }
  44. gin.DefaultWriter = io.MultiWriter(os.Stdout, fd)
  45. gin.DefaultErrorWriter = io.MultiWriter(os.Stderr, fd)
  46. }
  47. }
  48. func LogInfo(ctx context.Context, msg string) {
  49. logHelper(ctx, loggerINFO, msg)
  50. }
  51. func LogWarn(ctx context.Context, msg string) {
  52. logHelper(ctx, loggerWarn, msg)
  53. }
  54. func LogError(ctx context.Context, msg string) {
  55. logHelper(ctx, loggerError, msg)
  56. }
  57. func LogDebug(ctx context.Context, msg string, args ...any) {
  58. if common.DebugEnabled {
  59. if len(args) > 0 {
  60. msg = fmt.Sprintf(msg, args...)
  61. }
  62. logHelper(ctx, loggerDebug, msg)
  63. }
  64. }
  65. func logHelper(ctx context.Context, level string, msg string) {
  66. writer := gin.DefaultErrorWriter
  67. if level == loggerINFO {
  68. writer = gin.DefaultWriter
  69. }
  70. id := "SYSTEM"
  71. if ctx != nil {
  72. if v := ctx.Value(common.RequestIdKey); v != nil {
  73. id = fmt.Sprintf("%v", v)
  74. }
  75. }
  76. now := time.Now()
  77. _, _ = fmt.Fprintf(writer, "[%s] %v | %s | %s \n", level, now.Format("2006/01/02 - 15:04:05"), id, msg)
  78. logCount++ // we don't need accurate count, so no lock here
  79. if logCount > maxLogCount && !setupLogWorking {
  80. logCount = 0
  81. setupLogWorking = true
  82. gopool.Go(func() {
  83. SetupLogger()
  84. })
  85. }
  86. }
  87. func LogQuota(quota int) string {
  88. // 新逻辑:根据额度展示类型输出
  89. q := float64(quota)
  90. switch operation_setting.GetQuotaDisplayType() {
  91. case operation_setting.QuotaDisplayTypeCNY:
  92. usd := q / common.QuotaPerUnit
  93. cny := usd * operation_setting.USDExchangeRate
  94. return fmt.Sprintf("¥%.6f 额度", cny)
  95. case operation_setting.QuotaDisplayTypeCustom:
  96. usd := q / common.QuotaPerUnit
  97. rate := operation_setting.GetGeneralSetting().CustomCurrencyExchangeRate
  98. symbol := operation_setting.GetGeneralSetting().CustomCurrencySymbol
  99. if symbol == "" {
  100. symbol = "¤"
  101. }
  102. if rate <= 0 {
  103. rate = 1
  104. }
  105. v := usd * rate
  106. return fmt.Sprintf("%s%.6f 额度", symbol, v)
  107. case operation_setting.QuotaDisplayTypeTokens:
  108. return fmt.Sprintf("%d 点额度", quota)
  109. default: // USD
  110. return fmt.Sprintf("$%.6f 额度", q/common.QuotaPerUnit)
  111. }
  112. }
  113. func FormatQuota(quota int) string {
  114. q := float64(quota)
  115. switch operation_setting.GetQuotaDisplayType() {
  116. case operation_setting.QuotaDisplayTypeCNY:
  117. usd := q / common.QuotaPerUnit
  118. cny := usd * operation_setting.USDExchangeRate
  119. return fmt.Sprintf("¥%.6f", cny)
  120. case operation_setting.QuotaDisplayTypeCustom:
  121. usd := q / common.QuotaPerUnit
  122. rate := operation_setting.GetGeneralSetting().CustomCurrencyExchangeRate
  123. symbol := operation_setting.GetGeneralSetting().CustomCurrencySymbol
  124. if symbol == "" {
  125. symbol = "¤"
  126. }
  127. if rate <= 0 {
  128. rate = 1
  129. }
  130. v := usd * rate
  131. return fmt.Sprintf("%s%.6f", symbol, v)
  132. case operation_setting.QuotaDisplayTypeTokens:
  133. return fmt.Sprintf("%d", quota)
  134. default:
  135. return fmt.Sprintf("$%.6f", q/common.QuotaPerUnit)
  136. }
  137. }
  138. // LogJson 仅供测试使用 only for test
  139. func LogJson(ctx context.Context, msg string, obj any) {
  140. jsonStr, err := common.Marshal(obj)
  141. if err != nil {
  142. LogError(ctx, fmt.Sprintf("json marshal failed: %s", err.Error()))
  143. return
  144. }
  145. LogDebug(ctx, fmt.Sprintf("%s | %s", msg, string(jsonStr)))
  146. }