GoToSocial/vendor/codeberg.org/gruf/go-runners/context.go

65 lines
1.1 KiB
Go

package runners
import (
"context"
"time"
)
// closedctx is an always closed context.
var closedctx = func() context.Context {
ctx := make(chan struct{})
close(ctx)
return CancelCtx(ctx)
}()
// Closed returns an always closed context.
func Closed() context.Context {
return closedctx
}
// CtxWithCancel returns a new context.Context impl with cancel.
func CtxWithCancel() (context.Context, context.CancelFunc) {
ctx := make(chan struct{})
cncl := func() { close(ctx) }
return CancelCtx(ctx), cncl
}
// CancelCtx is the simplest possible cancellable context.
type CancelCtx (<-chan struct{})
func (CancelCtx) Deadline() (time.Time, bool) {
return time.Time{}, false
}
func (ctx CancelCtx) Done() <-chan struct{} {
return ctx
}
func (ctx CancelCtx) Err() error {
select {
case <-ctx:
return context.Canceled
default:
return nil
}
}
func (CancelCtx) Value(key interface{}) interface{} {
return nil
}
func (ctx CancelCtx) String() string {
var state string
select {
case <-ctx:
state = "closed"
default:
state = "open"
}
return "CancelCtx{state:" + state + "}"
}
func (ctx CancelCtx) GoString() string {
return "runners." + ctx.String()
}