在处理程序之后访问 HTTP 请求上下文

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《在处理程序之后访问 HTTP 请求上下文》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

在我的日志记录中间件（链中的第一个）中，我需要访问一些在链下游的某些身份验证中间件中编写的上下文，并且仅在处理程序本身执行之后。

旁注：需要首先调用日志记录中间件，因为我需要记录请求的持续时间，包括在中间件中花费的时间。此外，当权限不足时，身份验证中间件能够中止请求。在这种情况下，我还需要记录失败的请求。

我的问题是，从 http.request 指针读取上下文不会返回我期望的身份验证数据。请参阅下面的示例：

package main

import (
    "context"
    "fmt"
    "net/http"
    "time"
)

const (
    contextkeyusername = "username"
)

func authmiddleware(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        ctx := r.context()
        ctx = context.withvalue(ctx, contextkeyusername, "user123")
        next.servehttp(w, r.withcontext(ctx))
    })
}

func logmiddleware(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        defer func(start time.time) {
            ctx := r.context()
            username := ctx.value(contextkeyusername)
            if username != nil {
                fmt.printf("user %s has accessed %s, took %d\n", username,
                    r.url.path, time.since(start).milliseconds())
            } else {
                fmt.printf("annonyous has accessed %s, took %d\n",
                    r.url.path, time.since(start).milliseconds())
            }
        }(time.now())
        next.servehttp(w, r)
    })
}

func welcome(w http.responsewriter, r *http.request) {
    ctx := r.context()
    username := ctx.value(contextkeyusername)
    if username != nil {
        fmt.fprintf(w, fmt.sprintf("hello %s", username.(string)))
    } else {
        fmt.fprintf(w, "hello")
    }
}

func main() {
    mux := http.newservemux()
    mux.handlefunc("/welcome", welcome)
    chain := logmiddleware(authmiddleware(mux))
    http.listenandserve(":5050", chain)
}

虽然对 127.0.0.1:5050/welcome 的 get 请求确实返回了预期的字符串 hello user123，但日志的输出为：

annonyous has accessed /welcome, took 0

由于请求作为指针传递，我预计在执行延迟时，上下文将包含预期的 username 值。

我在这里缺少什么？

正确答案

WithContext 返回请求的浅表副本，即 authmiddleware 创建的请求与 logmiddleware 从中读取上下文的请求不是同一请求。

您可以让根中间件（在本例中为 logmiddleware）创建 context-with-value 和浅请求副本，但存储一个非零指针，而不是普通字符串 在上下文中，然后让 authmiddleware 使用指针间接分配指针指向的值，然后 logmiddleware 在 next 退出后，可以取消引用该指针来访问该值。

为了避免令人不快的取消引用，您可以使用指向带有字符串字段的结构的指针，而不是指向字符串的指针。

type ctxkey uint8

const userkey ctxkey = 0

type user struct{ name string }

func logmiddleware(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        u := new(user)
        r = r.withcontext(context.withvalue(r.context(), userkey, u))

        defer func(start time.time) {
            if u.name != "" {
                fmt.printf("user %s has accessed %s, took %s\n", u.name, r.url.path, time.since(start))
            } else {
                fmt.printf("annonyous has accessed %s, took %s\n", r.url.path, time.since(start))
            }
        }(time.now())

        next.servehttp(w, r)
    })
}

func authmiddleware(next http.handler) http.handler {
    return http.handlerfunc(func(w http.responsewriter, r *http.request) {
        if u, ok := r.context().value(userkey).(*user); ok {
            u.name = "user123"
        }
        next.servehttp(w, r)
    })
}

func welcome(w http.ResponseWriter, r *http.Request) {
    if u, ok := r.Context().Value(userKey).(*user); ok && u.name != "" {
        fmt.Fprintf(w, "hello %s", u.name)
    } else {
        fmt.Fprintf(w, "hello")
    }
}

https://go.dev/play/p/N7vmjQ7iLM1

本篇关于《在处理程序之后访问 HTTP 请求上下文》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！