Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

pkg/wait: add comment and make List private

Browse Source
This commit is contained in:
Xiang Li 2016-04-07 22:09:10 -07:00 committed by Gyu-Ho Lee
parent c309d745a6
commit d818ef2c76
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/wait/wait.go
View File

@ -21,22 +21,29 @@ import (
"sync" "sync"
) )
// Wait is an interface that provides the ability to wait and trigger events that
// are associated with IDs.
type Wait interface { type Wait interface {
// Register waits returns a chan that waits on the given ID.
// The chan will be triggered when Trigger is called with
// the same ID.
Register(id uint64) <-chan interface{} Register(id uint64) <-chan interface{}
// Trigger triggers the waiting chans with the given ID.
Trigger(id uint64, x interface{}) Trigger(id uint64, x interface{})
IsRegistered(id uint64) bool IsRegistered(id uint64) bool
} }
type List struct { type list struct {
l sync.Mutex l sync.Mutex
m map[uint64]chan interface{} m map[uint64]chan interface{}
} }
func New() *List { // New creates a Wait.
return &List{m: make(map[uint64]chan interface{})} func New() Wait {
return &list{m: make(map[uint64]chan interface{})}
} }
func (w *List) Register(id uint64) <-chan interface{} { func (w *list) Register(id uint64) <-chan interface{} {
w.l.Lock() w.l.Lock()
defer w.l.Unlock() defer w.l.Unlock()
ch := w.m[id] ch := w.m[id]
@ -49,7 +56,7 @@ func (w *List) Register(id uint64) <-chan interface{} {
return ch return ch
} }
func (w *List) Trigger(id uint64, x interface{}) { func (w *list) Trigger(id uint64, x interface{}) {
w.l.Lock() w.l.Lock()
ch := w.m[id] ch := w.m[id]
delete(w.m, id) delete(w.m, id)
@ -60,7 +67,7 @@ func (w *List) Trigger(id uint64, x interface{}) {
} }
} }
func (w *List) IsRegistered(id uint64) bool { func (w *list) IsRegistered(id uint64) bool {
w.l.Lock() w.l.Lock()
defer w.l.Unlock() defer w.l.Unlock()
_, ok := w.m[id] _, ok := w.m[id]