mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 14:46:38 +08:00
71171c63f5
* ♻️ refactor: move file directory * ♻️ refactor: move file directory * ♻️ refactor: support multiple config methods * 🔥 del: remove unused code * 💩 refactor: Refactor channel management and synchronization * 💄 improve: add channel website * ✨ feat: allow recording 0 consumption
135 lines
2.7 KiB
Go
135 lines
2.7 KiB
Go
package common
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"os"
|
||
"path/filepath"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
const (
|
||
loggerINFO = "INFO"
|
||
loggerWarn = "WARN"
|
||
loggerError = "ERR"
|
||
)
|
||
const (
|
||
RequestIdKey = "X-Oneapi-Request-Id"
|
||
)
|
||
|
||
const maxLogCount = 1000000
|
||
|
||
var logCount int
|
||
var setupLogLock sync.Mutex
|
||
var setupLogWorking bool
|
||
|
||
var defaultLogDir = "./logs"
|
||
|
||
func SetupLogger() {
|
||
logDir := getLogDir()
|
||
if logDir == "" {
|
||
return
|
||
}
|
||
|
||
ok := setupLogLock.TryLock()
|
||
if !ok {
|
||
log.Println("setup log is already working")
|
||
return
|
||
}
|
||
defer func() {
|
||
setupLogLock.Unlock()
|
||
setupLogWorking = false
|
||
}()
|
||
logPath := filepath.Join(logDir, fmt.Sprintf("oneapi-%s.log", time.Now().Format("20060102")))
|
||
fd, err := os.OpenFile(logPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||
if err != nil {
|
||
log.Fatal("failed to open log file")
|
||
}
|
||
gin.DefaultWriter = io.MultiWriter(os.Stdout, fd)
|
||
gin.DefaultErrorWriter = io.MultiWriter(os.Stderr, fd)
|
||
}
|
||
|
||
func getLogDir() string {
|
||
logDir := viper.GetString("log_dir")
|
||
if logDir == "" {
|
||
logDir = defaultLogDir
|
||
}
|
||
|
||
var err error
|
||
logDir, err = filepath.Abs(viper.GetString("log_dir"))
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
return ""
|
||
}
|
||
|
||
if !IsFileExist(logDir) {
|
||
err = os.Mkdir(logDir, 0777)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
return ""
|
||
}
|
||
}
|
||
|
||
return logDir
|
||
}
|
||
|
||
func SysLog(s string) {
|
||
t := time.Now()
|
||
_, _ = fmt.Fprintf(gin.DefaultWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s)
|
||
}
|
||
|
||
func SysError(s string) {
|
||
t := time.Now()
|
||
_, _ = fmt.Fprintf(gin.DefaultErrorWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s)
|
||
}
|
||
|
||
func LogInfo(ctx context.Context, msg string) {
|
||
logHelper(ctx, loggerINFO, msg)
|
||
}
|
||
|
||
func LogWarn(ctx context.Context, msg string) {
|
||
logHelper(ctx, loggerWarn, msg)
|
||
}
|
||
|
||
func LogError(ctx context.Context, msg string) {
|
||
logHelper(ctx, loggerError, msg)
|
||
}
|
||
|
||
func logHelper(ctx context.Context, level string, msg string) {
|
||
writer := gin.DefaultErrorWriter
|
||
if level == loggerINFO {
|
||
writer = gin.DefaultWriter
|
||
}
|
||
id := ctx.Value(RequestIdKey)
|
||
now := time.Now()
|
||
_, _ = fmt.Fprintf(writer, "[%s] %v | %s | %s \n", level, now.Format("2006/01/02 - 15:04:05"), id, msg)
|
||
logCount++ // we don't need accurate count, so no lock here
|
||
if logCount > maxLogCount && !setupLogWorking {
|
||
logCount = 0
|
||
setupLogWorking = true
|
||
go func() {
|
||
SetupLogger()
|
||
}()
|
||
}
|
||
}
|
||
|
||
func FatalLog(v ...any) {
|
||
t := time.Now()
|
||
_, _ = fmt.Fprintf(gin.DefaultErrorWriter, "[FATAL] %v | %v \n", t.Format("2006/01/02 - 15:04:05"), v)
|
||
os.Exit(1)
|
||
}
|
||
|
||
func LogQuota(quota int) string {
|
||
if DisplayInCurrencyEnabled {
|
||
return fmt.Sprintf("＄%.6f 额度", float64(quota)/QuotaPerUnit)
|
||
} else {
|
||
return fmt.Sprintf("%d 点额度", quota)
|
||
}
|
||
}
|