Go语言:select的典型应用场景之IO超时控制的示例(采用default,非阻塞方式)

select语句当使用default分支时,则采用非阻塞方式从所有收/发操作准备就绪(或者关闭)的channel中则随机选择一个。select最为常见的应用就是IO超时控制,既可以采用阻塞方式也可以采用非阻塞方式来实现,阻塞方式的写法请见另一篇博客《Go语言:select的典型应用场景之IO超时控制的示例(阻塞方式)》

非阻塞方式示例:

请求超时设为50毫秒,模拟的IO操作响应延迟为0-100毫秒之间的随机数。多运行几次则小于50毫秒延时的有响应,其余为超时报错。

select.go

package main

import (
	"errors"
	"fmt"
	"math/rand"
	"time"
)

type Response struct {
	body   string        //响应内容
	elapse time.Duration //响应耗时
}

func request(timeout time.Duration) (Response, error) {
	//用于触发超时的计时器
	timer := time.NewTimer(timeout)
	defer timer.Stop()

	start := time.Now()

	var respData Response
	var err error

	//同一个协程中执行IO操作
	respData.body = doIO()

	//如果超时计时器已处在就绪状态,则生成一个error的实例,否则继续往下执行。
	select {
	case <-timer.C:
		err = errors.New("request timeout1")
	default:
	}

	//记录总耗时
	respData.elapse = time.Now().Sub(start)

	return respData, err
}

func doIO() string{
	//随机产生一个[0,100)毫秒的延迟,以模拟IO延时延迟
	rand.Seed(time.Now().UnixNano())
	delay := time.Duration(rand.Intn(100)) * time.Millisecond
	fmt.Printf("delay=%v\n", delay)
	time.Sleep(delay)

	return "Hello World"
}

func main() {
	resp, err := request(50 * time.Millisecond)
	if err != nil {
		fmt.Printf("error: %s elapse=%s\n", err.Error(), resp.elapse)
		return
	}

	fmt.Printf("response: body=%s elapse=%s\n", resp.body, resp.elapse)
}

输出:

[root@dev tutorial]# go run select.go
delay=84ms
error: request timeout1 elapse=84.174263ms
[root@dev tutorial]# go run select.go
delay=78ms
error: request timeout1 elapse=78.152999ms
[root@dev tutorial]# go run select.go
delay=39ms
response: body=Hello World elapse=39.177298ms
[root@dev tutorial]# go run select.go
delay=19ms
response: body=Hello World elapse=19.196615ms

 

Go源码里采用非阻塞方式的例子:

net/fd_unix.go的函数

func (fd *netFD) connect(ctx context.Context, la, ra syscall.Sockaddr) (rsa syscall.Sockaddr, ret error)

代码节选,文件描述符fd等待写操作,如果报错,则以非阻塞方式检查上下文是否超时,是则返回超时错误,否则返回等待写操作的报错。

		if err := fd.pfd.WaitWrite(); err != nil {
			select {
			case <-ctx.Done():
				return nil, mapErr(ctx.Err())
			default:
			}
			return nil, err
		}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值