chore: init bj_power monorepo (dashboard/mes/wms/wms_client/workstation), clear subproject git metadata and align naming to folder names
This commit is contained in:
SunYF
@@ -0,0 +1,15 @@
|
||||
package log
|
||||
|
||||
type LogConf struct {
|
||||
Level string `json:",default=INFO"` // 日志级别: DEBUG, INFO, WARN, ERROR
|
||||
Format string `json:",default=text"` // 日志格式: text, json
|
||||
TimeFormat string `json:",default=2006-01-02 15:04:05"` // 时间格式
|
||||
Filename string `json:",default=spcm.log"` // 日志文件名
|
||||
MaxSize int `json:",default=100"` // 单个日志文件最大大小(MB)
|
||||
MaxBackups int `json:",default=10"` // 保留的旧日志文件数量
|
||||
MaxAge int `json:",default=30"` // 日志文件保留天数
|
||||
Compress bool `json:",default=true"` // 是否压缩旧日志文件
|
||||
Console bool `json:",default=false"` // 是否同时输出到控制台
|
||||
AddSource bool `json:",default=false"` // 是否在日志中添加文件名和行号
|
||||
Stats bool `json:",default=false"`
|
||||
}
|
||||
@@ -0,0 +1,46 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
"runtime"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
type EntLogger struct {
|
||||
*slog.Logger
|
||||
}
|
||||
|
||||
func (l *EntLogger) DebugWithContext(ctx context.Context, v ...any) {
|
||||
var pcs [1]uintptr
|
||||
// skip [runtime.Callers, this function, this function's caller]
|
||||
|
||||
var pc uintptr
|
||||
for i := 6; i < 20; i++ {
|
||||
runtime.Callers(i, pcs[:])
|
||||
frames := runtime.CallersFrames(pcs[:])
|
||||
frame, _ := frames.Next()
|
||||
//fmt.Println(i, frame.Function, frame.File, frame.Line)
|
||||
if !strings.Contains(frame.File, "automatic_changer/ent") && !strings.Contains(frame.File, "entgo.io") {
|
||||
pc = pcs[0]
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
r := slog.NewRecord(time.Now(), slog.LevelDebug, fmt.Sprint(v...), pc)
|
||||
/*var attrs []slog.Attr
|
||||
spanCtx := sdktrace.SpanContextFromContext(ctx)
|
||||
if spanCtx.HasTraceID() {
|
||||
attrs = append(attrs, slog.String("trace", spanCtx.TraceID().String()))
|
||||
}
|
||||
if spanCtx.HasSpanID() {
|
||||
attrs = append(attrs, slog.String("span", spanCtx.SpanID().String()))
|
||||
}
|
||||
r.AddAttrs(attrs...)
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}*/
|
||||
_ = l.Handler().Handle(ctx, r)
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"log/slog"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
|
||||
sdktrace "go.opentelemetry.io/otel/trace"
|
||||
)
|
||||
|
||||
var projectRoot = getCompileTimeModuleRoot()
|
||||
|
||||
func getCompileTimeModuleRoot() string {
|
||||
// 获取当前函数的调用信息
|
||||
_, file, _, ok := runtime.Caller(0)
|
||||
if !ok {
|
||||
return ""
|
||||
}
|
||||
// file 就是编译时的绝对路径
|
||||
return filepath.Dir(filepath.Dir(file))
|
||||
}
|
||||
|
||||
func isProjectFile(file string) bool {
|
||||
// 检查文件是否在编译时的模块根目录下
|
||||
if projectRoot != "" {
|
||||
if rel, err := filepath.Rel(projectRoot, file); err == nil {
|
||||
if !strings.HasPrefix(rel, "..") {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 其他回退逻辑...
|
||||
return false
|
||||
}
|
||||
|
||||
func NewContextHandler(w io.Writer, encoding string, opts *slog.HandlerOptions) slog.Handler {
|
||||
if strings.EqualFold(encoding, "json") {
|
||||
return &ContextHandler{
|
||||
Handler: slog.NewJSONHandler(w, opts),
|
||||
}
|
||||
}
|
||||
return &ContextHandler{
|
||||
Handler: slog.NewTextHandler(w, opts),
|
||||
}
|
||||
}
|
||||
|
||||
type ContextHandler struct {
|
||||
slog.Handler
|
||||
encoding string
|
||||
}
|
||||
|
||||
func (h *ContextHandler) Handle(ctx context.Context, r slog.Record) error {
|
||||
var attrs []slog.Attr
|
||||
spanCtx := sdktrace.SpanContextFromContext(ctx)
|
||||
if spanCtx.HasTraceID() {
|
||||
attrs = append(attrs, slog.String("trace", spanCtx.TraceID().String()))
|
||||
}
|
||||
if spanCtx.HasSpanID() {
|
||||
attrs = append(attrs, slog.String("span", spanCtx.SpanID().String()))
|
||||
}
|
||||
r.AddAttrs(attrs...)
|
||||
return h.Handler.Handle(ctx, r)
|
||||
}
|
||||
@@ -0,0 +1,271 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"log/slog"
|
||||
"os"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/zeromicro/go-zero/core/logx"
|
||||
"go.opentelemetry.io/otel/trace"
|
||||
"gopkg.in/natefinch/lumberjack.v2"
|
||||
)
|
||||
|
||||
var workdir, _ = filepath.Abs(filepath.Dir(os.Args[0]))
|
||||
|
||||
type Logger struct {
|
||||
*slog.Logger
|
||||
}
|
||||
|
||||
func SetDefault(logger *Logger) {
|
||||
slog.SetDefault(logger.Logger)
|
||||
}
|
||||
|
||||
func NewLogger(conf LogConf) *Logger {
|
||||
var log *slog.Logger
|
||||
|
||||
logLevel := slog.LevelInfo
|
||||
switch strings.ToLower(conf.Level) {
|
||||
case "error":
|
||||
logLevel = slog.LevelError
|
||||
case "warn":
|
||||
logLevel = slog.LevelWarn
|
||||
case "info":
|
||||
logLevel = slog.LevelInfo
|
||||
case "debug":
|
||||
logLevel = slog.LevelDebug
|
||||
}
|
||||
|
||||
if !path.IsAbs(conf.Filename) {
|
||||
conf.Filename = filepath.Join(workdir, conf.Filename)
|
||||
}
|
||||
|
||||
// �匧予��𠧧�亙�嚗𡁏�憭拐�銝芣�隞塚��賢��澆� bj_power_mes-YYYY-MM-DD.log
|
||||
// 瘥誩予����?lumberjack �匧之撠𧶏�MaxSize嚗㗇��剁�靽萘� MaxBackups 銝芸�隞?
|
||||
dw := newDailyWriter(conf)
|
||||
|
||||
var writer io.Writer = dw
|
||||
if conf.Console {
|
||||
writer = io.MultiWriter(writer, os.Stdout)
|
||||
}
|
||||
log = slog.New(NewContextHandler(writer, conf.Format, &slog.HandlerOptions{
|
||||
Level: logLevel,
|
||||
AddSource: conf.AddSource,
|
||||
ReplaceAttr: func(_ []string, attr slog.Attr) slog.Attr {
|
||||
switch attr.Key {
|
||||
case "caller":
|
||||
return slog.Attr{}
|
||||
case slog.SourceKey:
|
||||
if source, ok := attr.Value.Any().(*slog.Source); ok {
|
||||
if source.File == "" {
|
||||
return slog.Attr{}
|
||||
}
|
||||
return slog.String(slog.SourceKey, " "+filepath.Base(source.File)+fmt.Sprintf(":%d", source.Line))
|
||||
}
|
||||
case slog.TimeKey:
|
||||
return slog.String(slog.TimeKey, attr.Value.Time().Format(conf.TimeFormat))
|
||||
}
|
||||
return attr
|
||||
},
|
||||
}))
|
||||
return &Logger{
|
||||
Logger: log,
|
||||
}
|
||||
}
|
||||
|
||||
func (l *Logger) Alert(v any) {
|
||||
l.Logger.Info("%v", v)
|
||||
}
|
||||
|
||||
func (l *Logger) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (l *Logger) buildAttrs(fields []logx.LogField) []slog.Attr {
|
||||
attrs := make([]slog.Attr, 0, len(fields))
|
||||
for _, field := range fields {
|
||||
attrs = append(attrs, slog.Attr{
|
||||
Key: field.Key,
|
||||
Value: slog.AnyValue(field.Value),
|
||||
})
|
||||
}
|
||||
return attrs
|
||||
}
|
||||
|
||||
func (l *Logger) DebugCtx(ctx context.Context, v ...any) {
|
||||
if !l.Enabled(ctx, slog.LevelDebug) {
|
||||
return
|
||||
}
|
||||
var pcs [1]uintptr
|
||||
// skip [runtime.Callers, this function, this function's caller]
|
||||
runtime.Callers(2, pcs[:])
|
||||
pc := pcs[0]
|
||||
|
||||
r := slog.NewRecord(time.Now(), slog.LevelDebug, fmt.Sprint(v...), pc)
|
||||
var attrs []slog.Attr
|
||||
spanCtx := trace.SpanContextFromContext(ctx)
|
||||
if spanCtx.HasTraceID() {
|
||||
attrs = append(attrs, slog.String("trace", spanCtx.TraceID().String()))
|
||||
}
|
||||
if spanCtx.HasSpanID() {
|
||||
attrs = append(attrs, slog.String("span", spanCtx.SpanID().String()))
|
||||
}
|
||||
r.AddAttrs(attrs...)
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
_ = l.Handler().Handle(ctx, r)
|
||||
}
|
||||
|
||||
func (l *Logger) log(ctx context.Context, level slog.Level, msg string, args ...any) {
|
||||
if !l.Enabled(ctx, level) {
|
||||
return
|
||||
}
|
||||
var pc uintptr
|
||||
var pcs [1]uintptr
|
||||
// skip [runtime.Callers, this function, this function's caller]
|
||||
runtime.Callers(5, pcs[:])
|
||||
pc = pcs[0]
|
||||
|
||||
r := slog.NewRecord(time.Now(), level, msg, pc)
|
||||
r.Add(args...)
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
_ = l.Handler().Handle(ctx, r)
|
||||
}
|
||||
func (l *Logger) logAttrs(ctx context.Context, level slog.Level, msg string, attrs ...slog.Attr) {
|
||||
if !l.Enabled(ctx, level) {
|
||||
return
|
||||
}
|
||||
var pcs [1]uintptr
|
||||
// skip [runtime.Callers, this function, this function's caller]
|
||||
runtime.Callers(5, pcs[:])
|
||||
pc := pcs[0]
|
||||
|
||||
r := slog.NewRecord(time.Now(), level, msg, pc)
|
||||
r.AddAttrs(attrs...)
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
_ = l.Handler().Handle(ctx, r)
|
||||
}
|
||||
|
||||
func (l *Logger) Debug(v any, fields ...logx.LogField) {
|
||||
attrs := l.buildAttrs(fields)
|
||||
l.logAttrs(context.Background(), slog.LevelDebug, fmt.Sprintf("%v", v), attrs...)
|
||||
}
|
||||
|
||||
func (l *Logger) Error(v any, fields ...logx.LogField) {
|
||||
attrs := l.buildAttrs(fields)
|
||||
l.logAttrs(context.Background(), slog.LevelError, fmt.Sprintf("%v", v), attrs...)
|
||||
}
|
||||
|
||||
// skipHttpPollLog 餈�誘擃㗛�頧株砭�亙�嚗𡁜�蝡航蔭霂Ｘ𦻖����𣂼� GET嚗㗇�蝘埝㺭甈∪�撅𧶏�
|
||||
// �䭾��靝遠�潛凒�乩腺撘���蹱�雿頣�POST/PUT/DELETE嚗劐��?2xx �躰秤�滚�靽萘��?
|
||||
func skipHttpPollLog(v any) bool {
|
||||
msg, ok := v.(string)
|
||||
if !ok || !strings.HasPrefix(msg, "[HTTP]") {
|
||||
return false
|
||||
}
|
||||
// �澆�: [HTTP] 200 - GET /api/v1/jobs - 127.0.0.1:49864 - UA
|
||||
parts := strings.SplitN(msg, " - ", 3)
|
||||
if len(parts) < 3 {
|
||||
return false
|
||||
}
|
||||
code := strings.TrimSpace(strings.TrimPrefix(parts[0], "[HTTP]"))
|
||||
if len(code) < 1 || code[0] != '2' {
|
||||
return false
|
||||
}
|
||||
return strings.HasPrefix(parts[1], "GET ")
|
||||
}
|
||||
|
||||
func (l *Logger) Info(v any, fields ...logx.LogField) {
|
||||
if skipHttpPollLog(v) {
|
||||
return
|
||||
}
|
||||
attrs := l.buildAttrs(fields)
|
||||
l.logAttrs(context.Background(), slog.LevelInfo, fmt.Sprintf("%v", v), attrs...)
|
||||
}
|
||||
|
||||
func (l *Logger) Severe(v any) {
|
||||
l.log(context.Background(), slog.LevelError, fmt.Sprintf("%v", v))
|
||||
}
|
||||
|
||||
func (l *Logger) Slow(v any, fields ...logx.LogField) {
|
||||
if skipHttpPollLog(v) {
|
||||
return
|
||||
}
|
||||
attrs := l.buildAttrs(fields)
|
||||
l.logAttrs(context.Background(), slog.LevelWarn, fmt.Sprintf("%v", v), attrs...)
|
||||
}
|
||||
|
||||
func (l *Logger) Stack(v any) {
|
||||
l.log(context.Background(), slog.LevelError, fmt.Sprintf("%v", v))
|
||||
}
|
||||
|
||||
func (l *Logger) Stat(v any, fields ...logx.LogField) {
|
||||
attrs := l.buildAttrs(fields)
|
||||
l.logAttrs(context.Background(), slog.LevelInfo, fmt.Sprintf("%v", v), attrs...)
|
||||
}
|
||||
|
||||
// dailyWriter �匧予��𠧧�亙��坔��具�?
|
||||
// 瘥𤩺活 Write �嗆��亙��齿𠯫���頝典予�嗉䌊�典��Ｘ�隞嗚�?
|
||||
// 瘥誩予��眏 lumberjack �匧之撠𤩺��剁�靽萘� MaxBackups 銝芸�隞賬�?
|
||||
type dailyWriter struct {
|
||||
mu sync.Mutex
|
||||
conf LogConf
|
||||
baseDir string
|
||||
baseName string // 銝滚鉄�拙��滨��箇���辣�㵪�憒?"bj_power_mes"
|
||||
ext string // �拙��㵪�憒?".log"
|
||||
currentDay string
|
||||
lj *lumberjack.Logger
|
||||
}
|
||||
|
||||
func newDailyWriter(conf LogConf) *dailyWriter {
|
||||
dir := filepath.Dir(conf.Filename)
|
||||
ext := filepath.Ext(conf.Filename)
|
||||
base := strings.TrimSuffix(filepath.Base(conf.Filename), ext)
|
||||
// 蝖桐��亙��桀�摮睃銁
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
// �𥕦遣�桀�憭梯揖�嗅����啣虾�扯���辣�桀�嚗屸��滨�摨誩鍳�典仃韐?
|
||||
slog.Error("log: �𥕦遣�亙��桀�憭梯揖嚗�����啣極雿𦦵𤌍敶?, "dir", dir, "error", err)
|
||||
dir = workdir
|
||||
}
|
||||
return &dailyWriter{
|
||||
conf: conf,
|
||||
baseDir: dir,
|
||||
baseName: base,
|
||||
ext: ext,
|
||||
}
|
||||
}
|
||||
|
||||
func (w *dailyWriter) Write(p []byte) (n int, err error) {
|
||||
w.mu.Lock()
|
||||
defer w.mu.Unlock()
|
||||
|
||||
day := time.Now().Format("2006-01-02")
|
||||
if day != w.currentDay {
|
||||
// 頝典予嚗𡁜��剜唂 writer嚗��撱箸鰵��辣
|
||||
if w.lj != nil {
|
||||
w.lj.Close()
|
||||
}
|
||||
filename := filepath.Join(w.baseDir, w.baseName+"-"+day+w.ext)
|
||||
w.lj = &lumberjack.Logger{
|
||||
Filename: filename,
|
||||
MaxSize: w.conf.MaxSize,
|
||||
MaxBackups: w.conf.MaxBackups,
|
||||
MaxAge: w.conf.MaxAge,
|
||||
Compress: w.conf.Compress,
|
||||
}
|
||||
w.currentDay = day
|
||||
}
|
||||
return w.lj.Write(p)
|
||||
}
|
||||
Reference in New Issue
Block a user