have concat close the streams, not just the stream-of-streams, fixes #82

clj-commons · May 27, 2016 · f6fe13e · f6fe13e
1 parent 71ba8cb
commit f6fe13e
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/manifold/stream.clj b/src/manifold/stream.clj
@@ -795,10 +795,16 @@
     (d/loop []
       (d/chain' (take! in ::none)
         (fn [s']
-          (if (identical? ::none s')
+          (cond
+            (closed? out)
+            (close! s')
+
+            (identical? ::none s')
             (do
               (close! out)
               s')
+
+            :else
             (d/loop []
               (d/chain' (take! s' ::none)
                 (fn [msg]
@@ -807,7 +813,7 @@
                     (put! out msg)))
                 (fn [result]
                   (case result
-                    false (close! in)
+                    false (do (close! s') (close! in))
                     ::none nil
                     (d/recur)))))))
         (fn [result]