RxPortal; Close (#132)

openziti · Aug 26, 2021 · 8572db4 · 8572db4
1 parent 858ba31
commit 8572db4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/rxportal.go b/rxportal.go
@@ -23,6 +23,7 @@ type RxPortal struct {
 	txp          *TxPortal
 	seq          *util.Sequence
 	closer       *Closer
+	closed       bool
 }
 
 type RxRead struct {
@@ -114,5 +115,13 @@ func (rxp *RxPortal) Rx(wm *WireMessage) (err error) {
 	return err
 }
 
+func (rxp *RxPortal) Close() {
+	if !rxp.closed {
+		rxp.reads <- &RxRead{nil, 0, true}
+		rxp.closed = true
+		close(rxp.rxs)
+	}
+}
+
 func (rxp *RxPortal) run() {
 }