Cleanup code in cmux and bench_test.

This commit is a partial cherry pick of pull request #3, for code cleanup.
This commit is contained in:
Tamir Duberstein
2015-12-19 22:34:29 -05:00
committed by Soheil Hassas Yeganeh
parent 89dd8ce1fd
commit 563c371a98
2 changed files with 6 additions and 12 deletions

14
cmux.go
View File

@@ -109,14 +109,12 @@ func (m *cMux) Serve() error {
func (m *cMux) serve(c net.Conn) {
muc := newMuxConn(c)
matched := false
for _, sl := range m.sls {
for _, s := range sl.ss {
matched = s(muc.sniffer())
matched := s(muc.sniffer())
muc.reset()
if matched {
select {
// TODO(soheil): threre is a possiblity of having unclosed connection.
case sl.l.connc <- muc:
case <-sl.l.donec:
c.Close()
@@ -126,12 +124,10 @@ func (m *cMux) serve(c net.Conn) {
}
}
if !matched {
c.Close()
err := ErrNotMatched{c: c}
if !m.handleErr(err) {
m.root.Close()
}
c.Close()
err := ErrNotMatched{c: c}
if !m.handleErr(err) {
m.root.Close()
}
}