Skip to content

Instantly share code, notes, and snippets.

@roman
Created October 22, 2011 02:18
Show Gist options
  • Save roman/1305449 to your computer and use it in GitHub Desktop.
Save roman/1305449 to your computer and use it in GitHub Desktop.
Haskell Iteratees in clojure (naive impl)
(ns iteratee
(:require [clojure.contrib.types :as adt]))
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
(adt/defadt ::stream
eof
(chunks xs))
(adt/defadt ::iteratee
(yield value stream)
(continue consumer)
(error e))
(defn sum [xs] (reduce + 0 xs))
(defn $$ [enum it] (enum (it)))
(defn run_ [it]
(adt/match it
(error e) e
(yield result _) result
(continue consumer)
(do
(let [next-it (consumer eof)]
(do
(adt/match next-it
(error e) e
(yield result _) result
(continue _) (do
(println "missbehaving iteratee")
nil)))))))
(defn enum-seq [chunk-size xs0 it]
(let [ [a xs] (split-at chunk-size xs0) ]
(if (seq a)
(adt/match it
(error _) it
(yield _ _) it
(continue consumer)
(recur chunk-size xs (consumer (chunks a))))
it)))
(defn sum-it []
(letfn [(go [acc stream]
(adt/match stream
eof
(yield acc eof)
(chunks xs)
(do
(continue (partial go (+ (sum xs) acc))))))]
(continue (partial go 0))))
(defn -main [& args]
(run_ ($$ (partial enum-seq 5 [1 2 3 4 5 6]) sum-it)))
@roman
Copy link
Author

roman commented Oct 22, 2011

Maybe ($$) should be a macro...

@tavisrudd
Copy link

You can take out all the uses of (do 1sexp) and just leave 1sexp. You only need do in cases like (do 1sexp 2sexp 3sexp Nsexp).

@tavisrudd
Copy link

What does this buy you over Clojure's built-in sequence abstraction with first, rest, empty?, etc.?

@tavisrudd
Copy link

@roman
Copy link
Author

roman commented Oct 22, 2011

@tavisrudd:

  1. Yes, do's are there given that I was debugging and doing println in between sexp (lazy lazy not removing the do's, I know)

  2. This a really naive impl that tries to emulate haskell owns Iteratee framework, the advantage of using this approach is that the consumer has the ability to stop the stream generation, using this approach you will have another version of first, rest, empty using iteratees adt and stream adt.

  3. I have been wanting to check out that library, I did this snippet just to improve my clojure foo a little bit by playing with a concept really familiar to me in Haskell, I didn't intend to start replacing anything existing in the clojure library, it is likely this impl is not the "clojure way".

Thanks for your feedback :-).

@tavisrudd
Copy link

Yeah, I know you're just experimenting. I'm just not familiar enough with the Haskell way to spot the differences yet and am curious what they are so I can improve my understanding of it independent of the type system. The consumer is also in charge using the sequence abstraction. Ditto with Python's generator/iterators.

@tavisrudd
Copy link

Btw, would there be any harm removing $$ and just doing (run_ (enum-seq 5 [1 2 3 4 5 6] (sum-it)))? I'm not sure I understand its purpose. If you were to give it an English name, what would it be?

@roman
Copy link
Author

roman commented Oct 22, 2011

The sole purpose of ($$) is to apply an enumerator into an iteratee getting a new iterate out of that...

It would only abstract the notion of:

(partial enum-seq 5 [1 2 3 4 5 6]) and the calling of the (it) so you can just use it like:

($$ (enum-seq 5 [1 2 3 4 5 6]) it)

A problem I'm seeing that is a bit annoying though is that ($$ ... ...) is returning an "Iteratee" type instead of a function that returns an Iteratee type (as the first sum-it), and it will cause problems when chaining two ($$) calls together

($$ enum-terminal-input ($$ (enum-seq 5 [1 2 3 4 5 6]) sum-it))

@tavisrudd
Copy link

tavisrudd commented Oct 22, 2011 via email

@roman
Copy link
Author

roman commented Oct 22, 2011

I read the clojure.contrib.stream-utils lib and I found a hard time to understand it... don't know exactly how the actors involved work.

  1. Chunking is one
  2. Stream composition
  3. Stream transformation

Of course in this impl #3 is not supported, but in practice you are able to do that... One problem I'm seeing however is that you are doing recursion on the consumers, (as you can see I'm using recursion on the sum-it, which if there is to many things to sum, it is likely that is going to throw a StackLimitException).

We definitely need to play with the stream-utils (first understand it of course) and see if that library supports all of this features.

@tavisrudd
Copy link

tavisrudd commented Oct 22, 2011 via email

@tavisrudd
Copy link

tavisrudd commented Oct 22, 2011 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment