Add more debugging output in WebsocketHandler

cloudfoundry-attic · Oct 7, 2014 · 3aacc5c · 3aacc5c
1 parent 00d4959
commit 3aacc5c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/server/handlers/websocket_handler.go b/server/handlers/websocket_handler.go
@@ -25,6 +25,7 @@ func (h *websocketHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
 	ws, err := websocket.Upgrade(rw, r, nil, 0, 0)
 	if err != nil {
 		http.Error(rw, "Not a websocket handshake", http.StatusBadRequest)
+		h.logger.Debugf("websocket handler: Not a websocket handshake: %s", err.Error())
 		return
 	}
 	defer ws.Close()
@@ -44,6 +45,7 @@ func (h *websocketHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
 	go func() {
 		server.NewKeepAlive(ws, h.keepAlive).Run()
 		close(keepAliveExpired)
+		h.logger.Debugf("websocket handler: Connection from %s timed out", r.RemoteAddr)
 	}()
 
 	for {
@@ -56,6 +58,7 @@ func (h *websocketHandler) ServeHTTP(rw http.ResponseWriter, r *http.Request) {
 			}
 			err = ws.WriteMessage(websocket.BinaryMessage, message)
 			if err != nil {
+				h.logger.Debugf("websocket handler: Error writing to websocket: %s", err.Error())
 				return
 			}
 		}