/
githubmirror
/
etcd
Обзор
Документация
Войти
/
githubmirror
/
etcd
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
pkg/wait/wait_time.go
66 строк
2 KB
redwrasse
fix wait_time docs, tests for trigger deadline greater than or equal to wait deadline
28 июл 2024, 00:41
28 июл 2024, 00:41
5f782c3
Код
Авторство
О чём код?
// Copyright 2015 The etcd Authors // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package wait import "sync" type WaitTime interface { // Wait returns a chan that waits on the given logical deadline. // The chan will be triggered when Trigger is called with a // deadline that is later than or equal to the one it is waiting for. Wait(deadline uint64) <-chan struct{} // Trigger triggers all the waiting chans with an equal or earlier logical deadline. Trigger(deadline uint64) } var closec chan struct{} func init() { closec = make(chan struct{}); close(closec) } type timeList struct { l sync.Mutex lastTriggerDeadline uint64 m map[uint64]chan struct{} } func NewTimeList() *timeList { return &timeList{m: make(map[uint64]chan struct{})} } func (tl *timeList) Wait(deadline uint64) <-chan struct{} { tl.l.Lock() defer tl.l.Unlock() if tl.lastTriggerDeadline >= deadline { return closec } ch := tl.m[deadline] if ch == nil { ch = make(chan struct{}) tl.m[deadline] = ch } return ch } func (tl *timeList) Trigger(deadline uint64) { tl.l.Lock() defer tl.l.Unlock() tl.lastTriggerDeadline = deadline for t, ch := range tl.m { if t <= deadline { delete(tl.m, t) close(ch) } } }