gRPC go：在服务级别应用拦截器

Question

gRPC拦截器通过ServerOption应用于服务器。参见doc 如何在服务级别应用拦截器。例如，我可能只需要将身份验证器拦截器应用于受保护的服务。可以吗？

Answer 1

除了eric先前的回答，您还可以执行以下操作：

type key int

const (
    sessionIDKey key = iota
)

var (
    needTobeAllocate = [1]string{"allocate"}
)

func run() {
    // server option 
    opts := []grpc.ServerOption{}
    opts = append(opts, grpc.UnaryInterceptor(unaryInterceptor))
}


func unaryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
    if checkAllocate(info.FullMethod) {
        ctx = context.WithValue(ctx, sessionIDKey, "testsession")   
    }

    return handler(ctx, req)
}

func checkAllocate(method string) bool {
    for _, v := range needTobeAllocate {
        if strings.Contains(strings.ToLower(method), v) {
            return true
        }
    }

    return false
}

Answer 2

gRPC Go拦截器接收StreamServerInfo（或UnaryServerInfo）参数，该参数除其他外包含调用的服务和方法名称。您可以使用它来根据服务/方法过滤拦截器行为。