logger.go

package log

import (
	"context"
	"encoding/binary"

	"go.opencensus.io/trace"
	"go.uber.org/zap"
)

type logger struct {
	logger *zap.SugaredLogger
}

func (l logger) WithField(key string, value interface{}) Logger {
	return logger{l.logger.With(zap.Any(key, value))}
}

func (l logger) WithFields(fields Fields) Logger {
	return logger{l.logger.With(fields)}
}

func (l logger) WithError(err error) Logger {
	return logger{l.logger.With(zap.Error(err))}
}

// WithTracing will take an OpenCensus trace and add log fields for Datadog.
// based on convertSpan in https://github.com/DataDog/opencensus-go-exporter-datadog/blob/master/span.go
// and https://docs.datadoghq.com/tracing/advanced/connect_logs_and_traces/?tab=go
func (l logger) WithTracing(ctx context.Context) (returnedLogger Logger) {
	if span := trace.FromContext(ctx); span != nil {
		traceID := span.SpanContext().TraceID
		spanID := span.SpanContext().SpanID
		returnedLogger = l.
			WithField("dd.trace_id", binary.BigEndian.Uint64(traceID[8:])).
			WithField("dd.span_id", binary.BigEndian.Uint64(spanID[:]))
	}
	return
}

func (l logger) Debugf(format string, args ...interface{}) {
	l.logger.Debugf(format, args...)
}

func (l logger) Infof(format string, args ...interface{}) {
	l.logger.Infof(format, args...)
}

func (l logger) Warnf(format string, args ...interface{}) {
	l.logger.Warnf(format, args...)
}

func (l logger) Warningf(format string, args ...interface{}) {
	l.logger.Warnf(format, args...)
}

func (l logger) Errorf(format string, args ...interface{}) {
	l.logger.Errorf(format, args...)
}

func (l logger) Fatalf(format string, args ...interface{}) {
	l.logger.Fatalf(format, args...)
}

func (l logger) Panicf(format string, args ...interface{}) {
	l.logger.Panicf(format, args...)
}

func (l logger) Debug(args ...interface{}) {
	l.logger.Debug(args...)
}

func (l logger) Info(args ...interface{}) {
	l.logger.Info(args...)
}

func (l logger) Warn(args ...interface{}) {
	l.logger.Warn(args...)
}

func (l logger) Warning(args ...interface{}) {
	l.logger.Warn(args...)
}

func (l logger) Error(args ...interface{}) {
	l.logger.Error(args...)
}

func (l logger) Fatal(args ...interface{}) {
	l.logger.Fatal(args...)
}

func (l logger) Panic(args ...interface{}) {
	l.logger.Panic(args...)
}

func (l logger) Sync() error {
	return l.logger.Sync()
}