From ec6c3f74438c9aa6784b16262bd3ad2258f65d89 Mon Sep 17 00:00:00 2001
From: Itxaka <itxaka.garcia@spectrocloud.com>
Date: Fri, 17 Mar 2023 14:48:23 +0100
Subject: [PATCH] Add OnSignal misc util

Signed-off-by: Itxaka <itxaka.garcia@spectrocloud.com>
---
 utils/utils.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/utils/utils.go b/utils/utils.go
index 21849fc..2f5e03c 100644
--- a/utils/utils.go
+++ b/utils/utils.go
@@ -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()
+	}()
+}