fix: run shutdown once

flanksource · Oct 15, 2024 · cc304fd · cc304fd
1 parent 8d95a2e
commit cc304fd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/shutdown/shutdown.go b/shutdown/shutdown.go
@@ -3,13 +3,14 @@ package shutdown
 import (
 	"os"
 	"os/signal"
+	"sync"
 
 	"github.com/flanksource/commons/logger"
 )
 
 var shutdownHooks []func()
 
-func Shutdown() {
+var Shutdown = sync.OnceFunc(func() {
 	if len(shutdownHooks) == 0 {
 		return
 	}
@@ -18,7 +19,7 @@ func Shutdown() {
 		fn()
 	}
 	shutdownHooks = []func(){}
-}
+})
 
 func ShutdownAndExit(code int, msg string) {
 	Shutdown()