interrupt: also protect cancel with the mutex

Michael Muré created 6 years ago

Change summary

util/interrupt/cleaner.go | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)

Detailed changes

util/interrupt/cleaner.go

@@ -33,7 +33,12 @@ func RegisterCleaner(cleaner CleanerFunc) CancelFunc {
 	defer mu.Unlock()
 
 	w := &wrapper{f: cleaner}
-	cancel := func() { w.disabled = true }
+
+	cancel := func() {
+		mu.Lock()
+		defer mu.Unlock()
+		w.disabled = true
+	}
 
 	// prepend to later execute then in reverse order
 	cleaners = append([]*wrapper{w}, cleaners...)