make sure streaming response channel is closed when connection is closed

clj-commons · Sep 6, 2012 · a312591 · a312591
1 parent 3ac67b3
commit a312591
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/aleph/http/core.clj b/src/aleph/http/core.clj
@@ -172,7 +172,13 @@
 
 (defn expand-writes [f honor-keep-alive? ch]
   (let [ch* (channel)
-        default-charset (options/charset)]
+        default-charset (options/charset)
+        current-stream (atom nil)]
+
+    (on-closed ch
+      #(when-let [stream @current-stream]
+         (close stream)))
+
     (bridge-join ch "aleph.http.core/expand-writes"
       (fn [m]
         (let [{:keys [msg chunks write-callback]} (f m)
@@ -184,6 +190,8 @@
                               (close ch*)
                               true))]
 
+          (reset! current-stream chunks)
+
           (if-not chunks
 
             ;; non-streaming response