提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  后端

当前栏目

Go-标准库-context（二）

Go标准 Context
2023-06-13 09:18:47 时间

传递值

除了用于传递取消信号外，context还可以用于传递请求范围的值。可以通过context.WithValue函数将一个值保存到context中，然后在需要的地方通过context.Value函数获取该值。

下面是一个使用context传递请求范围的值的示例程序：：

package main

import (
	"context"
	"fmt"
)

type key int

const (
	nameKey key = iota
	ageKey
)

func main() {
	ctx := context.WithValue(context.Background(), nameKey, "Alice")
	ctx = context.WithValue(ctx, ageKey, 20)

	name, ok := ctx.Value(nameKey).(string)
	if !ok {
		fmt.Println("name not found")
		return
	}
	age, ok := ctx.Value(ageKey).(int)
	if !ok {
		fmt.Println("age not found")
		return
	}

	fmt.Printf("name: %s\n", name)
	fmt.Printf("age: %d\n", age)
}

在这个示例程序中，我们通过context.WithValue函数将一个名为"name"的字符串和一个年龄为20的整数保存到context中。然后，我们分别通过context.Value函数获取这两个值，并将它们转换为相应的类型。最后，我们输出这两个值。

需要注意的是，使用context传递请求范围的值并不是一个好的设计，因为这样会让程序变得复杂，而且容易出错。如果需要在多个goroutine之间共享数据，应该使用同步原语（例如mutex或channel）来保证数据的安全性。

实际应用

context包被广泛用于Go中的网络编程和多线程编程中。在网络编程中，可以使用context包来传递请求的截止时间和取消信号，以便控制网络连接的超时和关闭。在多线程编程中，可以使用context包来控制多个goroutine之间的交互，避免出现竞争条件和死锁等问题。

下面是一个使用context包进行网络编程的示例程序：

package main

import (
	"context"
	"fmt"
	"io/ioutil"
	"net/http"
	"time"
)

func main() {
	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
	defer cancel()

	req, err := http.NewRequest("GET", "https://www.google.com", nil)
	if err != nil {
		fmt.Println("http.NewRequest error:", err)
		return
	}

	req = req.WithContext(ctx)

	client := http.DefaultClient
	resp, err := client.Do(req)
	if err != nil {
		fmt.Println("client.Do error:", err)
		return
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("ioutil.ReadAll error:", err)
		return
	}

	fmt.Println(string(body))
}

在这个示例程序中，我们创建了一个带有2秒超时的Cancel context，并将其作为请求的Context。然后，我们使用http.NewRequest函数创建了一个GET请求，并将请求的Context设置为上面创建的Cancel context。最后，我们使用http.DefaultClient执行请求，并读取响应的内容。

需要注意的是，在网络编程中，要注意合理设置请求的超时时间，避免请求超时或一直等待的情况发生。

相关文章