From 3b0b066cc11879bc3c56a81dffc9f6f02224cc51 Mon Sep 17 00:00:00 2001
From: Milan Nikolic <gen2brain@gmail.com>
Date: Sun, 6 Oct 2019 02:43:58 +0200
Subject: [PATCH] Move context

---
 handlers/socket.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/handlers/socket.go b/handlers/socket.go
index 0890b70..16dc99f 100644
--- a/handlers/socket.go
+++ b/handlers/socket.go
@@ -32,6 +32,8 @@ func (s *Socket) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	ctx := context.Background()
+
 	for {
 		img, err := s.reader.Read()
 		if err != nil {
@@ -49,7 +51,7 @@ func (s *Socket) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 		b64 := image.EncodeToString(w.Bytes())
 
-		err = conn.Write(context.Background(), websocket.MessageText, []byte(b64))
+		err = conn.Write(ctx, websocket.MessageText, []byte(b64))
 		if err != nil {
 			break
 		}