Fix dmsg transport.HandleFrame logic.

skycoin · Jun 21, 2019 · e9e11d0 · e9e11d0
1 parent 3896594
commit e9e11d0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/pkg/dmsg/transport.go b/pkg/dmsg/transport.go
@@ -130,18 +130,18 @@ func (tp *Transport) Type() string {
 
 // HandleFrame allows 'tp.Serve' to handle the frame (typically from 'ClientConn').
 func (tp *Transport) HandleFrame(f Frame) error {
-	if tp.IsClosed() {
-		return io.ErrClosedPipe
-	}
-
 	tp.inMx.RLock()
 	defer tp.inMx.RUnlock()
 
+handleFrame:
 	select {
-	case <-tp.done:
-		return io.ErrClosedPipe
 	case tp.inCh <- f:
 		return nil
+	default:
+		if tp.IsClosed() {
+			return io.ErrClosedPipe
+		}
+		goto handleFrame
 	}
 }