From 4f0a33ac3d5de20505022455625020cbbd575071 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kenfe-Micka=C3=ABl=20Laventure?=
 <mickael.laventure@gmail.com>
Date: Tue, 30 Aug 2016 09:47:21 -0700
Subject: [PATCH] Dump runtime stacks on USR1 reception (#292)

Signed-off-by: Kenfe-Mickael Laventure <mickael.laventure@gmail.com>
---
 containerd/main.go | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/containerd/main.go b/containerd/main.go
index a6eee73..ee255b7 100644
--- a/containerd/main.go
+++ b/containerd/main.go
@@ -6,6 +6,7 @@ import (
 	"net"
 	"os"
 	"os/signal"
+	"runtime"
 	"strings"
 	"sync"
 	"syscall"
@@ -87,6 +88,32 @@ var daemonFlags = []cli.Flag{
 	},
 }
 
+// DumpStacks dumps the runtime stack.
+func dumpStacks() {
+	var (
+		buf       []byte
+		stackSize int
+	)
+	bufferLen := 16384
+	for stackSize == len(buf) {
+		buf = make([]byte, bufferLen)
+		stackSize = runtime.Stack(buf, true)
+		bufferLen *= 2
+	}
+	buf = buf[:stackSize]
+	logrus.Infof("=== BEGIN goroutine stack dump ===\n%s\n=== END goroutine stack dump ===", buf)
+}
+
+func setupDumpStacksTrap() {
+	c := make(chan os.Signal, 1)
+	signal.Notify(c, syscall.SIGUSR1)
+	go func() {
+		for range c {
+			dumpStacks()
+		}
+	}()
+}
+
 func main() {
 	logrus.SetFormatter(&logrus.TextFormatter{TimestampFormat: time.RFC3339Nano})
 	app := cli.NewApp()
@@ -99,6 +126,7 @@ func main() {
 	app.Usage = usage
 	app.Flags = daemonFlags
 	app.Before = func(context *cli.Context) error {
+		setupDumpStacksTrap()
 		if context.GlobalBool("debug") {
 			logrus.SetLevel(logrus.DebugLevel)
 			if context.GlobalDuration("metrics-interval") > 0 {
@@ -106,7 +134,6 @@ func main() {
 					return err
 				}
 			}
-
 		}
 		if p := context.GlobalString("pprof-address"); len(p) > 0 {
 			pprof.Enable(p)