From 1ad1b879a10cb0e92ddf01a0fdd093489b61cd83 Mon Sep 17 00:00:00 2001
From: tharvik <tharvik@users.noreply.github.com>
Date: Mon, 4 Jan 2021 15:30:23 +0100
Subject: [PATCH] websocket: do not error on service's close

---
 websocket.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/websocket.go b/websocket.go
index 6843a92a..5c8af372 100644
--- a/websocket.go
+++ b/websocket.go
@@ -350,9 +350,11 @@ outerReadLoop:
 				break outerReadLoop
 			case reply, ok := <-outChan:
 				if !ok {
-					err = xerrors.New("service finished streaming")
+					ws.WriteControl(websocket.CloseMessage,
+						websocket.FormatCloseMessage(websocket.CloseNormalClosure, "service finished streaming"),
+						time.Now().Add(time.Millisecond*500))
 					close(clientInputs)
-					break outerReadLoop
+					return
 				}
 				tx += len(reply)