From 045e5977875f4a7ffb571b42401543b9d78bac80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Magiera?= <magik6k@gmail.com>
Date: Sat, 11 Jul 2020 01:21:48 +0200
Subject: [PATCH] remove open windows when dropping workers

---
 sched.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/sched.go b/sched.go
index d926e753a..241440beb 100644
--- a/sched.go
+++ b/sched.go
@@ -528,6 +528,16 @@ func (sh *scheduler) dropWorker(wid WorkerID) {
 	w := sh.workers[wid]
 	delete(sh.workers, wid)
 
+	newWindows := make([]*schedWindowRequest, 0, len(sh.openWindows))
+	for _, window := range sh.openWindows {
+		if window.worker != wid {
+			newWindows = append(newWindows, window)
+		}
+	}
+	sh.openWindows = newWindows
+
+	// TODO: sync close worker goroutine
+
 	go func() {
 		if err := w.w.Close(); err != nil {
 			log.Warnf("closing worker %d: %+v", err)