From cea6c1658bd621d5f82cdeab3562e44fa6a8a4be Mon Sep 17 00:00:00 2001
From: Richard Patel <terorie@alphakevin.club>
Date: Sun, 3 Feb 2019 17:02:44 +0100
Subject: [PATCH] Bugfix: Don't schedule new tasks during shutdown

---
 scheduler.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/scheduler.go b/scheduler.go
index 6a00420..c7fa4cb 100644
--- a/scheduler.go
+++ b/scheduler.go
@@ -21,8 +21,13 @@ var totalQueued int64
 func Schedule(c context.Context, remotes <-chan *OD) {
 	go Stats(c)
 
-	for remote := range remotes {
-		if !scheduleNewTask(c, remote) {
+	for {
+		select {
+		case remote := <-remotes:
+			if !scheduleNewTask(c, remote) {
+				return
+			}
+		case <-c.Done():
 			return
 		}
 	}