golang context的具体使用场景举例-rpc并发调用

最新推荐文章于 2025-05-07 10:10:33 发布

原创

最新推荐文章于 2025-05-07 10:10:33 发布 · 1.2k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#go

本文介绍了在Go语言中，如何利用context包处理RPC并发调用的超时控制。通过设置整体任务的超时时间，在超时后无论RPC是否完成，都会返回结果。同时，详细阐述了context的传递特性，例如可以创建子Context来设定更严格的超时限制，当父Context被取消时，所有子Context也会随之取消。

rpc调用是在实际开发场景中经常用到的，假设某个用户请求的处理需要3个RPC并行请求，设置一个整体任务的超时时间，如果到达超时时间后，不管RPC有没有执行完毕任务都返回，实现逻辑见下面代码：

package main

import (
	"context"
	"fmt"
	"strconv"
	"sync"
	"time"
)

//rpc调用
func Rpc(ctx context.Context, url string) string {
	result := make(chan string)
	//开启一个新的协程去执行rpc调用逻辑
	go func() {
		//...balabala
		if url == "http://rpc_2_url" { //假设rpc-2服务比较耗时 就假设要2s吧
			time.Sleep(2*time.Second)
		}
		rpcResult := url+"xxxx" //假设调用rpc得到的数据就是这个样子
		result<-rpcResult
	}()
	//当前协程监听通道事件(整体任务的超时时间到/当前rpc调用完成了)
	select {
	case <- ctx.Done(): //context的超时事件
		// 如果规定的时间到了，没有得到结果也立即返回
		return ""
	case r:=<- result: //本RPC调用成功，返回结果
		return r
	}
}


func main() {
	//设置整个任务超时时间为1s
	ctx, _ := context.WithTimeout(context.Background(),1*time.Second)
	wg := sync.WaitGroup{}
	var lastRes string

	rcpUrl:=[]string{
		"http://rpc_1_url",
		"http://rpc_2_url",
		"http://rpc_3_url",
	}
	wg.Add(len(rcpUrl))
	for i,url := range rcpUrl {
		//并发执行rpc调用
		go func(i i

最低0.47元/天解锁文章