From 37ae21001dfc042c2da8570bd31a44d34de53ecb Mon Sep 17 00:00:00 2001
From: Benny Ng <benny.tpng@gmail.com>
Date: Tue, 7 Jun 2016 13:01:03 +0800
Subject: [PATCH] Close idle connections after graceful shutdown timeout

---
 caddyhttp/httpserver/server.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/caddyhttp/httpserver/server.go b/caddyhttp/httpserver/server.go
index 2156bcf4..37ed0165 100644
--- a/caddyhttp/httpserver/server.go
+++ b/caddyhttp/httpserver/server.go
@@ -52,6 +52,16 @@ func NewServer(addr string, group []*SiteConfig) (*Server, error) {
 		connTimeout: GracefulTimeout,
 	}
 	s.Server.Handler = s // this is weird, but whatever
+	s.Server.ConnState = func(c net.Conn, cs http.ConnState) {
+		if cs == http.StateIdle {
+			s.listenerMu.Lock()
+			// server stopped, close idle connection
+			if s.listener == nil {
+				c.Close()
+			}
+			s.listenerMu.Unlock()
+		}
+	}
 
 	// Disable HTTP/2 if desired
 	if !HTTP2 {
@@ -288,6 +298,7 @@ func (s *Server) Stop() (err error) {
 	s.listenerMu.Lock()
 	if s.listener != nil {
 		err = s.listener.Close()
+		s.listener = nil
 	}
 	s.listenerMu.Unlock()