From 68def82d9070ef0f50ac5f2e3d1ddfedbb23c349 Mon Sep 17 00:00:00 2001
From: Will Brown <will@rebeagle.com>
Date: Sun, 2 Oct 2022 13:16:36 -0400
Subject: [PATCH] fix ondisconnect never resolving (#135)

---
 ntex-io/src/io.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ntex-io/src/io.rs b/ntex-io/src/io.rs
index e989fba4..0eea153f 100644
--- a/ntex-io/src/io.rs
+++ b/ntex-io/src/io.rs
@@ -893,6 +893,7 @@ impl OnDisconnect {
             Poll::Ready(())
         } else if let Some(on_disconnect) = self.inner.on_disconnect.take() {
             on_disconnect[self.token].register(cx.waker());
+            self.inner.on_disconnect.set(Some(on_disconnect));
             Poll::Pending
         } else {
             Poll::Ready(())