mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
* Adding a way to inject a request ID It is very useful to associate a request ID to each incoming request, this change allows to provide a function to do that via Server Option. The change comes with a default function which will generate a new request ID. The request ID is put in the request context along with a common logger which always logs the request-id We add gRPC interceptors to the server so it can get the request ID out of the gRPC metadata and put it in the common logger stored in the context so as all the log lines using the common logger from the context will have the request ID logged
36 lines
1.3 KiB
Go
36 lines
1.3 KiB
Go
// Copyright 2017 Michal Witkowski. All Rights Reserved.
|
|
// See LICENSE for licensing terms.
|
|
|
|
package grpc_logging
|
|
|
|
import (
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
)
|
|
|
|
// ErrorToCode function determines the error code of an error
|
|
// This makes using custom errors with grpc middleware easier
|
|
type ErrorToCode func(err error) codes.Code
|
|
|
|
func DefaultErrorToCode(err error) codes.Code {
|
|
return grpc.Code(err)
|
|
}
|
|
|
|
// Decider function defines rules for suppressing any interceptor logs
|
|
type Decider func(fullMethodName string, err error) bool
|
|
|
|
// DefaultDeciderMethod is the default implementation of decider to see if you should log the call
|
|
// by default this if always true so all calls are logged
|
|
func DefaultDeciderMethod(fullMethodName string, err error) bool {
|
|
return true
|
|
}
|
|
|
|
// ServerPayloadLoggingDecider is a user-provided function for deciding whether to log the server-side
|
|
// request/response payloads
|
|
type ServerPayloadLoggingDecider func(ctx context.Context, fullMethodName string, servingObject interface{}) bool
|
|
|
|
// ClientPayloadLoggingDecider is a user-provided function for deciding whether to log the client-side
|
|
// request/response payloads
|
|
type ClientPayloadLoggingDecider func(ctx context.Context, fullMethodName string) bool
|