TheArchive/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2024-12-24 16:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

add util.CancellationChild() and run gofmt

Browse Source
This commit is contained in:
Arceliar 2019-07-17 21:15:02 -05:00
parent 06e8403aaf
commit 6bf182e341
1 changed files with 60 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/util/cancellation.go
View File

@ -2,9 +2,9 @@ package util
import (
"errors"
"runtime"
"sync"
"time"
"runtime"
)
type Cancellation interface {
@ -49,7 +49,7 @@ func (c *cancellation) Finished() <-chan struct{} {
func (c *cancellation) Cancel(err error) error {
select {
case c.signal<-err:
case c.signal <- err:
return nil
case <-c.cancel:
return c.Error()
@ -63,14 +63,25 @@ func (c *cancellation) Error() error {
return err
}
func CancellationWithTimeout(parent Cancellation, timeout time.Duration) Cancellation {
func CancellationChild(parent Cancellation) Cancellation {
child := NewCancellation()
go func() {
select {
case <-child.Finished():
case <-parent.Finished():
child.Cancel(parent.Error())
}
}()
return child
}
func CancellationWithTimeout(parent Cancellation, timeout time.Duration) Cancellation {
child := CancellationChild(parent)
go func() {
timer := time.NewTimer(timeout)
defer TimerStop(timer)
select {
case <-parent.Finished():
child.Cancel(parent.Error())
case <-child.Finished():
case <-timer.C:
child.Cancel(errors.New("timeout"))
}