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

Tries have a human-readable representation

Browse source
This commit is contained in:
Tommi Reiman 2017-11-11 16:39:11 +02:00
parent 2f6bfb33f8
commit 1f27021c29
1 changed files with 26 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
modules/reitit-core/src/reitit/trie.cljc
View file

@ -21,7 +21,9 @@
(lookup [this path params]) (lookup [this path params])
(get-segment [this]) (get-segment [this])
(update-segment [this subs lcs]) (update-segment [this subs lcs])
(get-data [this])
(set-data [this data]) (set-data [this data])
(get-chidren [this])
(add-child [this key child]) (add-child [this key child])
(insert-child [this key path-spec data])) (insert-child [this key path-spec data]))
@ -36,6 +38,7 @@
(let [?wild (maybe-wild-node children) (let [?wild (maybe-wild-node children)
?catch (maybe-catch-all-node children) ?catch (maybe-catch-all-node children)
children' (impl/fast-map children)] children' (impl/fast-map children)]
^{:type ::node}
(reify (reify
Node Node
(lookup [_ path params] (lookup [_ path params]
@ -51,26 +54,36 @@
(->Match data (assoc params param path))))) (->Match data (assoc params param path)))))
(get-segment [_] (get-segment [_]
segment) segment)
(get-data [_]
data)
(set-data [_ data] (set-data [_ data]
(wild-node segment param children data)) (wild-node segment param children data))
(get-chidren [_]
children)
(add-child [_ key child] (add-child [_ key child]
(wild-node segment param (assoc children key child) data)) (wild-node segment param (assoc children key child) data))
(insert-child [_ key path-spec child-data] (insert-child [_ key path-spec child-data]
(wild-node segment param (update children key insert path-spec child-data) data))))) (wild-node segment param (update children key insert path-spec child-data) data)))))
(defn- catch-all-node [segment children param data] (defn- catch-all-node [segment children param data]
^{:type ::node}
(reify (reify
Node Node
(lookup [_ path params] (lookup [_ path params]
(->Match data (assoc params param path))) (->Match data (assoc params param path)))
(get-segment [_] (get-segment [_]
segment))) segment)
(get-data [_]
data)
(get-chidren [_]
children)))
(defn- static-node [^String segment children data] (defn- static-node [^String segment children data]
(let [size (count segment) (let [size (count segment)
?wild (maybe-wild-node children) ?wild (maybe-wild-node children)
?catch (maybe-catch-all-node children) ?catch (maybe-catch-all-node children)
children' (impl/fast-map children)] children' (impl/fast-map children)]
^{:type ::node}
(reify (reify
Node Node
(lookup [_ path params] (lookup [_ path params]
@ -87,8 +100,12 @@
segment) segment)
(update-segment [_ subs lcs] (update-segment [_ subs lcs]
(static-node (subs segment lcs) children data)) (static-node (subs segment lcs) children data))
(get-data [_]
data)
(set-data [_ data] (set-data [_ data]
(static-node segment children data)) (static-node segment children data))
(get-chidren [_]
children)
(add-child [_ key child] (add-child [_ key child]
(static-node segment (assoc children key child) data)) (static-node segment (assoc children key child) data))
(insert-child [_ key path-spec child-data] (insert-child [_ key path-spec child-data]
@ -196,3 +213,11 @@
:else :else
(split node path-spec data lcs)))))) (split node path-spec data lcs))))))
(defn view
"Returns a view representation of a prefix-tree."
[x]
(vec (concat
[(get-segment x)]
(some->> (get-chidren x) vals seq (map view))
(some->> (get-data x) vector))))