From 4667bb9f0ed528739bd8173840553399191e086b Mon Sep 17 00:00:00 2001
From: Clint Armstrong <clint@clintarmstrong.net>
Date: Fri, 9 Feb 2018 20:41:15 -0500
Subject: [PATCH] expose shutdown

---
 sdk/handler.go | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/sdk/handler.go b/sdk/handler.go
index c0d042e..16f55ba 100644
--- a/sdk/handler.go
+++ b/sdk/handler.go
@@ -1,6 +1,7 @@
 package sdk
 
 import (
+	"context"
 	"crypto/tls"
 	"fmt"
 	"net"
@@ -13,7 +14,8 @@ const activatePath = "/Plugin.Activate"
 // Handler is the base to create plugin handlers.
 // It initializes connections and sockets to listen to.
 type Handler struct {
-	mux *http.ServeMux
+	mux    *http.ServeMux
+	server *http.Server
 }
 
 // NewHandler creates a new Handler with an http mux.
@@ -25,16 +27,21 @@ func NewHandler(manifest string) Handler {
 		fmt.Fprintln(w, manifest)
 	})
 
-	return Handler{mux: mux}
+	server := &http.Server{
+		Handler: mux,
+	}
+
+	return Handler{mux: mux, server: server}
 }
 
 // Serve sets up the handler to serve requests on the passed in listener
 func (h Handler) Serve(l net.Listener) error {
-	server := http.Server{
-		Addr:    l.Addr().String(),
-		Handler: h.mux,
-	}
-	return server.Serve(l)
+	h.server.Addr = l.Addr().String()
+	return h.server.Serve(l)
+}
+
+func (h Handler) Shutdown(ctx context.Context) error {
+	return h.server.Shutdown(ctx)
 }
 
 // ServeTCP makes the handler to listen for request in a given TCP address.