我非常清楚如何将数据从处理程序传递到它包装的处理程序，但是是否有惯用的方法从包装的处理程序中取回一些东西？这是一个激励示例:我有一个 accessLogHandler 和一个 authHandler。 accessLogHandler 记录每个 http 请求，包括时间和其他请求信息，例如当前登录的用户 ID(如果有的话)。 authHandler 适用于需要登录用户的路由，当用户未登录时为 403。我想用 authHandler<>，并用 accessLogHandler 包装我的所有路由。如果用户已登录，我希望我的 accessLogHandler 将用户信息与访问日志一起记录。

现在，我想出了一个我不喜欢的解决方案。我将添加代码，然后解释我的一些问题。

// Log the timings of each request optionally including user data
// if there is a logged in user
func accessLogHandler(fn http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        accessLog := newAccessLog()
        ctx := context.WithValue(r.Context(), accessLogKey, accessLog)
        fn.ServeHTTP(w, r.WithContext(ctx))

        // Logs the http access, ommit user info if not set
        accessLog.Log()
    }
}

// pull some junk off the request/cookies/whatever and check if somebody is logged in
func authHandler(fn http.HandlerFunc) http.HandlerFunc {
    return func (w http.ResponseWriter, r *http.Request) {
        //Do some authorization
        user, err := auth(r)
        if err != nil{
            //No userId, don't set anything on the accesslogger
            w.WriteHeader(http.StatusForbiddend)
            return
        }
        //Success a user is logged in, let's make sure the access logger knows
        acessLog := r.Context().Value(accessLogKey).(*AccessLog)
        accessLog.Set("userID", user.ID)
        fn.ServeHTTP(w, r)
    }
}

基本上，我在这里所做的是在 accessLogHandler 和 authHandler 中将一个 accessLog 结构附加到我的上下文中从上下文中读取 accessLog 并调用 accessLog.Set 以通知记录器用户 ID 存在。

我不喜欢这种方法的一些地方:

1. context 是不可变的，但我在其上粘贴了一个可变结构，并在下游的其他地方改变了所述结构。感觉像黑客。
2. 我的 authHandler 现在对 accessLog 包具有包级依赖性，因为我正在对 *AccessLog 进行类型断言。
3. 理想情况下，我的 authHandler 可以通过某种方式将用户数据通知请求堆栈的任何部分，而无需将其自身与所述部分紧密耦合。

最佳答案

Context 本身是一个接口(interface)，因此您可以在记录器中间件中创建一个新的记录器上下文，它具有获取所需行为所需的方法。

像这样:

type Logger struct{}

func (l *Logger) SetLogField(key string, value interface{}) {// set log field }
func (l *Logger) Log(){// log request}

type LoggerCtx struct {
    context.Context
    *Logger
}

func newAccessLog() *Logger {
    return &Logger{}
}

func accessLogHandler(fn http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // create new logger context
        ctx := &LoggerCtx{}
        ctx.Context = r.Context()
        ctx.Logger = newAccessLog()

        fn.ServeHTTP(w, r.WithContext(ctx))

        // Logs the http access, ommit user info if not set
        ctx.Log()
    }
}

// pull some junk off the request/cookies/whatever and check if somebody is logged in
func authHandler(fn http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        //Do some authorization
        user, err := auth(r)
        if err != nil {
            //No userId, don't set anything on the accesslogger
            w.WriteHeader(http.StatusForbiddend)
            return
        }

        //Success a user is logged in, let's make sure the access logger knows
        ctx := r.Context()

        // this could be moved - here for clarity
        type setLog interface {
            SetLogField(string, interface{})
        }

        if lctx, ok := ctx.(setLog); ok {
            lctx.SetLogField("userID", user.ID)
        }

        fn.ServeHTTP(w, r.WithContext(ctx))
    }
}

关于go - 将数据传递给父中间件的方法？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/42401197/