Rework blocking logic, up to 0.1.2

This commit is contained in:
jar3b 2019-04-09 17:55:36 +03:00
parent a980f1f8d2
commit ac0f979ef9
2 changed files with 7 additions and 8 deletions

View File

@ -10,5 +10,6 @@ waiter.AddCloseHandler(func() {
nacl.FinalizeStan() nacl.FinalizeStan()
}, false) }, false)
waiter.Wait(true) // blocking wait, if no need to block (with http server, for example), you can omit .Wait() call
waiter.Wait()
``` ```

View File

@ -63,12 +63,10 @@ func (w *Waiter) Halt(err error) {
} }
} }
func (w *Waiter) Wait(blockingMode bool) { func (w *Waiter) Wait() {
w.blockingMode = blockingMode w.blockingMode = true
if blockingMode { log.Info("Waiting...")
log.Info("Waiting...") w.waitGroup.Wait()
w.waitGroup.Wait()
}
} }
func (w *Waiter) onSignal(sig os.Signal) { func (w *Waiter) onSignal(sig os.Signal) {
@ -78,7 +76,7 @@ func (w *Waiter) onSignal(sig os.Signal) {
func NewWaiter() *Waiter { func NewWaiter() *Waiter {
w := Waiter{ w := Waiter{
true, false,
sync.WaitGroup{}, sync.WaitGroup{},
make([]*CloseHandler, 0), make([]*CloseHandler, 0),
} }