From 57ad1b98fff64fd7cd81f1247cb543c1bbf16fe0 Mon Sep 17 00:00:00 2001
From: Amir Raminfar <findamir@gmail.com>
Date: Tue, 27 Nov 2018 12:54:32 -0800
Subject: [PATCH] Adds error channel for events

---
 main.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index b1676c2f..90d91c57 100644
--- a/main.go
+++ b/main.go
@@ -166,7 +166,7 @@ func streamEvents(w http.ResponseWriter, r *http.Request) {
 
     ctx, cancel := context.WithCancel(context.Background())
     defer cancel()
-    messages, _ := dockerClient.Events(ctx)
+    messages, error := dockerClient.Events(ctx)
 
 Loop:
     for {
@@ -195,6 +195,9 @@ Loop:
         case <-r.Context().Done():
             cancel()
             break Loop
+        case <-error:
+            cancel()
+            break Loop
         }
     }
 }