- working on new cmd and logger setup

2 files changed, 120 insertions, 0 deletions

diff --git a/plugins/logger/config.go b/plugins/logger/config.go
new file mode 100644
index 00000000..12badade
--- /dev/null
+++ b/plugins/logger/config.go
@@ -0,0 +1,9 @@
+package logger
+
+type Config struct {
+	Squash   bool
+	Channels map[string]LoggerConfig
+}
+
+type LoggerConfig struct {
+}
diff --git a/plugins/logger/zap_logger.go b/plugins/logger/zap_logger.go
new file mode 100644
index 00000000..747cbf77
--- /dev/null
+++ b/plugins/logger/zap_logger.go
@@ -0,0 +1,111 @@
+package logger
+
+import (
+	"github.com/fatih/color"
+	"github.com/spiral/endure"
+	"github.com/spiral/roadrunner/v2/plugins/config"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"strings"
+	"time"
+)
+
+// ServiceName declares service name.
+const ServiceName = "logs"
+
+type LogFactory interface {
+	// GlobalLogger returns global log instance.
+	GlobalLogger() *zap.Logger
+
+	// NamedLogger returns logger dedicated to the specific channel. Similar to Named() but also reads the core params.
+	NamedLogger(name string) *zap.Logger
+}
+
+// ZapLogger manages zap logger.
+type ZapLogger struct {
+	base *zap.Logger
+	cfg  Config
+}
+
+func (z *ZapLogger) Init(cfg config.Provider) (err error) {
+	err = cfg.UnmarshalKey(ServiceName, &z.cfg)
+	if err != nil {
+		return err
+	}
+
+	if z.base == nil {
+		cfg := zap.Config{
+			Level:    zap.NewAtomicLevelAt(zap.DebugLevel),
+			Encoding: "console",
+			EncoderConfig: zapcore.EncoderConfig{
+				MessageKey: "message",
+				LevelKey:   "level",
+				TimeKey:    "time",
+				NameKey:    "name",
+				EncodeName: func(s string, enc zapcore.PrimitiveArrayEncoder) {
+					if len(s) < 12 {
+						s = s + strings.Repeat(" ", 12-len(s))
+					}
+
+					enc.AppendString(color.HiGreenString(s))
+				},
+				EncodeLevel: func(level zapcore.Level, enc zapcore.PrimitiveArrayEncoder) {
+					switch level {
+					case zapcore.DebugLevel:
+						enc.AppendString(color.HiWhiteString(level.CapitalString()))
+					case zapcore.InfoLevel:
+						enc.AppendString(color.HiCyanString(level.CapitalString()))
+					case zapcore.WarnLevel:
+						enc.AppendString(color.HiYellowString(level.CapitalString()))
+					case zapcore.ErrorLevel, zapcore.DPanicLevel:
+						enc.AppendString(color.HiRedString(level.CapitalString()))
+					case zapcore.PanicLevel, zapcore.FatalLevel:
+						enc.AppendString(color.HiMagentaString(level.CapitalString()))
+					}
+				},
+				EncodeTime: func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
+					enc.AppendString(t.UTC().Format("2006/01/02 15:04:05"))
+				},
+				EncodeCaller: zapcore.ShortCallerEncoder,
+			},
+			OutputPaths:      []string{"stderr"},
+			ErrorOutputPaths: []string{"stderr"},
+		}
+
+		z.base, err = cfg.Build()
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+// GlobalLogger returns global log instance.
+func (z *ZapLogger) GlobalLogger() *zap.Logger {
+	return z.base
+}
+
+// NamedLogger returns logger dedicated to the specific channel. Similar to Named() but also reads the core params.
+func (z *ZapLogger) NamedLogger(name string) *zap.Logger {
+	// todo: automatically configure
+	return z.base.Named(name)
+}
+
+// Provides declares factory methods.
+func (z *ZapLogger) Provides() []interface{} {
+	return []interface{}{
+		z.DefaultLogger,
+		z.AllocateLogger,
+	}
+}
+
+// AllocateLogger allocates logger for the service.
+func (z *ZapLogger) AllocateLogger(n endure.Named) (*zap.Logger, error) {
+	return z.NamedLogger(n.Name()), nil
+}
+
+// DefaultLogger returns default logger.
+func (z *ZapLogger) DefaultLogger() (*zap.Logger, error) {
+	return z.GlobalLogger(), nil
+}