From af35d1826ec79503040f0ca1deb94cb4e7e60b13 Mon Sep 17 00:00:00 2001
From: xtaci <daniel820313@gmail.com>
Date: Sat, 3 Sep 2016 23:08:31 +0800
Subject: [PATCH] optimize Sesion.Close()

---
 session.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/session.go b/session.go
index 2537450..4da9327 100644
--- a/session.go
+++ b/session.go
@@ -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()