34 lines
494 B
Go
34 lines
494 B
Go
package stream
|
|
|
|
import (
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
type broadcaster struct {
|
|
sync.RWMutex
|
|
closed uint32
|
|
*sync.Cond
|
|
}
|
|
|
|
func newBroadcaster() *broadcaster {
|
|
var b broadcaster
|
|
b.Cond = sync.NewCond(b.RWMutex.RLocker())
|
|
return &b
|
|
}
|
|
|
|
func (b *broadcaster) Wait() {
|
|
if b.IsOpen() {
|
|
b.Cond.Wait()
|
|
}
|
|
}
|
|
|
|
func (b *broadcaster) IsOpen() bool {
|
|
return atomic.LoadUint32(&b.closed) == 0
|
|
}
|
|
|
|
func (b *broadcaster) Close() error {
|
|
atomic.StoreUint32(&b.closed, 1)
|
|
b.Cond.Broadcast()
|
|
return nil
|
|
}
|