Documentation

Overview

`grpc_ctxtags` adds a Tag object to the context that can be used by other middleware to add context about a request.

Request Context Tags

Tags describe information about the request, and can be set and used by other middleware, or handlers. Tags are used for logging and tracing of requests. Tags are populated both upwards, *and* downwards in the interceptor-handler stack.

You can automatically extract tags (in `grpc.request.<field_name>`) from request payloads.

For unary and server-streaming methods, pass in the `WithFieldExtractor` option. For client-streams and bidirectional-streams, you can use `WithFieldExtractorForInitialReq` which will extract the tags from the first message passed from client to server. Note the tags will not be modified for subsequent requests, so this option only makes sense when the initial message establishes the meta-data for the stream.

If a user doesn't use the interceptors that initialize the `Tags` object, all operations following from an `Extract(ctx)` will be no-ops. This is to ensure that code doesn't panic if the interceptors weren't used.

Tags fields are typed, and shallow and should follow the OpenTracing semantics convention: https://github.com/opentracing/specification/blob/master/semantic_conventions.md

Example (InitialisationWithOptions)

Example using WithFieldExtractorForInitialReq

Code:

opts := []grpc_ctxtags.Option{
	grpc_ctxtags.WithFieldExtractorForInitialReq(grpc_ctxtags.TagBasedRequestFieldExtractor("log_fields")),
}
_ = grpc.NewServer(
	grpc.StreamInterceptor(grpc_ctxtags.StreamServerInterceptor(opts...)),
	grpc.UnaryInterceptor(grpc_ctxtags.UnaryServerInterceptor(opts...)),
)
Example (Initialization)

Simple example of server initialization code, with data automatically populated from `log_fields` Golang tags.

Code:

opts := []grpc_ctxtags.Option{
	grpc_ctxtags.WithFieldExtractor(grpc_ctxtags.TagBasedRequestFieldExtractor("log_fields")),
}
_ = grpc.NewServer(
	grpc.StreamInterceptor(grpc_ctxtags.StreamServerInterceptor(opts...)),
	grpc.UnaryInterceptor(grpc_ctxtags.UnaryServerInterceptor(opts...)),
)

Index

Examples

Constants

This section is empty.

Variables

View Source
var (

	// NoopTags is a trivial, minimum overhead implementation of Tags for which all operations are no-ops.
	NoopTags = &noopTags{}
)

Functions

func CodeGenRequestFieldExtractor

func CodeGenRequestFieldExtractor(fullMethod string, req interface{}) map[string]interface{}

CodeGenRequestFieldExtractor is a function that relies on code-generated functions that export log fields from requests. These are usually coming from a protoc-plugin that generates additional information based on custom field options.

func SetInContext

func SetInContext(ctx context.Context, tags Tags) context.Context

func StreamServerInterceptor

func StreamServerInterceptor(opts ...Option) grpc.StreamServerInterceptor

StreamServerInterceptor returns a new streaming server interceptor that sets the values for request tags.

func UnaryServerInterceptor

func UnaryServerInterceptor(opts ...Option) grpc.UnaryServerInterceptor

UnaryServerInterceptor returns a new unary server interceptors that sets the values for request tags.

Types

type Option

type Option func(*options)

func WithFieldExtractor

func WithFieldExtractor(f RequestFieldExtractorFunc) Option

WithFieldExtractor customizes the function for extracting log fields from protobuf messages, for unary and server-streamed methods only.

func WithFieldExtractorForInitialReq

func WithFieldExtractorForInitialReq(f RequestFieldExtractorFunc) Option

WithFieldExtractorForInitialReq customizes the function for extracting log fields from protobuf messages, for all unary and streaming methods. For client-streams and bidirectional-streams, the tags will be extracted from the first message from the client.

type RequestFieldExtractorFunc

type RequestFieldExtractorFunc func(fullMethod string, req interface{}) map[string]interface{}

RequestFieldExtractorFunc is a user-provided function that extracts field information from a gRPC request. It is called from tags middleware on arrival of unary request or a server-stream request. Keys and values will be added to the context tags of the request. If there are no fields, you should return a nil.

func TagBasedRequestFieldExtractor

func TagBasedRequestFieldExtractor(tagName string) RequestFieldExtractorFunc

TagBasedRequestFieldExtractor is a function that relies on Go struct tags to export log fields from requests. These are usually coming from a protoc-plugin, such as Gogo protobuf.

message Metadata {
   repeated string tags = 1 [ (gogoproto.moretags) = "log_field:\"meta_tags\"" ];
}

The tagName is configurable using the tagName variable. Here it would be "log_field".

type Tags

type Tags interface {
	// Set sets the given key in the metadata tags.
	Set(key string, value interface{}) Tags
	// Has checks if the given key exists.
	Has(key string) bool
	// Values returns a map of key to values.
	// Do not modify the underlying map, please use Set instead.
	Values() map[string]interface{}
}

Tags is the interface used for storing request tags between Context calls. The default implementation is *not* thread safe, and should be handled only in the context of the request.

func Extract

func Extract(ctx context.Context) Tags

Extracts returns a pre-existing Tags object in the Context. If the context wasn't set in a tag interceptor, a no-op Tag storage is returned that will *not* be propagated in context.

func NewTags

func NewTags() Tags

Directories

Path Synopsis
logrus
zap