Close stream

cloudfoundry · Jun 4, 2020 · 0bb6325 · 0bb6325
1 parent 1deac6d
commit 0bb6325
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/jetstream/plugins/kubernetes/terminal/start.go b/src/jetstream/plugins/kubernetes/terminal/start.go
@@ -192,13 +192,15 @@ func pumpStdout(ws *websocket.Conn, source *websocket.Conn, done chan bool) {
 		_, r, err := source.ReadMessage()
 		if err != nil {
 			// Close
+			ws.Close()
 			done <- true
 			break
 		}
 		ws.SetWriteDeadline(time.Now().Add(writeWait))
 		bytes := fmt.Sprintf("% x\n", r[1:])
 		if err := ws.WriteMessage(websocket.TextMessage, []byte(bytes)); err != nil {
 			log.Errorf("Kubernetes Terminal failed to write message: %+v", err)
+			ws.Close()
 			break
 		}
 	}