im-kulikov/helium

package grace

import (
    "context"
    "os/signal"
    "syscall"

    "go.uber.org/zap"

    "github.com/im-kulikov/helium/module"
)

// Module graceful context.
// nolint:gochecknoglobals
var Module = module.Module{
    {Constructor: NewGracefulContext},
}

// NewGracefulContext returns graceful context.
func NewGracefulContext(l *zap.Logger) context.Context {
    ctx, _ := signal.NotifyContext(context.Background(),
        syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)

    go func() {
        <-ctx.Done()
        l.Info("receive stop signal")
    }()

    return ctx
}