luciano/reitit
1
0
Fork 0
mirror of https://github.com/metosin/reitit.git synced 2025-12-16 16:01:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tailor keywordize implementation to concrete types

Browse source 
Even faster
This commit is contained in:
Ben Sless 2024-08-25 18:50:43 +03:00
parent 7ab6021630
commit dcb7258caf
1 changed files with 38 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

77
modules/reitit-core/src/reitit/walk.clj
View file

@ -1,52 +1,51 @@
(ns ^:no-doc reitit.walk)
(defprotocol Walkable
(-walk [coll f]))
(defprotocol IKeywordize
(-keywordize [coll]))
(extend-protocol Walkable
nil
(-walk [_ _] nil)
Object
(-walk [x _] x)
clojure.lang.IMapEntry
(-walk [e f] (clojure.lang.MapEntry. (f (.key e)) (f (.val e))))
clojure.lang.ISeq
(-walk [coll f] (map f coll))
clojure.lang.PersistentList
(-walk [coll f] (apply list (map f coll)))
clojure.lang.PersistentList$EmptyList
(-walk [x _] x)
clojure.lang.IRecord
(-walk [r f] (reduce (fn [r x] (conj r (f x))) r r)))
(defn- -keywordize-map
[m]
(persistent!
(reduce-kv
(fn [m k v]
(if (string? k)
(assoc! m (keyword k) (-keywordize v))
(assoc! m (-keywordize k) (-keywordize v))))
(transient (empty m))
m)))
(defn- -walk-default
[coll f]
(into (empty coll) (map f) coll))
(defn- -keywordize-default
[coll]
(into (empty coll) (map -keywordize) coll))
(doseq [type [clojure.lang.PersistentArrayMap
clojure.lang.PersistentHashMap
clojure.lang.PersistentHashSet
(doseq [type [clojure.lang.PersistentHashSet
clojure.lang.PersistentVector
clojure.lang.PersistentQueue
clojure.lang.PersistentStructMap
clojure.lang.PersistentTreeMap
clojure.lang.PersistentTreeSet]]
(extend type Walkable {:-walk -walk-default}))
(extend type IKeywordize {:-keywordize -keywordize-default}))
(defn walk
[inner outer form]
(outer (-walk form inner)))
(doseq [type [clojure.lang.PersistentArrayMap
clojure.lang.PersistentHashMap]]
(extend type IKeywordize {:-keywordize -keywordize-map}))
(defn postwalk [f form] (walk (partial postwalk f) f form))
(extend-protocol IKeywordize
nil
(-keywordize [_] nil)
Object
(-keywordize [x] x)
clojure.lang.MapEntry
(-keywordize [e] (clojure.lang.MapEntry/create
(-keywordize (.key e))
(-keywordize (.val e))))
clojure.lang.ISeq
(-keywordize [coll] (map -keywordize coll))
clojure.lang.PersistentList
(-keywordize [coll] (apply list (map -keywordize coll)))
clojure.lang.PersistentList$EmptyList
(-keywordize [x] x)
clojure.lang.IRecord
(-keywordize [r] (reduce (fn [r x] (conj r (-keywordize x))) r r)))
(defn- keywordize
[m]
(persistent!
(reduce-kv
(fn [m k v] (if (string? k) (assoc! m (keyword k) v) m))
(transient {})
m)))
(defn keywordize-keys
[m]
(postwalk (fn [x] (if (map? x) (keywordize m) x)) m))
(defn keywordize-keys [m] (-keywordize m))