Compare commits
2 Commits
09aa19e7c1
...
c6f54927b0
Author | SHA1 | Date | |
---|---|---|---|
c6f54927b0 | |||
0f9d8fb102 |
36
sync/locker.go
Normal file
36
sync/locker.go
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
package usync
|
||||||
|
|
||||||
|
import "sync"
|
||||||
|
|
||||||
|
// Locker guards access to a value
|
||||||
|
type Locker[T any] struct {
|
||||||
|
value T
|
||||||
|
mutex sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewLocker creates a new locker with the specified value.
|
||||||
|
func NewLocker[T any] (value T) Locker[T] {
|
||||||
|
return Locker[T] {
|
||||||
|
value: value,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set sets the value of the locker.
|
||||||
|
func (this *Locker[T]) Set (value T) {
|
||||||
|
this.mutex.Lock()
|
||||||
|
defer this.mutex.Unlock()
|
||||||
|
this.value = value
|
||||||
|
}
|
||||||
|
|
||||||
|
// Borrow borrows the value from the locker, and returns a function that must
|
||||||
|
// immediately be deferred, like this:
|
||||||
|
//
|
||||||
|
// value, done := locker.Borrow()
|
||||||
|
// defer done()
|
||||||
|
//
|
||||||
|
// From the time Borrow is called to the time the done function is called, it is
|
||||||
|
// safe to access the locked object from within the current goroutine.
|
||||||
|
func (this *Locker[T]) Borrow () (T, func ()) {
|
||||||
|
this.mutex.Lock()
|
||||||
|
return this.value, this.mutex.Unlock
|
||||||
|
}
|
@ -4,9 +4,15 @@ import "slices"
|
|||||||
import "context"
|
import "context"
|
||||||
import "reflect"
|
import "reflect"
|
||||||
|
|
||||||
|
// ChannelRecv is a type constraint that includes all channel types that can be
|
||||||
|
// recieved from.
|
||||||
|
type ChannelRecv[T any] interface {
|
||||||
|
~chan T | ~<- chan T
|
||||||
|
}
|
||||||
|
|
||||||
// A type-safe wrapper around reflect.Select. Taken from:
|
// A type-safe wrapper around reflect.Select. Taken from:
|
||||||
// https://stackoverflow.com/questions/19992334
|
// https://stackoverflow.com/questions/19992334
|
||||||
func Select[T any] (ctx context.Context, channels ...chan T) (int, T, bool) {
|
func Select[T any, C ChannelRecv[T]] (ctx context.Context, channels ...C) (int, T, bool) {
|
||||||
var zero T
|
var zero T
|
||||||
|
|
||||||
// add all channels as select cases
|
// add all channels as select cases
|
||||||
@ -39,11 +45,12 @@ func Select[T any] (ctx context.Context, channels ...chan T) (int, T, bool) {
|
|||||||
return chosen, ret, true
|
return chosen, ret, true
|
||||||
}
|
}
|
||||||
return chosen, zero, false
|
return chosen, zero, false
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Combine returns a channel that continuously returns the result of the select
|
// Combine returns a channel that continuously returns the result of the select
|
||||||
// until all input sources are exhauste, or the context is canceled.
|
// until all input sources are exhauste, or the context is canceled.
|
||||||
func Combine[T any] (ctx context.Context, channels ...chan T) <- chan T {
|
func Combine[T any, C ChannelRecv[T]] (ctx context.Context, channels ...C) <- chan T {
|
||||||
channel := make(chan T)
|
channel := make(chan T)
|
||||||
// our silly slection routine
|
// our silly slection routine
|
||||||
go func () {
|
go func () {
|
||||||
|
Reference in New Issue
Block a user