1bfffa975e
* Define and use an interface for logging. This allows alternative implementations to use whatever logging package they want. Currently the interface just mimicks what logrus already implements, with minor modifications to not rely on logrus itself. I think the interface is pretty solid in terms of logging implementations being able to do what they need to. * Make tracing interface to coalesce logging/tracing Allows us to share data between the tracer and the logger so we can simplify log/trace handling wher we generally want data to go both places.
59 lines
2.0 KiB
Go
59 lines
2.0 KiB
Go
// Package trace abstracts virtual-kubelet's tracing capabilties into a set of
|
|
// interfaces.
|
|
// While this does allow consumers to use whatever tracing library they want,
|
|
// the primary goal is to share logging data between the configured logger and
|
|
// tracing spans instead of duplicating calls.
|
|
package trace
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/virtual-kubelet/virtual-kubelet/log"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// Status is an alias to opencensus's trace status.
|
|
// The main reason we use this instead of implementing our own is library re-use,
|
|
// namely for converting an error to a tracing status.
|
|
// In the future this may be defined completely in this package.
|
|
type Status = trace.Status
|
|
|
|
// Tracer is the interface used for creating a tracing span
|
|
type Tracer interface {
|
|
// StartSpan starts a new span. The span details are emebedded into the returned
|
|
// context
|
|
StartSpan(context.Context, string) (context.Context, Span)
|
|
}
|
|
|
|
var (
|
|
// T is the Tracer to use this should be initialized before starting up
|
|
// virtual-kubelet
|
|
T Tracer = nopTracer{}
|
|
)
|
|
|
|
// StartSpan starts a span from the configured default tracer
|
|
func StartSpan(ctx context.Context, name string) (context.Context, Span) {
|
|
ctx, span := T.StartSpan(ctx, name)
|
|
ctx = log.WithLogger(ctx, span.Logger())
|
|
return ctx, span
|
|
}
|
|
|
|
// Span encapsulates a tracing event
|
|
type Span interface {
|
|
End()
|
|
SetStatus(Status)
|
|
|
|
// WithField and WithFields adds attributes to an entire span
|
|
//
|
|
// This interface is a bit weird, but allows us to manage loggers in the context
|
|
// It is expected that implementations set `log.WithLogger` so the logger stored
|
|
// in the context is updated with the new fields.
|
|
WithField(context.Context, string, interface{}) context.Context
|
|
WithFields(context.Context, log.Fields) context.Context
|
|
|
|
// Logger is used to log individual entries.
|
|
// Calls to functions like `WithField` and `WithFields` on the logger should
|
|
// not affect the rest of the span but rather individual entries.
|
|
Logger() log.Logger
|
|
}
|