From de94b9067d8fb4ca170604b111581b09f562bbb6 Mon Sep 17 00:00:00 2001
From: Eric Daniels <eric@erdaniels.com>
Date: Wed, 14 Aug 2024 18:13:07 -0400
Subject: [PATCH] set close state before unregisterStream

---
 association.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/association.go b/association.go
index 5be88389..29e9978c 100644
--- a/association.go
+++ b/association.go
@@ -573,6 +573,7 @@ func (a *Association) readLoop() {
 		a.closeWriteLoopOnce.Do(func() { close(a.closeWriteLoopCh) })
 
 		a.lock.Lock()
+		a.setState(closed)
 		for _, s := range a.streams {
 			a.unregisterStream(s, closeErr)
 		}