mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-17 17:16:38 +08:00
150 lines
3.3 KiB
Go
150 lines
3.3 KiB
Go
package logger
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/songquanpeng/one-api/common/config"
|
|
"github.com/songquanpeng/one-api/common/helper"
|
|
)
|
|
|
|
type loggerLevel string
|
|
|
|
const (
|
|
loggerDEBUG loggerLevel = "DEBUG"
|
|
loggerINFO loggerLevel = "INFO"
|
|
loggerWarn loggerLevel = "WARN"
|
|
loggerError loggerLevel = "ERROR"
|
|
loggerFatal loggerLevel = "FATAL"
|
|
)
|
|
|
|
var setupLogOnce sync.Once
|
|
|
|
func SetupLogger() {
|
|
setupLogOnce.Do(func() {
|
|
if LogDir != "" {
|
|
var logPath string
|
|
if config.OnlyOneLogFile {
|
|
logPath = filepath.Join(LogDir, "oneapi.log")
|
|
} else {
|
|
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 SysLog(s string) {
|
|
logHelper(nil, loggerINFO, s)
|
|
}
|
|
|
|
func SysLogf(format string, a ...any) {
|
|
logHelper(nil, loggerINFO, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func SysError(s string) {
|
|
logHelper(nil, loggerError, s)
|
|
}
|
|
|
|
func SysErrorf(format string, a ...any) {
|
|
logHelper(nil, loggerError, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func Debug(ctx context.Context, msg string) {
|
|
if !config.DebugEnabled {
|
|
return
|
|
}
|
|
logHelper(ctx, loggerDEBUG, msg)
|
|
}
|
|
|
|
func Info(ctx context.Context, msg string) {
|
|
logHelper(ctx, loggerINFO, msg)
|
|
}
|
|
|
|
func Warn(ctx context.Context, msg string) {
|
|
logHelper(ctx, loggerWarn, msg)
|
|
}
|
|
|
|
func Error(ctx context.Context, msg string) {
|
|
logHelper(ctx, loggerError, msg)
|
|
}
|
|
|
|
func Debugf(ctx context.Context, format string, a ...any) {
|
|
logHelper(ctx, loggerDEBUG, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func Infof(ctx context.Context, format string, a ...any) {
|
|
logHelper(ctx, loggerINFO, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func Warnf(ctx context.Context, format string, a ...any) {
|
|
logHelper(ctx, loggerWarn, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func Errorf(ctx context.Context, format string, a ...any) {
|
|
logHelper(ctx, loggerError, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func FatalLog(s string) {
|
|
logHelper(nil, loggerFatal, s)
|
|
}
|
|
|
|
func FatalLogf(format string, a ...any) {
|
|
logHelper(nil, loggerFatal, fmt.Sprintf(format, a...))
|
|
}
|
|
|
|
func logHelper(ctx context.Context, level loggerLevel, msg string) {
|
|
writer := gin.DefaultErrorWriter
|
|
if level == loggerINFO {
|
|
writer = gin.DefaultWriter
|
|
}
|
|
var requestId string
|
|
if ctx != nil {
|
|
rawRequestId := helper.GetRequestID(ctx)
|
|
if rawRequestId != "" {
|
|
requestId = fmt.Sprintf(" | %s", rawRequestId)
|
|
}
|
|
}
|
|
lineInfo, funcName := getLineInfo()
|
|
now := time.Now()
|
|
_, _ = fmt.Fprintf(writer, "[%s] %v%s%s %s%s \n", level, now.Format("2006/01/02 - 15:04:05"), requestId, lineInfo, funcName, msg)
|
|
SetupLogger()
|
|
if level == loggerFatal {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func getLineInfo() (string, string) {
|
|
funcName := "[unknown] "
|
|
pc, file, line, ok := runtime.Caller(3)
|
|
if ok {
|
|
if fn := runtime.FuncForPC(pc); fn != nil {
|
|
parts := strings.Split(fn.Name(), ".")
|
|
funcName = "[" + parts[len(parts)-1] + "] "
|
|
}
|
|
} else {
|
|
file = "unknown"
|
|
line = 0
|
|
}
|
|
parts := strings.Split(file, "one-api/")
|
|
if len(parts) > 1 {
|
|
file = parts[1]
|
|
}
|
|
return fmt.Sprintf(" | %s:%d", file, line), funcName
|
|
}
|