wip

ntex-rs · May 12, 2024 · 0598c1c · 0598c1c
1 parent 61b7c43
commit 0598c1c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/v5/dispatcher.rs b/src/v5/dispatcher.rs
@@ -548,6 +548,7 @@ where
         Err(err) => {
             // do not handle nested error
             if error {
+                inner.sink.drop_sink();
                 return Err(err);
             } else {
                 // handle error from control service
@@ -562,20 +563,19 @@ where
         }
     };
 
-    if error {
+    let response = if error {
         if let Some(pkt) = result.packet {
             let _ = inner.sink.encode_packet(pkt);
         }
-        if result.disconnect {
-            inner.sink.drop_sink();
-        }
         Ok(None)
     } else {
-        if result.disconnect {
-            inner.sink.drop_sink();
-        }
         Ok(result.packet)
+    };
+
+    if result.disconnect {
+        inner.sink.drop_sink();
     }
+    response
 }
 
 #[cfg(test)]