luciano/specter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ALL and MAP-VALS perf improvement for small maps

Browse source
This commit is contained in:
Nathan Marz 2016-06-15 17:18:58 -04:00
parent e8f0a873fa
commit c709b16eab
1 changed files with 35 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
src/clj/com/rpl/specter/impl.cljx
View file

@ -837,7 +837,24 @@
(all-transform [structure next-fn] (all-transform [structure next-fn]
(mapv next-fn structure)) (mapv next-fn structure))
#+clj clojure.lang.PersistentArrayMap #+cljs cljs.core/PersistentArrayMap #+clj
clojure.lang.PersistentArrayMap
#+clj
(all-transform [structure next-fn]
(let [k-it (.keyIterator structure)
v-it (.valIterator structure)]
(loop [ret {}]
(if (.hasNext k-it)
(let [k (.next k-it)
v (.next v-it)
[newk newv] (next-fn [k v])]
(recur (assoc ret newk newv)))
ret
))))
#+cljs
cljs.core/PersistentArrayMap
#+cljs
(all-transform [structure next-fn] (all-transform [structure next-fn]
(non-transient-map-all-transform structure next-fn {}) (non-transient-map-all-transform structure next-fn {})
) )
@ -923,7 +940,23 @@
nil nil
) )
#+clj clojure.lang.PersistentArrayMap #+cljs cljs.core/PersistentArrayMap #+clj
clojure.lang.PersistentArrayMap
#+clj
(map-vals-transform [structure next-fn]
(let [k-it (.keyIterator structure)
v-it (.valIterator structure)]
(loop [ret {}]
(if (.hasNext k-it)
(let [k (.next k-it)
v (.next v-it)]
(recur (assoc ret k (next-fn v))))
ret
))))
#+cljs
cljs.core/PersistentArrayMap
#+cljs
(map-vals-transform [structure next-fn] (map-vals-transform [structure next-fn]
(map-vals-non-transient-transform structure {} next-fn) (map-vals-non-transient-transform structure {} next-fn)
) )