No longer accept to send after close

snapview · Sep 12, 2019 · bed9bd4 · bed9bd4
1 parent 534a2b1
commit bed9bd4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/protocol/mod.rs b/src/protocol/mod.rs
@@ -231,7 +231,9 @@ impl WebSocketContext {
         Stream: Read + Write,
     {
         // Do not write to already closed connections.
-        self.state.check_active()?;
+        if !self.state.is_active() {
+            return Err(Error::AlreadyClosed);
+        }
 
         if let Some(max_send_queue) = self.config.max_send_queue {
             if self.send_queue.len() >= max_send_queue {