1 Commits

Author SHA1 Message Date
jar3b
a980f1f8d2 Add blocking mode support 2019-04-09 17:50:55 +03:00
2 changed files with 16 additions and 4 deletions

View File

@@ -10,5 +10,5 @@ waiter.AddCloseHandler(func() {
nacl.FinalizeStan()
}, false)
waiter.Wait()
waiter.Wait(true)
```

View File

@@ -9,6 +9,7 @@ import (
)
type Waiter struct {
blockingMode bool
waitGroup sync.WaitGroup
closeHandlers []*CloseHandler
}
@@ -53,11 +54,21 @@ func (w *Waiter) Halt(err error) {
} else {
log.Info("Program was terminated gracefully.")
}
if !w.blockingMode {
if err != nil {
os.Exit(1)
} else {
os.Exit(0)
}
}
}
func (w *Waiter) Wait() {
func (w *Waiter) Wait(blockingMode bool) {
w.blockingMode = blockingMode
if blockingMode {
log.Info("Waiting...")
w.waitGroup.Wait()
}
}
func (w *Waiter) onSignal(sig os.Signal) {
@@ -67,6 +78,7 @@ func (w *Waiter) onSignal(sig os.Signal) {
func NewWaiter() *Waiter {
w := Waiter{
true,
sync.WaitGroup{},
make([]*CloseHandler, 0),
}