Merge branch 'ztellman'

src/taoensso/nippy.clj

@@ -87,14 +87,29 @@
 
 (defmacro ^:private write-biginteger [s x] `(write-bytes ~s (.toByteArray ~x)))
 (defmacro ^:private write-utf8       [s x] `(write-bytes ~s (.getBytes ~x "UTF-8")))
-(defmacro ^:private freeze-to-stream
-  "Like `freeze-to-stream*` but with metadata support."
-  [s x]
-  `(let [x# ~x s# ~s]
-     (when-let [m# (meta x#)]
-       (write-id  s# ~id-meta)
-       (freeze-to-stream* m# s#))
-     (freeze-to-stream* x# s#)))
+
+(defn- freeze-to-stream
+  "Like `freeze-to-stream*`, but with metadata support and certain fast-paths encoded."
+  [^DataOutputStream s x]
+  (condp instance? x
+    Number (condp instance? x
+             Long    (do (write-id s id-long) (.writeLong s x))
+             Double  (do (write-id s id-double) (.writeDouble s x))
+             (freeze-to-stream* x s))
+    Keyword (do
+              (write-id s id-keyword)
+              (write-utf8 s
+                (if-let [ns (namespace x)]
+                  (str ns "/" (name x))
+                  (name x))))
+    String (do
+             (write-id s id-string)
+             (write-utf8 s ^String x))
+    (do
+      (when-let [m (meta x)]
+        (write-id s id-meta)
+        (freeze-to-stream* m s))
+      (freeze-to-stream* x s))))
 
 (defn freeze-to-stream!
   "Low-level API. Serializes arg (any Clojure data type) to a DataOutputStream."
@@ -114,17 +129,30 @@
   "Extends Freezable to simple collection types."
   [type id & body]
   `(freezer ~type ~id
+     (if (counted? ~'x)
+       (do
          (.writeInt ~'s (count ~'x))
-    (doseq [i# ~'x] (freeze-to-stream ~'s i#))))
+         (doseq [i# ~'x] (freeze-to-stream ~'s i#)))
+       (let [bas# (ByteArrayOutputStream.)
+             s# (DataOutputStream. bas#)
+             cnt# (loop [cnt# 0, x# ~'x]
+                    (if (empty? x#)
+                      cnt#
+                      (do
+                        (freeze-to-stream! s# (first x#))
+                        (recur (unchecked-inc cnt#) (rest x#)))))
+             ba# (.toByteArray bas#)]
+         (.writeInt ~'s cnt#)
+         (.write ~'s ba# 0 (alength ba#))))))
 
 (defmacro ^:private kv-freezer
   "Extends Freezable to key-value collection types."
   [type id & body]
   `(freezer ~type ~id
      (.writeInt ~'s (* 2 (count ~'x)))
-    (doseq [[k# v#] ~'x]
-      (freeze-to-stream ~'s k#)
-      (freeze-to-stream ~'s v#))))
+     (doseq [kv# ~'x]
+       (freeze-to-stream ~'s (key kv#))
+       (freeze-to-stream ~'s (val kv#)))))
 
 (freezer (Class/forName "[B") id-bytes   (write-bytes s ^bytes x))
 (freezer nil                  id-nil)

src/taoensso/nippy/benchmarks.clj

@@ -66,6 +66,23 @@
   ;; (bench {:reader? false :laps 1})
   ;; (bench {:reader? false :laps 2})
 
+  ;;; 11 Oct 2013: Nippy v2.2.0, with both ztellman mods
+  ;; {:defaults  {:round 4319, :freeze 2950, :thaw 1446, :data-size 12369}}
+  ;; {:encrypted {:round 7675, :freeze 4479, :thaw 3160, :data-size 12388}}
+  ;; {:fast      {:round 3928, :freeze 2530, :thaw 1269, :data-size 13277}}
+  ;; {:defaults-delta {:round 0.84 :freeze 0.79 :thaw 1.14}} ; vs 2.2.0
+
+  ;;; 11 Oct 2013: Nippy v2.2.0, with first ztellman mod
+  ;; {:defaults  {:round 4059, :freeze 2578, :thaw 1351, :data-size 12342}}
+  ;; {:encrypted {:round 7248, :freeze 4058, :thaw 3041, :data-size 12372}}
+  ;; {:fast      {:round 3430, :freeze 2085, :thaw 1229, :data-size 13277}}
+  ;; {:defaults-delta {:round  0.79 :freeze 0.69 :thaw 1.07}} ; vs 2.2.0
+
+  ;;; 11 Oct 2013: Nippy v2.2.0
+  ;; {:defaults  {:round 5135, :freeze 3711, :thaw 1266, :data-size 12393}}
+  ;; {:encrypted {:round 8655, :freeze 5323, :thaw 3036, :data-size 12420}}
+  ;; {:fast      {:round 4670, :freeze 3282, :thaw 1294, :data-size 13277}}
+
   ;;; 7 Auguest 2013: Nippy v2.2.0-RC1
   ;; {:reader    {:round 71582, :freeze 13656, :thaw 56730, :data-size 22964}}
   ;; {:defaults  {:round 5619,  :freeze 3710,  :thaw 1783,  :data-size 12368}}