From 59c174e01b3e65e386a9d082d4de26ce577b7dd7 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Wed, 11 Dec 2013 16:31:59 -0800
Subject: [PATCH] packer/rpc: log when a muxconn is shut down

---
 packer/rpc/server.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/packer/rpc/server.go b/packer/rpc/server.go
index 5d11b403a..27ebafc98 100644
--- a/packer/rpc/server.go
+++ b/packer/rpc/server.go
@@ -31,11 +31,14 @@ type Server struct {
 	mux      *MuxConn
 	streamId uint32
 	server   *rpc.Server
+	closeMux bool
 }
 
 // NewServer returns a new Packer RPC server.
 func NewServer(conn io.ReadWriteCloser) *Server {
-	return NewServerWithMux(NewMuxConn(conn), 0)
+	result := NewServerWithMux(NewMuxConn(conn), 0)
+	result.closeMux = true
+	return result
 }
 
 func NewServerWithMux(mux *MuxConn, streamId uint32) *Server {
@@ -43,11 +46,17 @@ func NewServerWithMux(mux *MuxConn, streamId uint32) *Server {
 		mux:      mux,
 		streamId: streamId,
 		server:   rpc.NewServer(),
+		closeMux: false,
 	}
 }
 
 func (s *Server) Close() error {
-	return s.mux.Close()
+	if s.closeMux {
+		log.Printf("[WARN] Shutting down mux conn in Server")
+		return s.mux.Close()
+	}
+
+	return nil
 }
 
 func (s *Server) RegisterArtifact(a packer.Artifact) {