micro-config-consul/logger.go

package consul

import (
	"bytes"
	"context"
	"fmt"
	"io"
	"log"
	"os"

	hclog "github.com/hashicorp/go-hclog"
	"go.unistack.org/micro/v4/logger"
)

// to check implementation
var _ hclog.Logger = &consulLogger{}

type consulLogger struct {
	logger logger.Logger
}

func (l *consulLogger) Name() string {
	return l.logger.String()
}

func (l *consulLogger) GetLevel() hclog.Level {
	switch l.logger.Options().Level {
	case logger.DebugLevel:
		return hclog.Debug
	case logger.TraceLevel:
		return hclog.Trace
	case logger.InfoLevel:
		return hclog.Info
	case logger.WarnLevel:
		return hclog.Warn
	case logger.ErrorLevel:
		return hclog.Error
	default:
		return hclog.Info
	}
}

func (l *consulLogger) With(args ...interface{}) hclog.Logger {
	fields := make([]interface{}, len(args))
	for i := 0; i < int(len(args)/2); i += 2 {
		fields = append(fields, fmt.Sprintf("%v", args[i]), args[i+1])
	}
	return &consulLogger{logger: l.logger.Clone(logger.WithAttrs(fields...))}
}

func (l *consulLogger) Debug(format string, msg ...interface{}) {
	l.logger.Debug(context.TODO(), fmt.Sprintf(format, msg...))
}

func (l *consulLogger) Error(format string, msg ...interface{}) {
	l.logger.Error(context.TODO(), fmt.Sprintf(format, msg...))
}

func (l *consulLogger) Info(format string, msg ...interface{}) {
	l.logger.Info(context.TODO(), fmt.Sprintf(format, msg...))
}

func (l *consulLogger) Warn(format string, msg ...interface{}) {
	l.logger.Warn(context.TODO(), fmt.Sprintf(format, msg...))
}

func (l *consulLogger) Trace(format string, msg ...interface{}) {
	l.logger.Trace(context.TODO(), fmt.Sprintf(format, msg...))
}

func (l *consulLogger) ImpliedArgs() []interface{} {
	fields := make([]interface{}, len(l.logger.Options().Attrs)*2)
	for k, v := range l.logger.Options().Attrs {
		fields = append(fields, k, v)
	}
	return fields
}

func (l *consulLogger) Named(name string) hclog.Logger {
	var newname string
	var oldname string

	fields := l.logger.Options().Attrs
	for i := 0; i < len(fields); i += 2 {
		if fields[i].(string) == "name" {
			oldname = fields[i+1].(string)
		}
	}

	if len(oldname) > 0 {
		newname = fmt.Sprintf("%s.%s", oldname, name)
	} else {
		newname = name
	}
	return &consulLogger{logger: l.logger.Clone(logger.WithAttrs("name", newname))}
}

func (l *consulLogger) ResetNamed(name string) hclog.Logger {
	return &consulLogger{logger: l.logger.Clone(logger.WithAttrs("name", name))}
}

func (l *consulLogger) SetLevel(level hclog.Level) {
	// TODO: add logic when logger.Logger supports this method
}

func (l *consulLogger) StandardLogger(opts *hclog.StandardLoggerOptions) *log.Logger {
	// TODO: add logic
	return log.New(os.Stderr, "", log.Ldate|log.Ltime|log.Llongfile|log.LUTC)
}

func (l *consulLogger) StandardWriter(opts *hclog.StandardLoggerOptions) io.Writer {
	buf := bytes.NewBuffer(nil)
	return buf
}

func (l *consulLogger) IsDebug() bool {
	return l.logger.V(logger.DebugLevel)
}

func (l *consulLogger) IsError() bool {
	return l.logger.V(logger.ErrorLevel)
}

func (l *consulLogger) IsInfo() bool {
	return l.logger.V(logger.InfoLevel)
}

func (l *consulLogger) IsTrace() bool {
	return l.logger.V(logger.TraceLevel)
}

func (l *consulLogger) IsWarn() bool {
	return l.logger.V(logger.WarnLevel)
}

func (l *consulLogger) Log(level hclog.Level, msg string, args ...interface{}) {
	switch level {
	case hclog.Trace:
		l.Trace(msg, args...)
	case hclog.Debug:
		l.Debug(msg, args...)
	case hclog.Info:
		l.Info(msg, args...)
	case hclog.Warn:
		l.Warn(msg, args...)
	case hclog.Error:
		l.Error(msg, args...)
	case hclog.NoLevel:
		l.Info(msg, args...)
	}
}