mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
34 lines
474 B
Go
34 lines
474 B
Go
package utils
|
|
|
|
import (
|
|
"sync/atomic"
|
|
)
|
|
|
|
type RefCount struct {
|
|
_ NoCopy
|
|
|
|
refCount uint32
|
|
zeroCh chan struct{}
|
|
}
|
|
|
|
func NewRefCounter() *RefCount {
|
|
rc := &RefCount{
|
|
refCount: 1,
|
|
zeroCh: make(chan struct{}),
|
|
}
|
|
return rc
|
|
}
|
|
|
|
func (rc *RefCount) Zero() <-chan struct{} {
|
|
return rc.zeroCh
|
|
}
|
|
|
|
func (rc *RefCount) Add() {
|
|
atomic.AddUint32(&rc.refCount, 1)
|
|
}
|
|
|
|
func (rc *RefCount) Sub() {
|
|
if atomic.AddUint32(&rc.refCount, ^uint32(0)) == 0 {
|
|
close(rc.zeroCh)
|
|
}
|
|
}
|