🎉logger的v2版本,暂未支持环境变量配置

loki.go

@@ -0,0 +1,70 @@
+package zap_logger
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	customencoder "gitee.ltd/lxh/logger/v2/encoder"
+	"gitee.ltd/lxh/logger/v2/write"
+	"github.com/grafana/loki-client-go/loki"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+type LokiLogger struct {
+	Encoder     string `json:"encoder"`
+	Level       string `json:"level"`
+	Enable      bool   `yaml:"enable"`
+	Host        string `yaml:"host"`
+	Port        int    `yaml:"port"`
+	Source      string `yaml:"source"`
+	Service     string `yaml:"service"`
+	Job         string `yaml:"job"`
+	Environment string `yaml:"environment"`
+
+	lokiClient *loki.Client
+}
+
+func newLokiLogger(lokiConf *Loki) *LokiLogger {
+	lokiConfig := &LokiLogger{
+		Enable:      lokiConf.Enable,
+		Encoder:     lokiConf.Encoder,
+		Level:       lokiConf.Level,
+		Host:        lokiConf.Host,
+		Port:        lokiConf.Port,
+		Source:      lokiConf.Source,
+		Service:     lokiConf.Service,
+		Job:         lokiConf.Job,
+		Environment: lokiConf.Environment,
+	}
+
+	lc, err := loki.NewWithDefault(fmt.Sprintf("%v:%v/loki/api/v1/push", lokiConfig.Host, lokiConfig.Port))
+	if err != nil {
+		return nil
+	}
+
+	lokiConfig.lokiClient = lc
+	return lokiConfig
+}
+
+func (l *LokiLogger) Init() zapcore.Core {
+	encoderConfig := zap.NewProductionEncoderConfig()
+	encoderConfig.TimeKey = "time"
+	encoderConfig.EncodeTime = func(time time.Time, encoder zapcore.PrimitiveArrayEncoder) {
+		encoder.AppendString(time.Format("2006-01-02 15:04:05.000"))
+	}
+
+	writer := zapcore.AddSync(write.NewLokiWriter(l.lokiClient, l.Encoder, l.Job, l.Source, l.Environment))
+	var encoder zapcore.Encoder
+	switch l.Encoder {
+	case "json":
+		encoder = customencoder.NewJsonEncoder(encoderConfig)
+	case "console":
+		encoder = zapcore.NewConsoleEncoder(encoderConfig)
+	}
+
+	level, _ := zapcore.ParseLevel(strings.ToLower(l.Level))
+
+	return zapcore.NewCore(encoder, writer, level)
+}