handshake.go RawHistoryFolder

1package client
2
3import (
4	"context"
5	"sync"
6	"time"
7
8	"github.com/bnema/vev/internal/ports"
9)
10
11type systemClock struct{}
12type systemTimer struct{ timer *time.Timer }
13
14func (systemClock) Now() time.Time { return time.Now() }
15func (systemClock) NewTimer(delay time.Duration) ports.Timer {
16	return systemTimer{timer: time.NewTimer(delay)}
17}
18func (t systemTimer) C() <-chan time.Time        { return t.timer.C }
19func (t systemTimer) Reset(d time.Duration) bool { return t.timer.Reset(d) }
20func (t systemTimer) Stop() bool                 { return t.timer.Stop() }
21
22// newBoundedContext owns one deadline of the requested length. The caller must
23// invoke finish when the bounded operation ends.
24func newBoundedContext(parent context.Context, clock ports.Clock, timeout time.Duration) (context.Context, <-chan struct{}, func()) {
25	if parent == nil {
26		parent = context.Background()
27	}
28	ctx, cancel := context.WithCancel(parent)
29	timedOut := make(chan struct{})
30	if clock == nil {
31		clock = systemClock{}
32	}
33	timer := clock.NewTimer(timeout)
34	if timer == nil {
35		timer = systemClock{}.NewTimer(timeout)
36	}
37	stop := make(chan struct{})
38	done := make(chan struct{})
39	go func() {
40		defer close(done)
41		select {
42		case <-timer.C():
43			close(timedOut)
44			cancel()
45		case <-stop:
46		}
47	}()
48	var once sync.Once
49	finish := func() {
50		once.Do(func() {
51			close(stop)
52			<-done
53			timer.Stop()
54			cancel()
55		})
56	}
57	return ctx, timedOut, finish
58}