From 05e3b3397c888807719d70af4ed3c73397d9374f Mon Sep 17 00:00:00 2001
From: xengineering <me@xengineering.eu>
Date: Sat, 11 Feb 2023 18:17:01 +0100
Subject: Move shutdown code to main.go

This has nothing to do with the database. Because the db is now a global
pointer the shutdown code can live in main.go.
---
 main.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 405ff0a..c1e0571 100644
--- a/main.go
+++ b/main.go
@@ -3,7 +3,10 @@ package main
 
 import (
 	"log"
+	"os"
+	"os/signal"
 	"database/sql"
+	"syscall"
 )
 
 var config RuntimeConfig
@@ -13,5 +16,18 @@ func main() {
 	log.Printf("Started Ceres recipe server.\n")
 	config = GetRuntimeConfig()
 	db = setupDatabase()
+	provideShutdown()
 	runServer()
 }
+
+func provideShutdown() {
+	var listener = make(chan os.Signal)
+	signal.Notify(listener, syscall.SIGTERM)
+	signal.Notify(listener, syscall.SIGINT)
+	go func() {
+		signal := <-listener
+		log.Printf("Got signal '%+v'. Shutdown is started.\n", signal)
+		dbCleanup(db)
+		os.Exit(0)
+	}()
+}
-- 
cgit v1.2.3-70-g09d2