使用Golang的Context管理上下文的方法
golang 1.7版本中context库被很多标准库的模块所使用,比如net/http和os的一些模块中,利用这些原生模块,我们就不需要自己再写上下文的管理器了,直接调用函数接口即可实现,利用context我们可以实现一些比如请求的声明周期内的变量管理,执行一些操作的超时等等。
保存上下文对象
这里我们通过一个简单的例子来看一下如何使用context的特性来实现上下文的对象保存,这里我们写了一个简单的http server,具有登录和退出,状态检查路由(检查用户是否登录)
func main(){ mux:=http.newservemux() mux.handlefunc("/",statushandler) mux.handlefunc("/login",loginhandler) mux.handlefunc("/logout",logouthandler) contextedmux:=addcontextsupport(mux) log.fatal(http.listenandserve(":8080",contextedmux)) }
其中的addcontextsupport是一个中间件,用来绑定一个context到原来的handler中,所有的请求都必须先经过该中间件后才能进入各自的路由处理中。具体的实现代码如下:
func addcontextsupport(next http.handler)http.handler{ return http.handlerfunc(func(w http.responsewriter, r *http.request) { log.println(r.method, "-", r.requesturi) cookie, _ := r.cookie("username") if cookie != nil { ctx := context.withvalue(r.context(), "username", cookie.value) // withcontext returns a shallow copy of r with its context changed // to ctx. the provided ctx must be non-nil. next.servehttp(w, r.withcontext(ctx)) } else { next.servehttp(w, r) } }) }
该中间件可以打印每次请求的方法和请求的url,然后获得请求的cookie值,如果cookie为空的话则继续传递到对应的路由处理函数中,否则保存cookie的值到context, 注意这里的context()是request对象的方法, 将创建一个新的上下文对象(如果context为空),context.withvalue()函数将key和value保存在新的上下文对象中并返回该对象。
其余的路由处理函数代码如下, 分别用来保存cookie的登录路由loginhandler(),还有删除cookie的退出路由处理函数logouthandler()。
func loginhandler(w http.responsewriter,r *http.request){ expitation := time.now().add(24*time.hour) var username string if username=r.url.query().get("username");username==""{ username = "guest" } cookie:=http.cookie{name:"username",value:username,expires:expitation} http.setcookie(w,&cookie) } func logouthandler(w http.responsewriter, r *http.request) { expiration := time.now().adddate(0, 0, -1) cookie := http.cookie{name: "username", value: "alice_cooper@gmail.com", expires: expiration} http.setcookie(w, &cookie) }
这里我们在请求/login的时候,可以传递一个参数username到函数中,比如/login?username=alice , 默认为”guest”用户. 设置的cookie有效期为1天,删除的时候我们只需要设置cookie为之前的日期即可。
另外一个关键的部分是读取上下文保存内容的 statushandler() 路由处理函数,该函数将调用r.context()获得request的上下文,如果我们执行了login后,那我们在中间件中已经设置了该对象,所以请求将查看是否上下文对象中保存了一个名为username的对象,如果有的话则回应一个欢迎页面。否则告知用户没有登录。
func statushandler(w http.responsewriter,r *http.request){ if username:=r.context().value("username"); username!=nil{ w.writeheader(http.statusok) w.write([]byte("hi username:"+username.(string)+"\n")) }else{ w.writeheader(http.statusnotfound) w.write([]byte("not logged in")) } }
我们不仅仅可以在上下文中保存简单的类型,我们可以保存任何类型的数据,因为value() 返回的对象是一个interface{}对象,所以我们需要转换一下才能使用。
超时处理
对于简单的保存和传递对象,使用context的确很方便,但是该库的使用不仅仅是保存变量,还可以创建一个超时和取消的行为,比如说我们web端去请求了其他的资源,但是该资源的处理比较耗时,我们无法预见什么时候能够返回,如果让用户超时的话,实在是不太好,所以我们需要创建一个超时的操作,主动判断是否超时,然后传递一个合适的行为给用户。
这里我们现在路由中增加一个长期运行的job路由
mux.handlefunc("/longjob",jobwithcancelhandler)
具体的处理如下,我们的handler会利用withcancel() 返回一个新的(如果没有创建)或者原来已保存的上下文,还有一个cancel对象,这个对象可以用来手动执行取消操作。另外我们的url中可以指定这个任务模拟执行的长度,比如/longjob?jobtime=10则代表模拟的任务将会执行超过10秒。 执行任务的函数longrunningcalculation()返回一个chan该chan会在执行时间到期后写入一个done字符串。
handler中我们就可以使用select语句监听两个非缓存的channel,阻塞直到有数据写到任何一个channel中。比如代码中我们设置了超时是5秒,而任务执行10秒的话则5秒到期后ctx.done()会因为cancel()的调用而写入数据,这样该handler就会因为超时退出。否则的话则执行正常的job处理后获得传递的“done”退出。
func longrunningcalculation(timecost int)chan string{ result:=make(chan string) go func (){ time.sleep(time.second*(time.duration(timecost))) result<-"done" }() return result } func jobwithcancelhandler(w http.responsewriter, r * http.request){ var ctx context.context var cancel context.cancelfunc var jobtime string if jobtime=r.url.query().get("jobtime");jobtime==""{ jobtime = "10" } timecost,err:=strconv.atoi(jobtime) if err!=nil{ timecost=10 } log.println("job will cost : "+jobtime+"s") ctx,cancel = context.withcancel(r.context()) defer cancel() go func(){ time.sleep(5*time.second) cancel() }() select{ case <-ctx.done(): log.println(ctx.err()) return case result:=<-longrunningcalculation(timecost): io.writestring(w,result) } return }
这就是使用context的一些基本方式,其实context还有很多函数这里没有涉及,比如withtimeout和withdeadline等,但是使用上都比较相似,这里就不在举例。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
推荐阅读
-
linux下管理员强行踢出用户的命令使用方法
-
Firefox 鼠标手势使用时被右键的上下文菜单挡住解决方法
-
KVM虚拟化技术之使用Qemu-kvm创建和管理虚拟机的方法
-
gearman管理工具GearmanManager的安装与php使用方法示例
-
vs怎么添加代码管理器? visual studio代码段管理器的使用方法
-
RHEL7使用ssm命令管理LVM的方法
-
在Linux中使用Vundle管理Vim插件的方法
-
mysql管理软件有什么(免费的mysql可视化工具使用方法)
-
Android编程实现全局获取Context及使用Intent传递对象的方法详解
-
C#使用semaphore来管理异步下载请求的方法