cancel transceivable.go context when closing is called

linxGnu · Sep 4, 2024 · aa97da7 · aa97da7
1 parent 0dcdeb9
commit aa97da7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/transceivable.go b/transceivable.go
@@ -177,6 +177,8 @@ func (t *transceivable) windowCleanup() {
 
 func (t *transceivable) closing(state State) {
 	if atomic.CompareAndSwapInt32(&t.aliveState, Alive, Closed) {
+		t.cancel()
+
 		t.in.closing(StoppingProcessOnly)
 		t.out.closing(StoppingProcessOnly)
 
@@ -187,6 +189,7 @@ func (t *transceivable) closing(state State) {
 		if t.settings.OnClosed != nil {
 			t.settings.OnClosed(state)
 		}
+		t.wg.Wait()
 	}
 	return
 }