mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 09:40:24 +08:00
16 lines
246 B
Go
16 lines
246 B
Go
|
package syncx
|
||
|
|
|
||
|
|
import "sync/atomic"
|
||
|
|
|
||
|
|
type OnceGuard struct {
|
||
|
|
done uint32
|
||
|
|
}
|
||
|
|
|
||
|
|
func (og *OnceGuard) Taken() bool {
|
||
|
|
return atomic.LoadUint32(&og.done) == 1
|
||
|
|
}
|
||
|
|
|
||
|
|
func (og *OnceGuard) Take() bool {
|
||
|
|
return atomic.CompareAndSwapUint32(&og.done, 0, 1)
|
||
|
|
}
|