aboutsummaryrefslogtreecommitdiff
path: root/rutina.go
diff options
context:
space:
mode:
Diffstat (limited to 'rutina.go')
-rwxr-xr-xrutina.go7
1 files changed, 5 insertions, 2 deletions
diff --git a/rutina.go b/rutina.go
index 0bdc7e8..703e9fc 100755
--- a/rutina.go
+++ b/rutina.go
@@ -122,10 +122,13 @@ func (r *Rutina) Go(doer func(ctx context.Context) error, opts ...Options) {
}
// OS signals handler
-func (r *Rutina) ListenOsSignals() {
+func (r *Rutina) ListenOsSignals(signals ...os.Signal) {
+ if len(signals) == 0 {
+ signals = []os.Signal{os.Kill, os.Interrupt}
+ }
r.Go(func(ctx context.Context) error {
sig := make(chan os.Signal, 1)
- signal.Notify(sig, os.Interrupt, os.Kill)
+ signal.Notify(sig, signals...)
select {
case s := <-sig:
if r.logger != nil {