Go Resiliency库中timeout实现原理及源码解析（go类库）学会了吗

目录1.go-resiliency简介2.timeout模式3.源码实现如下4.扩展一下

import (
“errors”
“testing”
“time”
)
func takesFiveSecond(stopper <-chan struct{}) error {
time.Sleep(5 * time.Second)
return nil
}
func takesTwentySecond(stopper <-chan struct{}) error {
time.Sleep(20 * time.Second)
return nil
}
func TestDeadline(t *testing.T) {
dl :=New(10 * time.Second)
//执行takesFiveSecond
if err :=dl.Run(takesFiveSecond); err !=nil {
t.Error(err)
}
//执行takesTwentySecond
if err :=dl.Run(takesTwentySecond); err==ErrTimedOut {
t.Error(err)
}
}

type Deadline struct {
timeout time.Duration
}
func New(timeout time.Duration) *Deadline {
return &Deadline{
timeout: timeout,
}
}

//1. 可以看到Run函数有一个入参是一个函数，函数的原型为func (<-chan struct{}))error 也就是说我们传入work变量就需要定义一个这个的签名函数。
//2. Run函数返回error，这个返回实际是入参work函数返回的。
//3.为什么work函数变量，要有一个chan了？ 这个主要为了能让work函数里来控制，Run提前退出
func (d *Deadline) Run(work func(<-chan struct{}) error) error {
result :=make(chan error)
stopper :=make(chan struct{})
//启动一个协程
go func() {
value :=work(stopper)
select {
case result <- value:
case <-stopper:
}
}()
//这里是判断是否超时常用手法，通过select监听2个chan,一个读取结果，一个为超时定时器。
//如果在timeout时间内未读取到执行结果，就触发time.After返回超时
select {
case ret :=<-result:
return ret
case <-time.After(d.timeout):
close(stopper)
return ErrTimedOut
}
}

import (
“context”
“errors”
“time”
)
var ErrTimedOut=errors.New(“timed out waiting for function to finish”)
type ContextTimeOut struct {
timeout time.Duration
}
// New constructs a new Deadline with the given timeout.
func New(timeout time.Duration) *ContextTimeOut {
return &ContextTimeOut{
timeout: timeout,
}
}
func (d *ContextTimeOut) Run(work func(<-chan struct{}) error) error {
result :=make(chan error)
stopper :=make(chan struct{})
go func() {
value :=work(stopper)
select {
case result <- value:
case <-stopper:
}
}()
ctx, _ :=context.WithTimeout(context.Background(), d.timeout)
select {
case ret :=<-result:
return ret
case <-ctx.Done():
close(stopper)
return ErrTimedOut
}
}

您可能感兴趣的文章:深入探究Golang中log标准库的使用基于Golang开发一个轻量级登录库/框架深入探究Golang中flag标准库的使用