From 7eba9bb4f74d863dbfbb4799e4800b4010bb777e Mon Sep 17 00:00:00 2001
From: Alberto Bertogli <albertito@blitiri.com.ar>
Date: Fri, 22 Jul 2016 01:42:21 +0100
Subject: [PATCH] chasquid: Flush logs periodically and before exiting

This patch makes chasquid flush logs periodically, and also before exiting,
even if it gets killed via the usual signals (including Ctrl-C).

Both things help troubleshooting when there is not much traffic, or we kill
the server manually (including during automated tests).
---
 chasquid.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/chasquid.go b/chasquid.go
index ba1065e..7fecaae 100644
--- a/chasquid.go
+++ b/chasquid.go
@@ -13,8 +13,10 @@ import (
 	"net/mail"
 	"net/textproto"
 	"os"
+	"os/signal"
 	"path/filepath"
 	"strings"
+	"syscall"
 	"time"
 
 	"blitiri.com.ar/go/chasquid/internal/auth"
@@ -44,6 +46,11 @@ var (
 func main() {
 	flag.Parse()
 
+	setupSignalHandling()
+
+	defer glog.Flush()
+	go periodicallyFlushLogs()
+
 	// Seed the PRNG, just to prevent for it to be totally predictable.
 	rand.Seed(time.Now().UnixNano())
 
@@ -134,6 +141,25 @@ func loadDomain(s *Server, name, dir string) {
 	}
 }
 
+// Flush logs periodically, to help troubleshooting if there isn't that much
+// traffic.
+func periodicallyFlushLogs() {
+	for range time.Tick(5 * time.Second) {
+		glog.Flush()
+	}
+}
+
+// Set up signal handling, to flush logs when we get killed.
+func setupSignalHandling() {
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+	go func() {
+		<-c
+		glog.Flush()
+		os.Exit(1)
+	}()
+}
+
 type Server struct {
 	// Main hostname, used for display only.
 	Hostname string