From 7f6a6a03c0add9888a6f9465db1ff640ea93fbd3 Mon Sep 17 00:00:00 2001
From: Amir Raminfar <findamir@gmail.com>
Date: Thu, 21 Jan 2021 10:50:09 -0800
Subject: [PATCH] Fixes possible goroutine leak

---
 docker/client.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/docker/client.go b/docker/client.go
index c02ab03c..99063f32 100644
--- a/docker/client.go
+++ b/docker/client.go
@@ -148,11 +148,15 @@ func (d *dockerClient) ContainerStats(ctx context.Context, id string, stats chan
 			)
 
 			if cpuPercent > 0 || memUsage > 0 {
-				stats <- ContainerStat{
+				select {
+				case <-ctx.Done():
+					return
+				case stats <- ContainerStat{
 					ID:            id,
 					CPUPercent:    cpuPercent,
 					MemoryPercent: memPercent,
 					MemoryUsage:   memUsage,
+				}:
 				}
 			}
 		}