mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-12 03:26:30 +00:00
35 lines
494 B
Go
35 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
|
||
|
|
}
|