35 lines
440 B
Go
35 lines
440 B
Go
package util
|
|
|
|
import (
|
|
"context"
|
|
)
|
|
|
|
type Signaler interface {
|
|
Signal()
|
|
Wait(ctx context.Context) error
|
|
}
|
|
|
|
type signaler chan struct{}
|
|
|
|
func NewSignaler() Signaler {
|
|
return make(signaler, 1)
|
|
}
|
|
|
|
var _ Signaler = (*signaler)(nil)
|
|
|
|
func (s signaler) Signal() {
|
|
select {
|
|
case s <- struct{}{}:
|
|
default:
|
|
}
|
|
}
|
|
|
|
func (s signaler) Wait(ctx context.Context) error {
|
|
select {
|
|
case <-s:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|