/
global.go
37 lines (27 loc) · 841 Bytes
/
global.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package graceful
import (
"context"
"os"
)
var defaultListener *Listener //nolint:gochecknoglobals
// Init subscribes to os signals and registers cancel func and group context.
func Init(ctx context.Context, signals ...os.Signal) (context.Context, context.CancelFunc) {
var cancelFunc context.CancelFunc
defaultListener, ctx, cancelFunc = NewListener(ctx, signals...)
return ctx, cancelFunc
}
// Register registers some service to listen and its "stop" function.
func Register(funcs ...interface{}) error {
if defaultListener == nil {
return ErrEmptyListener
}
defaultListener.Register(funcs...)
return nil
}
// Wait registers executor for "stop" callbacks and blocks until all registered callbacks will end.
func Wait() error {
if defaultListener == nil {
return ErrEmptyListener
}
return defaultListener.Wait()
}