distinct-by

core.clj

(defn distinct-by
  "Like clojure.core/distinct, but applys f"
  ([f]
   (fn [rf]
     (let [seen (volatile! #{})]
       (fn
         ([] (rf))
         ([result] (rf result))
         ([result input]
          (let [id (f input)]
            (if (contains? @seen id)
              result
              (do (vswap! seen conj id)
                  (rf result input)))))))))
  ([op coll]
   (let [step (fn step [xs seen]
                (lazy-seq
                  ((fn [[f :as xs] seen]
                     (when-let [s (seq xs)]
                       (let [v (op f)]
                         (if (contains? seen v)
                           (recur (rest s) seen)
                           (cons f (step (rest s) (conj seen v)))))))
                    xs seen)))]
     (step coll #{}))))