golang基础-context使用

前端之家收集整理的这篇文章主要介绍了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 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

猜你在找的Go相关文章