You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
898 B
Go
34 lines
898 B
Go
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform License
|
|
// that can be found in the LICENSE file.
|
|
|
|
package logger
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
type loggerKey struct{}
|
|
|
|
// WithContext returns a new context with the provided logger.
|
|
// Use in combination with logger.WithField for great effect.
|
|
func WithContext(ctx context.Context, logger Logger) context.Context {
|
|
return context.WithValue(ctx, loggerKey{}, logger)
|
|
}
|
|
|
|
// FromContext retrieves the current logger from the context.
|
|
func FromContext(ctx context.Context) Logger {
|
|
logger := ctx.Value(loggerKey{})
|
|
if logger == nil {
|
|
return Default
|
|
}
|
|
return logger.(Logger)
|
|
}
|
|
|
|
// FromRequest retrieves the current logger from the request. If no
|
|
// logger is available, the default logger is returned.
|
|
func FromRequest(r *http.Request) Logger {
|
|
return FromContext(r.Context())
|
|
}
|