125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
|
package loggerx
|
||
|
|
|
||
|
|
// author：黄新云
|
||
|
|
// lastTime:2023年6月30日21:28:04
|
||
|
|
// desc: 日志封装类
|
||
|
|
|
||
|
|
// 优化方向
|
||
|
|
// 可以写盘的时候添加缓存，但是有个难点就是如果采用缓存的方式必须保证这个是最后关闭的，如果不是则退出的时候会丢失日志（写在缓存里还没刷盘）
|
||
|
|
|
||
|
|
// TODO:自动清除过期日志
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"runtime"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
// "sync_log/global"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 需要实现io.Writer接口
|
||
|
|
type Logger struct {
|
||
|
|
file *os.File
|
||
|
|
fileName string
|
||
|
|
mu sync.Mutex
|
||
|
|
// channel string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewLogger(opts ...Option) *Logger {
|
||
|
|
opt := defaultOptions()
|
||
|
|
for _, apply := range opts {
|
||
|
|
apply(&opt)
|
||
|
|
}
|
||
|
|
|
||
|
|
l := &Logger{}
|
||
|
|
|
||
|
|
// 打开文件
|
||
|
|
err := l.createNewFile(true)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
log.SetOutput(l)
|
||
|
|
log.SetFlags(log.LstdFlags | log.Llongfile | log.Lmicroseconds) // log.Lshortfile | log.LUTC
|
||
|
|
|
||
|
|
// 保存Gin日志写入到文件+控制台
|
||
|
|
gin.DefaultWriter = io.MultiWriter(l, os.Stdout)
|
||
|
|
gin.DefaultErrorWriter = io.MultiWriter(l, os.Stdout)
|
||
|
|
|
||
|
|
// 赋值前缀
|
||
|
|
if prefix != "" {
|
||
|
|
log.SetPrefix(fmt.Sprintf("[%s]", prefix))
|
||
|
|
}
|
||
|
|
return l
|
||
|
|
}
|
||
|
|
|
||
|
|
// func (l *Logger) Channel(ch string) (r *Logger) {
|
||
|
|
// rr := l
|
||
|
|
// rr.channel = ch
|
||
|
|
// return rr
|
||
|
|
// }
|
||
|
|
|
||
|
|
// 超时删除
|
||
|
|
|
||
|
|
func (l *Logger) Write(b []byte) (n int, err error) {
|
||
|
|
|
||
|
|
if l.file == nil {
|
||
|
|
// 新建一个file连接
|
||
|
|
l.createNewFile(false)
|
||
|
|
}
|
||
|
|
if l.fileName != nowFileName() {
|
||
|
|
l.createNewFile(false)
|
||
|
|
}
|
||
|
|
|
||
|
|
n, err = l.file.Write(b)
|
||
|
|
if err == nil && n < len(b) {
|
||
|
|
err = io.ErrShortWrite
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
// 强制更新
|
||
|
|
l.createNewFile(true)
|
||
|
|
}
|
||
|
|
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Logger) Info(ctx context.Context, v ...any) {
|
||
|
|
l.logger(ctx, "info", v...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Logger) Infof(ctx context.Context, format string, v ...any) {
|
||
|
|
s := fmt.Sprintf(format, v...)
|
||
|
|
l.logger(ctx, "info", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Logger) Error(ctx context.Context, v ...any) {
|
||
|
|
l.logger(ctx, "error", v...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *Logger) Errorf(ctx context.Context, format string, v ...any) {
|
||
|
|
s := fmt.Sprintf(format, v...)
|
||
|
|
l.logger(ctx, "error", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// 添加固定的内容
|
||
|
|
// func (l *Logger) ContextWithFields(ctx context.Context, v ...any) {
|
||
|
|
// l.logger(ctx, "add", v...)
|
||
|
|
// }
|
||
|
|
// func (l *Logger) Field(key,val string) {
|
||
|
|
// l.logger(nil, "add", key,val)
|
||
|
|
// }
|
||
|
|
|
||
|
|
func getGID() string {
|
||
|
|
b := make([]byte, 64)
|
||
|
|
b = b[:runtime.Stack(b, false)]
|
||
|
|
b = bytes.TrimPrefix(b, []byte("goroutine "))
|
||
|
|
b = b[:bytes.IndexByte(b, ' ')]
|
||
|
|
return string(b)
|
||
|
|
}
|