From 20e58f06839966a5acf4bfed7420dae47f712463 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Mierzwa?= <l.mierzwa@gmail.com>
Date: Sun, 12 May 2019 15:18:50 +0100
Subject: [PATCH] feat(backend): graceful shutdown support

---
 main.go | 30 ++++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index dca37e46c..7fab3239a 100644
--- a/main.go
+++ b/main.go
@@ -1,11 +1,15 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"html/template"
 	"net/http"
+	"os"
+	"os/signal"
 	"path"
 	"strings"
+	"syscall"
 	"time"
 
 	"github.com/prymitive/karma/internal/alertmanager"
@@ -231,10 +235,28 @@ func main() {
 			log.Fatalf("Failed to setup proxy handlers for Alertmanager '%s': %s", am.Name, err)
 		}
 	}
+
 	listen := fmt.Sprintf("%s:%d", config.Config.Listen.Address, config.Config.Listen.Port)
-	log.Infof("Listening on %s", listen)
-	err := router.Run(listen)
-	if err != nil {
-		log.Fatal(err)
+	httpServer := &http.Server{
+		Addr:    listen,
+		Handler: router,
 	}
+
+	go func() {
+		if err := httpServer.ListenAndServe(); err != nil {
+			log.Infof("Listening on %s", listen)
+		}
+	}()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	log.Infof("Shutting down HTTP server")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+	if err := httpServer.Shutdown(ctx); err != nil {
+		log.Fatalf("Shutdown failed: %s", err)
+	}
+	log.Info("HTTP server shut down")
 }