fix a potential race in closing stream.die

xtaci · Sep 22, 2016 · 9f2b528 · 9f2b528
1 parent bc68dab
commit 9f2b528
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/stream.go b/stream.go
@@ -111,7 +111,12 @@ func (s *Stream) Close() error {
 func (s *Stream) sessionClose() {
 	s.dieLock.Lock()
 	defer s.dieLock.Unlock()
-	close(s.die)
+
+	select {
+	case <-s.die:
+	default:
+		close(s.die)
+	}
 }
 
 // pushBytes a slice into buffer