You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
428 B
Go
19 lines
428 B
Go
package syncx
|
|
|
|
import "sync/atomic"
|
|
|
|
// A OnceGuard is used to make sure a resource can be taken once.
|
|
type OnceGuard struct {
|
|
done uint32
|
|
}
|
|
|
|
// Taken checks if the resource is taken.
|
|
func (og *OnceGuard) Taken() bool {
|
|
return atomic.LoadUint32(&og.done) == 1
|
|
}
|
|
|
|
// Take takes the resource, returns true on success, false for otherwise.
|
|
func (og *OnceGuard) Take() bool {
|
|
return atomic.CompareAndSwapUint32(&og.done, 0, 1)
|
|
}
|