mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type logLevel int
|
|
|
|
const (
|
|
logLevelDebug logLevel = iota
|
|
logLevelInfo
|
|
logLevelWarn
|
|
logLevelError
|
|
)
|
|
|
|
var currentLogLevel = logLevelInfo
|
|
var currentLogLevelName = "info"
|
|
var commonLogWriter io.Writer = os.Stdout
|
|
var errorLogWriter io.Writer = os.Stderr
|
|
|
|
func configureGinWriters() {
|
|
if shouldLog(logLevelDebug) {
|
|
gin.DefaultWriter = commonLogWriter
|
|
} else {
|
|
gin.DefaultWriter = io.Discard
|
|
}
|
|
gin.DefaultErrorWriter = errorLogWriter
|
|
}
|
|
|
|
func SetLogLevel(level string) {
|
|
normalized := strings.TrimSpace(strings.ToLower(level))
|
|
switch normalized {
|
|
case "debug":
|
|
currentLogLevel = logLevelDebug
|
|
currentLogLevelName = "debug"
|
|
case "warn", "warning":
|
|
currentLogLevel = logLevelWarn
|
|
currentLogLevelName = "warn"
|
|
case "error":
|
|
currentLogLevel = logLevelError
|
|
currentLogLevelName = "error"
|
|
default:
|
|
currentLogLevel = logLevelInfo
|
|
currentLogLevelName = "info"
|
|
}
|
|
configureGinWriters()
|
|
}
|
|
|
|
func GetLogLevel() string {
|
|
return currentLogLevelName
|
|
}
|
|
|
|
func shouldLog(level logLevel) bool {
|
|
return level >= currentLogLevel
|
|
}
|
|
|
|
func SetupGinLog() {
|
|
if *LogDir != "" {
|
|
commonLogPath := filepath.Join(*LogDir, "common.log")
|
|
errorLogPath := filepath.Join(*LogDir, "error.log")
|
|
commonFd, err := os.OpenFile(commonLogPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
log.Fatal("failed to open log file")
|
|
}
|
|
errorFd, err := os.OpenFile(errorLogPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
log.Fatal("failed to open log file")
|
|
}
|
|
commonLogWriter = io.MultiWriter(os.Stdout, commonFd)
|
|
errorLogWriter = io.MultiWriter(os.Stderr, errorFd)
|
|
}
|
|
configureGinWriters()
|
|
}
|
|
|
|
func SysLog(s string) {
|
|
if !shouldLog(logLevelInfo) {
|
|
return
|
|
}
|
|
t := time.Now()
|
|
_, _ = fmt.Fprintf(commonLogWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s)
|
|
}
|
|
|
|
func SysError(s string) {
|
|
if !shouldLog(logLevelError) {
|
|
return
|
|
}
|
|
t := time.Now()
|
|
_, _ = fmt.Fprintf(errorLogWriter, "[SYS] %v | %s \n", t.Format("2006/01/02 - 15:04:05"), s)
|
|
}
|
|
|
|
func FatalLog(v ...any) {
|
|
t := time.Now()
|
|
_, _ = fmt.Fprintf(errorLogWriter, "[FATAL] %v | %v \n", t.Format("2006/01/02 - 15:04:05"), v)
|
|
os.Exit(1)
|
|
}
|