wait for persistence on signal

d-led · Jun 23, 2024 · cfcafc2 · cfcafc2
1 parent 8949148
commit cfcafc2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/mermaidlive/main.go b/cmd/mermaidlive/main.go
@@ -10,6 +10,7 @@ import (
 	"github.com/carlmjohnson/versioninfo"
 	"github.com/cskr/pubsub/v2"
 	"github.com/d-led/mermaidlive"
+	"github.com/d-led/percounter"
 )
 
 var transpileOnly *bool
@@ -51,7 +52,9 @@ func main() {
 		mermaidlive.GetFS(),
 		countdownDelay,
 	)
-	server.Run(*port)
+	go server.Run(*port)
+	percounter.GlobalEmergencyPersistence().Init()
+	percounter.GlobalEmergencyPersistence().PersistAndExitOnSignal()
 }
 
 func init() {