本篇文章给大家带来的内容是关于context的上下文值传递的介绍(代码示例),有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助。
本文开始将针对context的用法进行系统化讨论,在这里你将能够在工作中合理使用context解决一些比较棘手的问题。
context处理超时处理之外还可以用来保存数据,当你需要在多个上下文传递时传递数据,那么本文提到的知识可以排上用场。
示例代码
示例代码为一个简单的http服务,流程是登录之后会跳转首页,首页通过guard中间件进行鉴权。当然,示例代码未做其他诸如连接数据库之类的处理,这不是本文的重点。
守卫函数读取cookie之后将cookie值写入context并向下传递,在整个请求中可以说是“透明”的。当访问到需要保护的接口时检测到没有提供cookie,则直接终端请求,否则通过r.withcontext将username的值存入cookie,避免的业务接口直接读取cookie的弊端。因为如果后期更改鉴权算法的话,业务代码可以不用更改,直接更改中间件即可。
package mainimport ( context fmt log net/http time)func main() { mux := http.newservemux() mux.handlefunc(/, guard(home)) mux.handlefunc(/login, login) log.fatal(http.listenandserve(:8080, mux))}// 登录func login(w http.responsewriter, r *http.request) { if r.url.query().get(username) != root { http.error(w, http.statustext(401), 401) return } cookie := &http.cookie{name: username, value: root, expires: time.now().add(time.hour)} http.setcookie(w, cookie) http.redirect(w, r, /, 302)}func home(w http.responsewriter, r *http.request) { username := r.context().value(username) fmt.fprintf(w, welcome login: %s, username.(string))}// 守卫func guard(handlefunc http.handlerfunc) http.handlerfunc { return func(w http.responsewriter, r *http.request) { // check username log.printf(%s - %s\n, r.method, r.requesturi) cookie, err := r.cookie(username) if err != nil || cookie == nil { // 如果username为空直接拦截 http.error(w, http.statustext(401), 401) return } handlefunc(w, r.withcontext(context.withvalue(r.context(), username, cookie.value))) }}
本文的代码就这么多,内容也很少,希望大家能好好用上这个利器。
以上就是context的上下文值传递的介绍(代码示例)的详细内容。
