代码拉取完成,页面将自动刷新
package middleware
import (
"context"
"crypto/rand"
"encoding/base64"
"io"
"time"
"github.com/golang/protobuf/proto"
"goa.design/goa/middleware"
"google.golang.org/grpc"
"google.golang.org/grpc/metadata"
"google.golang.org/grpc/status"
)
// UnaryServerLog returns a middleware that logs incoming gRPC requests
// and outgoing responses. The middleware uses the request ID set by
// the RequestID middleware or creates a short unique request ID if
// missing for each incoming request and logs it with the request and
// corresponding response details.
//
// The middleware logs the incoming requests gRPC method. It also logs the
// response gRPC status code, message length (in bytes), and timing information.
func UnaryServerLog(l middleware.Logger) grpc.UnaryServerInterceptor {
return grpc.UnaryServerInterceptor(func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
var reqID string
{
md, ok := metadata.FromIncomingContext(ctx)
if !ok {
md = metadata.MD{}
}
reqID = MetadataValue(md, RequestIDMetadataKey)
if reqID == "" {
reqID = shortID()
}
}
started := time.Now()
// before executing rpc
l.Log("id", reqID,
"method", info.FullMethod,
"bytes", messageLength(req))
// invoke rpc
resp, err = handler(ctx, req)
// after executing rpc
s, _ := status.FromError(err)
l.Log("id", reqID,
"status", s.Code(),
"bytes", messageLength(resp),
"time", time.Since(started).String())
return resp, err
})
}
// StreamServerLog returns a middleware that logs incoming streaming gRPC
// requests and responses. The middleware uses the request ID set by the
// RequestID middleware or creates a short unique request ID if missing for
// each incoming request and logs it with the request and corresponding
// response details.
func StreamServerLog(l middleware.Logger) grpc.StreamServerInterceptor {
return grpc.StreamServerInterceptor(func(srv interface{}, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
var reqID string
{
md, ok := metadata.FromIncomingContext(ss.Context())
if !ok {
md = metadata.MD{}
}
reqID = MetadataValue(md, RequestIDMetadataKey)
if reqID == "" {
reqID = shortID()
}
}
started := time.Now()
// before executing rpc
l.Log("id", reqID,
"method", info.FullMethod,
"msg", "started stream")
// invoke rpc
err := handler(srv, ss)
// after executing rpc
s, _ := status.FromError(err)
l.Log("id", reqID,
"status", s.Code(),
"msg", "completed stream",
"time", time.Since(started).String())
return err
})
}
// shortID produces a " unique" 6 bytes long string.
// Do not use as a reliable way to get unique IDs, instead use for things like logging.
func shortID() string {
b := make([]byte, 6)
io.ReadFull(rand.Reader, b)
return base64.RawURLEncoding.EncodeToString(b)
}
func messageLength(msg interface{}) int64 {
var length int64
{
if m, ok := msg.(proto.Message); ok {
length = int64(proto.Size(m))
}
}
return length
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。