如何自定义Go中AuthFromMD的认证请求头名称？

我之前也遇到过类似的需求——需要支持多个自定义认证头，而不是局限于默认的authorization。问题的核心在于AuthFromMD是metautils包封装好的工具函数，它硬编码了默认的头名称，所以没法直接通过修改它来支持自定义头。不过我们可以绕开这个封装，直接操作元数据对象来实现需求。

下面给你两个实用的方案：

方案一：自定义提取单个认证头的函数 #

直接复用metautils提取元数据的能力，自己实现一个支持指定头名称的提取函数，完全替代AuthFromMD：

import (
    "context"
    "fmt"
    "strings"

    "github.com/grpc-ecosystem/go-grpc-middleware/util/metautils"
)

// CustomAuthFromMD 从指定的请求头中提取认证令牌，支持自定义头名称
func CustomAuthFromMD(ctx context.Context, headerName string) (string, error) {
    // gRPC元数据的key默认是小写存储的，统一转小写避免大小写问题
    headerName = strings.ToLower(headerName)
    md := metautils.ExtractIncoming(ctx)
    authValues := md.Get(headerName)

    if len(authValues) == 0 {
        return "", fmt.Errorf("请求中缺少%s头", headerName)
    }

    // 如果你的认证格式是类似"Bearer <token>"的，可以保留这段前缀校验；不需要的话直接返回authValues[0]即可
    parts := strings.SplitN(authValues[0], " ", 2)
    if len(parts) != 2 {
        return "", fmt.Errorf("%s头格式无效，应为'Bearer <token>'", headerName)
    }

    return parts[1], nil
}

使用的时候，你可以传入任意头名称，比如：

token, err := CustomAuthFromMD(ctx, "x-custom-auth")
// 或者用默认的authorization
token, err := CustomAuthFromMD(ctx, "authorization")

方案二：支持同时检查多个认证头 #

如果需要同时兼容多个认证头（比如同时支持authorization和x-internal-auth），可以写一个遍历检查的函数：

// MultiAuthFromMD 依次尝试从多个指定头中提取有效令牌
func MultiAuthFromMD(ctx context.Context, headerNames ...string) (string, error) {
    md := metautils.ExtractIncoming(ctx)

    for _, name := range headerNames {
        name = strings.ToLower(name)
        authValues := md.Get(name)
        if len(authValues) == 0 {
            continue
        }

        parts := strings.SplitN(authValues[0], " ", 2)
        if len(parts) == 2 {
            return parts[1], nil
        }
    }

    return "", fmt.Errorf("未找到有效认证头，支持的头名称：%v", headerNames)
}

调用示例：

token, err := MultiAuthFromMD(ctx, "authorization", "x-custom-auth", "x-internal-token")

为什么原来的AuthFromMD没法直接修改？ #

查看AuthFromMD的源码会发现，它内部直接使用了包级别的headerAuthorize常量（值为"authorization"），没有提供参数让我们自定义这个值。所以与其去修改源码或者做复杂的封装，不如直接基于metautils.ExtractIncoming返回的MD对象自己实现提取逻辑，这样更灵活也更可控。

内容的提问来源于stack exchange，提问作者venkatReddi