Add OnSignal misc util

Signed-off-by: Itxaka <itxaka.garcia@spectrocloud.com>
This commit is contained in:
Itxaka
2023-03-17 14:48:23 +01:00
parent 58ddd1c5df
commit ec6c3f7443

View File

@@ -5,6 +5,7 @@ import (
"fmt"
"os"
"os/exec"
"os/signal"
"strings"
"github.com/denisbrodbeck/machineid"
@@ -115,3 +116,12 @@ func SetEnv(env []string) {
}
}
}
func OnSignal(fn func(), sig ...os.Signal) {
sigs := make(chan os.Signal, 1)
signal.Notify(sigs, sig...)
go func() {
<-sigs
fn()
}()
}