首页> Java> java教程> 正文

context的上下文值传递的介绍(代码示例)

不言
发布: 2018-10-18 15:25:41
转载
3140 人浏览过

本篇文章给大家带来的内容是关于context的上下文值传递的介绍(代码示例),有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助。

本文开始将针对context的用法进行系统化讨论,在这里你将能够在工作中合理使用context解决一些比较棘手的问题。

context处理超时处理之外还可以用来保存数据,当你需要在多个上下文传递时传递数据,那么本文提到的知识可以排上用场。

示例代码

示例代码为一个简单的http服务,流程是登录之后会跳转首页,首页通过guard中间件进行鉴权。当然,示例代码未做其他诸如连接数据库之类的处理,这不是本文的重点。
守卫函数读取cookie之后将cookie值写入context并向下传递,在整个请求中可以说是“透明”的。当访问到需要保护的接口时检测到没有提供cookie,则直接终端请求,否则通过r.WithContext将username的值存入cookie,避免的业务接口直接读取cookie的弊端。因为如果后期更改鉴权算法的话,业务代码可以不用更改,直接更改中间件即可。

package main import ( "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的上下文值传递的介绍(代码示例)的详细内容。更多信息请关注PHP中文网其他相关文章!

相关标签:
来源:segmentfault.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责声明 Sitemap
PHP中文网:公益在线PHP培训,帮助PHP学习者快速成长!