diff --git a/README.md b/README.md index b70478a..44465a9 100644 --- a/README.md +++ b/README.md @@ -10,5 +10,6 @@ waiter.AddCloseHandler(func() { nacl.FinalizeStan() }, false) -waiter.Wait(true) +// blocking wait, if no need to block (with http server, for example), you can omit .Wait() call +waiter.Wait() ``` \ No newline at end of file diff --git a/waiter.go b/waiter.go index d1e5026..85d5d12 100644 --- a/waiter.go +++ b/waiter.go @@ -63,12 +63,10 @@ func (w *Waiter) Halt(err error) { } } -func (w *Waiter) Wait(blockingMode bool) { - w.blockingMode = blockingMode - if blockingMode { - log.Info("Waiting...") - w.waitGroup.Wait() - } +func (w *Waiter) Wait() { + w.blockingMode = true + log.Info("Waiting...") + w.waitGroup.Wait() } func (w *Waiter) onSignal(sig os.Signal) { @@ -78,7 +76,7 @@ func (w *Waiter) onSignal(sig os.Signal) { func NewWaiter() *Waiter { w := Waiter{ - true, + false, sync.WaitGroup{}, make([]*CloseHandler, 0), }