keel/log/with.go

package log

import (
	"context"
	"fmt"
	"net"
	"net/http"
	"reflect"
	"strings"

	"go.opentelemetry.io/otel/trace"
	"go.uber.org/zap"

	httputils "github.com/foomo/keel/net/http"
)

func WithError(l *zap.Logger, err error) *zap.Logger {
	return l.With(FError(err), FErrorType(reflect.TypeOf(err).String()))
}

func WithServiceName(l *zap.Logger, name string) *zap.Logger {
	return l.With(FServiceName(name))
}

func WithTraceID(l *zap.Logger, ctx context.Context) *zap.Logger { //nolint:golint
	if spanCtx := trace.SpanContextFromContext(ctx); spanCtx.IsValid() {
		l = l.With(FTraceID(spanCtx.TraceID().String()))
	}
	return l
}

func WithHTTPRequest(l *zap.Logger, r *http.Request) *zap.Logger {
	fields := []zap.Field{
		FHTTPRequestContentLength(r.ContentLength),
		FHTTPMethod(r.Method),
		FHTTPUserAgent(r.UserAgent()),
		FHTTPTarget(r.RequestURI),
	}

	if r.Host != "" {
		fields = append(fields, FHTTPHost(r.Host))
	}
	if id := r.Header.Get(httputils.HeaderXRequestID); id != "" {
		fields = append(fields, FHTTPRequestID(id))
	}
	if r.TLS != nil {
		fields = append(fields, FHTTPScheme("https"))
	} else {
		fields = append(fields, FHTTPScheme("http"))
	}
	if r.ProtoMajor == 1 {
		fields = append(fields, FHTTPFlavor(fmt.Sprintf("1.%d", r.ProtoMinor)))
	} else if r.ProtoMajor == 2 {
		fields = append(fields, FHTTPFlavor("2"))
	}

	var clientIP string
	if value := r.Header.Get(httputils.HeaderXForwardedFor); value != "" {
		if i := strings.IndexAny(value, ", "); i > 0 {
			clientIP = value[:i]
		} else {
			clientIP = value
		}
	} else if value := r.Header.Get(httputils.HeaderXRealIP); value != "" {
		clientIP = value
	} else if value, _, err := net.SplitHostPort(r.RemoteAddr); err == nil {
		clientIP = value
	} else {
		clientIP = r.RemoteAddr
	}
	if clientIP != "" {
		fields = append(fields, FHTTPClientIP(clientIP))
	}

	if spanCtx := trace.SpanContextFromContext(r.Context()); spanCtx.IsValid() {
		fields = append(fields, FTraceID(spanCtx.TraceID().String()))
	}

	return l.With(fields...)
}