optimize Sesion.Close()

2024-04-21 10:51:48 +00:00 · 2016-09-03 23:08:31 +08:00
parent b2ffd23649
commit af35d1826e
1 changed files with 5 additions and 0 deletions
@@ -104,6 +104,11 @@ func (s *Session) Close() error {
 	case <-s.die:
 		return errors.New(errBrokenPipe)
 	default:
+		s.streamLock.Lock()
+		for k := range s.streams {
+			s.streams[k].sessionClose()
+		}
+		s.streamLock.Unlock()
 		s.conn.Close()
 		close(s.die)
 		s.bucketCond.Signal()