yml2213
47b0dd5a49
- OPEN_API_DEBUG 记录鉴权、查单、推送业务字段与 req_id - LOG_FILE 默认 logs/app.log，标准日志/Gin/GORM 同时输出控制台与文件
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package applog
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
// Setup 将标准库 log、Gin 请求日志同时写到控制台与文件。
|
|
// logFile 为空则仅控制台。返回文件句柄供进程结束时 Close（可为 nil）。
|
|
func Setup(logFile string) (*os.File, error) {
|
|
writers := []io.Writer{os.Stdout}
|
|
var f *os.File
|
|
if logFile != "" {
|
|
dir := filepath.Dir(logFile)
|
|
if err := os.MkdirAll(dir, 0o755); err != nil {
|
|
return nil, err
|
|
}
|
|
var err error
|
|
f, err = os.OpenFile(logFile, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
writers = append(writers, f)
|
|
}
|
|
|
|
mw := io.MultiWriter(writers...)
|
|
log.SetOutput(mw)
|
|
log.SetFlags(log.LstdFlags | log.Lmicroseconds)
|
|
|
|
// Gin access / recovery 日志
|
|
gin.DefaultWriter = mw
|
|
gin.DefaultErrorWriter = mw
|
|
|
|
if logFile != "" {
|
|
abs, _ := filepath.Abs(logFile)
|
|
log.Printf("[applog] 日志同时输出到控制台与文件: %s", abs)
|
|
}
|
|
return f, nil
|
|
}
|
|
|
|
// NewGormLogger 返回写到当前 log 输出（已含 MultiWriter）的 GORM 日志
|
|
func NewGormLogger(mode string) logger.Interface {
|
|
level := logger.Info
|
|
if mode == "release" {
|
|
level = logger.Warn
|
|
}
|
|
return logger.New(
|
|
log.New(log.Writer(), "\r\n", log.LstdFlags),
|
|
logger.Config{
|
|
SlowThreshold: 200 * time.Millisecond,
|
|
LogLevel: level,
|
|
IgnoreRecordNotFoundError: true,
|
|
Colorful: false,
|
|
},
|
|
)
|
|
}
|