From 1e2aa4fc9062c608f40aa98c0807f660bb8a8457 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Tue, 27 Feb 2018 18:05:06 -0800
Subject: [PATCH] mark context as modified after writing data

---
 src/context.rs    | 1 +
 src/ws/context.rs | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/context.rs b/src/context.rs
index 02eda8d5c..aa6f4c49a 100644
--- a/src/context.rs
+++ b/src/context.rs
@@ -164,6 +164,7 @@ impl<A, S> HttpContext<A, S> where A: Actor<Context=Self> {
             self.stream = Some(SmallVec::new());
         }
         self.stream.as_mut().map(|s| s.push(frame));
+        self.inner.modify();
     }
 
     /// Handle of the running future
diff --git a/src/ws/context.rs b/src/ws/context.rs
index 8720b461c..56320c895 100644
--- a/src/ws/context.rs
+++ b/src/ws/context.rs
@@ -112,6 +112,7 @@ impl<A, S> WebsocketContext<A, S> where A: Actor<Context=Self> {
             }
             let stream = self.stream.as_mut().unwrap();
             stream.push(ContextFrame::Chunk(Some(data)));
+            self.inner.modify();
         } else {
             warn!("Trying to write to disconnected response");
         }
@@ -179,6 +180,7 @@ impl<A, S> WebsocketContext<A, S> where A: Actor<Context=Self> {
             self.stream = Some(SmallVec::new());
         }
         self.stream.as_mut().map(|s| s.push(frame));
+        self.inner.modify();
     }
 
     /// Handle of the running future