时间戳

首页 > 解决方案 > 即使 gin 的 Context 结构将 Header 作为 Map 也无法获取 Map 的键

http - 即使 gin 的 Context 结构将 Header 作为 Map 也无法获取 Map 的键

问题描述

我无法从 Gin 的 Context.Header（Golang 的 gin-gonic http/rest 框架）字段中获取所有键，即使 Header 被定义为 Map “type Header map[string][]string”（Header 来自net\http\Header.go，request 来自 net\hhtp\request.go，Context 来自 Gin-gonic 的包），奇怪的是即使在编译/构建时，Visual Studio 代码也不让我调用/ 在这个映射类型的 Header 上使用“MapKeys()”方法（鉴于 Golang 是静态类型语言，并且它在编译时已经知道它的数据类型）。

  1. 我需要将所有的HTTP headers复制到Logger中，这样当我记录任何消息时，我就可以放入相应的请求Headers。
  2. 而且我还需要将所有 HTTP 标头从 HTTP 传递到 gRPC 调用，以满足端到端调用可追溯性的需要。
func (l *Logger) InfoCtx(ctx *gin.Context, md metadata.MD) *zerolog.Event {
    headerName := "X-Request-Id" // Read all the headers from the ENV file
    // mapping := make(map[string]string)
    // mapping[headerName] = ctx.Request.Header[headerName][0]
    event := l.Logger.Info()
    
    // ctx.Request.Header ==> Even though this is a "map" type, 
    // which is known at the compilation time itself, 
    // it doesn't let me use any map functions.

    if ctx != nil && len(ctx.Request.Header[headerName]) > 0 {
        event = event.Str(headerName, ctx.Request.Header[headerName][0])

    } else if md != nil {
        // some other gRPC metadata context handling (not relevant for this question)
        
    }
    return event
}

能否请你帮忙？

标题对象

请求对象使用 Header 字段

显示标题是地图类型

标签： httpgoheadergrpcgo-gin

解决方案

同时对我有用的另一种方法是，

  1. 我创建了一个结构，其中包含“HttpHeadersMap map[string][]string”
type CommonContext struct {
    HttpHeadersMap map[string][]string
    RequestContext context.Context

    GrpcMDHeadersMap map[string][]string
}
  1. 将 Gin 的“ctx.Request.Header”分配给“HttpHeadersMap map[string][]string”
func GetCommonCtx(ctx *gin.Context, md metadata.MD) CommonContext {
    var commonContext CommonContext
    if ctx != nil {
        // event = event.Str(headerName, ctx.Request.Header[headerName][0])
        commonContext = CommonContext{ // don't return address, use valye type
            HttpHeadersMap: ctx.Request.Header,
            RequestContext: ctx.Request.Context(),
        }
    }
...
}

然后在“gRPC 拦截器（仅显示示例用例）”中，我可以使用“HttpHeadersMap”常规方式作为“headersMapVal.MapKeys()”来迭代 Map 键。

func clientInterceptor(
    ctx context.Context,
    method string,
    req interface{},
    reply interface{},
    cc *grpc.ClientConn,
    invoker grpc.UnaryInvoker,
    opts ...grpc.CallOption,
) error {
    start := time.Now()
    commonCtx := commonContext.GetCommonCtx(nil, metadata.MD{})
    if callOpt, ok := opts[0].(CustomDataCallOption); ok {

        headersMapVal := reflect.ValueOf(callOpt).FieldByName("HeadersMap")
        newMap := make(map[string]string)
        // allKeysMap := make(map[string]string)
        for _, key := range headersMapVal.MapKeys() {
            // fmt.Printf("headersMapVal.MapKeys(), e %v", e)
            // c_key := e.Convert(headersMapValueIndirectStr.Type().Key())
            keyValue := headersMapVal.MapIndex(key)

...
...
}

推荐阅读