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

Merge pull request #7831 from gyuho/cc

Browse Source 
pkg/wait: add comment and make List private
This commit is contained in:
Gyu-Ho Lee 2017-04-27 13:45:25 -07:00 committed by GitHub
commit 96d6f05391
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"
)
// Wait is an interface that provides the ability to wait and trigger events that
// are associated with IDs.
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{}
// Trigger triggers the waiting chans with the given ID.
Trigger(id uint64, x interface{})
IsRegistered(id uint64) bool
}
type List struct {
type list struct {
l sync.Mutex
m map[uint64]chan interface{}
}
func New() *List {
return &List{m: make(map[uint64]chan interface{})}
// New creates a Wait.
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()
defer w.l.Unlock()
ch := w.m[id]
@ -49,7 +56,7 @@ func (w *List) Register(id uint64) <-chan interface{} {
return ch
}
func (w *List) Trigger(id uint64, x interface{}) {
func (w *list) Trigger(id uint64, x interface{}) {
w.l.Lock()
ch := 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()
defer w.l.Unlock()
_, ok := w.m[id]