Move context

This commit is contained in:
Milan Nikolic
2019-10-06 02:43:58 +02:00
parent 277d6c5b48
commit 3b0b066cc1

View File

@@ -32,6 +32,8 @@ func (s *Socket) ServeHTTP(w http.ResponseWriter, r *http.Request) {
return return
} }
ctx := context.Background()
for { for {
img, err := s.reader.Read() img, err := s.reader.Read()
if err != nil { if err != nil {
@@ -49,7 +51,7 @@ func (s *Socket) ServeHTTP(w http.ResponseWriter, r *http.Request) {
b64 := image.EncodeToString(w.Bytes()) 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 { if err != nil {
break break
} }