golang基础-context使用

在 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 @H_403_8@interface {
    Deadline() (deadline time.Time,ok @H_403_8@bool)
    Done() <-chan @H_403_8@struct{}
    Err() error
    Value(key @H_403_8@interface{}) @H_403_8@interface{}
}

Deadline()返回一个time.Time，是当前 Context 的应该结束的时间，ok 表示是否有 deadline
Done()返回一个struct{}类型的只读 channel
Err()返回 Context 被取消时的错误
Value(key interface{}) 是 Context 自带的 K-V 存储功能

@H_403_8@package main

@H_403_8@import (
    "context"
    "fmt"
    "io/IoUtil"
    "net/http"
    "time"
)
@H_403_8@type Result @H_403_8@struct {
    r   *http.Response
    err error
}
@H_403_8@func process() {
    ctx,cancel := context.WithTimeout(context.Background(), 2*time.Second)
    //释放资源
    @H_403_8@defer cancel()
    tr := &http.Transport{}
    client := &http.Client{Transport: tr}
    resultChan := make(@H_403_8@chan Result, 1)
    //发起请求
    req,err := http.NewRequest("GET","http://www.baidu.com",nil)
    @H_403_8@if err != nil {
        fmt.Println("http request Failed,err:",err)
        @H_403_8@return
    }
    /* func (c *Client) Do(req *Request) (*Response,error) */
    @H_403_8@go @H_403_8@func() {
        resp,err := client.Do(req)
        pack := Result{r: resp,err: err}
        //将返回信息写入管道(正确或者错误的)
        resultChan <- pack
    }()
    @H_403_8@select {
    @H_403_8@case <-ctx.Done():
        tr.CancelRequest(req)
        er:= <-resultChan
        fmt.Println("Timeout!",er.err)
    @H_403_8@case res := <-resultChan:
        @H_403_8@defer res.r.Body.Close()
        out,_ := IoUtil.ReadAll(res.r.Body)
        fmt.Printf("Server Response: %s",out)
    }
    @H_403_8@return
}
@H_403_8@func main() {
    process()
}

输出如下：

如果修改下代码

req,err := http.NewRequest("GET","http://google.com",@H_403_8@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>

golang基础-context使用

猜你在找的Go相关文章