-
-
Save erikprice/2aa047fd06238f4c0b73 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
(use 'clojure.core.async) | |
(def output (atom [])) | |
(defn producer [ctrl k] | |
(go (loop [i 0] | |
(when-let [c (<! ctrl)] | |
(>! c [k i]) | |
(>! ctrl c) | |
(recur (inc i)))))) | |
(defn consumer | |
([stop] | |
(consumer stop nil)) | |
([stop buf-or-n] | |
(let [ctrl (chan 1) | |
c (chan buf-or-n)] | |
(>!! ctrl c) | |
(thread | |
(loop [] | |
(let [[v p] (alts!! [stop c] :priority true)] | |
(if-not (identical? stop p) | |
(do (swap! output conj v) | |
(Thread/sleep 500) ; <- for the sake of the demo | |
(recur)) | |
(close! ctrl))))) | |
ctrl))) | |
(defn start | |
([producers] | |
(start producers nil)) | |
([producers buf-or-n] | |
(let [stop (chan 1) | |
ctrl (consumer stop buf-or-n)] | |
(dotimes [k producers] | |
(producer ctrl k)) | |
(fn [] | |
(>!! stop true))))) | |
(comment | |
(def stop (start 10)) | |
(count @output) | |
) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment