Add IOverride cache

Closes #87

Author: bsless

src/fipp/edn.cljc

@@ -2,7 +2,7 @@
   "Provides a pretty document serializer and pprint fn for Clojure/EDN forms.
   See fipp.clojure for pretty printing Clojure code."
   (:require [clojure.string :as str]
-            [fipp.ednize :refer [edn record->tagged]]
+            [fipp.ednize :refer [edn record->tagged IOverride]]
             [fipp.visit :refer [visit visit*]]
             [fipp.engine :refer (pprint-document)]))
 
@@ -19,10 +19,26 @@
                    [:span sep "..."])]
     [:group open [:align ys ellipsis] close]))
 
-(defrecord EdnPrinter [symbols print-meta print-length print-level]
+(defn- cache-satisfies
+  ([prot]
+   (let [cache (volatile! {})]
+     (fn [x]
+       (cache-satisfies prot cache x))))
+  ([prot cache x]
+   (let [clazz (type x)
+         ret (get @cache clazz ::not-found)]
+     (if (identical? ::not-found ret)
+       (let [ret (satisfies? prot x)]
+         (vswap! cache assoc clazz ret)
+         ret)
+       ret))))
+
+(defrecord EdnPrinter [symbols print-meta print-length print-level cache]
 
   fipp.visit/IVisitor
 
+  (visit-ovrride? [_ x]
+    (cache-satisfies IOverride cache x))
 
   (visit-unknown [this x]
     (visit this (edn x)))
@@ -96,6 +112,7 @@
   ([x] (pretty x {}))
   ([x options]
    (let [defaults {:symbols {}
+                   :cache (volatile! {})
                    :print-length *print-length*
                    :print-level *print-level*
                    :print-meta *print-meta*}

src/fipp/visit.cljc

@@ -27,14 +27,16 @@
   (visit-var [this x])
   (visit-pattern [this x])
   (visit-record [this x])
+
+  (visit-ovrride? [this x])
   )
 
 (defn visit*
   "Visits objects, ignoring metadata."
   [visitor x]
   (cond
     (nil? x) (visit-nil visitor)
-    (override? x) (visit-unknown visitor x)
+    (visit-ovrride? visitor x) (visit-unknown visitor x)
     (util/boolean? x) (visit-boolean visitor x)
     (string? x) (visit-string visitor x)
     (util/char? x) (visit-character visitor x)