Miccah
5c0ffda618
* Add CancellableWrite helper function * Create SourceUnitEnumerator interface and EnumerationResult struct * Implement SourceUnitEnumerator for the filesystem Source * Omit explicit zero values
25 lines
533 B
Go
25 lines
533 B
Go
package common
|
|
|
|
import "context"
|
|
|
|
func IsDone(ctx context.Context) bool {
|
|
select {
|
|
case <-ctx.Done():
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
// CancellableWrite blocks on writing the item to the channel but can be
|
|
// cancelled by the context. If both the context is cancelled and the channel
|
|
// write would succeed, either operation will be performed randomly.
|
|
func CancellableWrite[T any](ctx context.Context, ch chan<- T, item T) error {
|
|
select {
|
|
case ch <- item:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|