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

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:=
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值