在 Go http包的Server中,每一个请求在都有一个对应的 goroutine 去处理。请求处理函数通常会启动额外的 goroutine 用来访问后端服务,比如数据库和RPC服务。用来处理一个请求的 goroutine 通常需要访问一些与请求特定的数据,比如终端用户的身份认证信息、验证相关的token、请求的截止时间。 当一个请求被取消或超时时,所有用来处理该请求的 goroutine 都应该迅速退出,然后系统才能释放这些 goroutine 占用的资源。
在Google 内部,开发了 Context 包,专门用来简化 对于处理单个请求的多个 goroutine 之间与请求域的数据、取消信号、截止时间等相关操作,这些操作可能涉及多个 API 调用。你可以通过 go get golang.org/x/net/context 命令获取这个包。本文要讲的就是如果使用这个包,同时也会提供一个完整的例子。
Context interface
type Context interface {
Deadline() (deadline time.Time,ok bool)
Done() <-chan struct{}
Err() error
Value(key interface{}) interface{}
}
Deadline()返回一个time.Time,是当前 Context 的应该结束的时间,ok 表示是否有 deadline Done()返回一个struct{}类型的只读 channel Err()返回 Context 被取消时的错误 Value(key interface{}) 是 Context 自带的 K-V 存储功能
package main
import (
"context"
"fmt"
"io/IoUtil"
"net/http"
"time"
)
type Result struct {
r *http.Response
err error
}
func process() {
ctx,cancel := context.WithTimeout(context.Background(), 2*time.Second)
//释放资源
defer cancel()
tr := &http.Transport{}
client := &http.Client{Transport: tr}
resultChan := make(chan Result, 1)
//发起请求
req,err := http.NewRequest("GET","http://www.baidu.com",nil)
if err != nil {
fmt.Println("http request Failed,err:",err)
return
}
/* func (c *Client) Do(req *Request) (*Response,error) */
go func() {
resp,err := client.Do(req)
pack := Result{r: resp,err: err}
//将返回信息写入管道(正确或者错误的)
resultChan <- pack
}()
select {
case <-ctx.Done():
tr.CancelRequest(req)
er:= <-resultChan
fmt.Println("Timeout!",er.err)
case res := <-resultChan:
defer res.r.Body.Close()
out,_ := IoUtil.ReadAll(res.r.Body)
fmt.Printf("Server Response: %s",out)
}
return
}
func main() {
process()
}
输出如下:
req,err := http.NewRequest("GET","http://google.com",nil)
请求超时,输出log信息如下
PS E:\golang\go_pro\src\safly> go build main.go
PS E:\golang\go_pro\src\safly> main.exe
Timeout! Get http://google.com: net/http: request canceled while waiting for connection
PS E:\golang\go_pro\src\safly>
原文链接:https://www.f2er.com/go/187655.html