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

cljs trie

Browse source
This commit is contained in:
Tommi Reiman 2019-02-08 19:17:22 +02:00
parent 5b9f90d283
commit 54d5550fae
1 changed files with 117 additions and 115 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,8 @@
(ns reitit.trie (ns reitit.trie
(:refer-clojure :exclude [compile]) (:refer-clojure :exclude [compile -assoc!])
(:require [clojure.string :as str]) (:require [clojure.string :as str])
(:import [reitit Trie Trie$Match Trie$Matcher] #?(:clj (:import [reitit Trie Trie$Match Trie$Matcher]
(java.net URLDecoder))) (java.net URLDecoder))))
(defrecord Wild [value]) (defrecord Wild [value])
(defrecord CatchAll [value]) (defrecord CatchAll [value])
@ -13,7 +13,7 @@
(defn catch-all? [x] (instance? CatchAll x)) (defn catch-all? [x] (instance? CatchAll x))
(defprotocol Matcher (defprotocol Matcher
(match [this i max ^chars path]) (match [this i max path])
(view [this]) (view [this])
(depth [this])) (depth [this]))
@ -116,27 +116,27 @@
(update :children dissoc "")) (update :children dissoc ""))
node'))) node')))
(set! *warn-on-reflection* true)
(defn decode! (defn decode!
([chars start end] ([path start end]
(let [s (String. ^chars chars ^int start ^int (- end start))] #?(:clj (let [s (subs path start end)]
(if (str/index-of s \%) (if (str/index-of s \%)
(URLDecoder/decode (URLDecoder/decode
(if (str/index-of s \+) (.replace ^String s "+" "%2B") s) (if (str/index-of s \+) (.replace ^String s "+" "%2B") s)
"UTF-8") "UTF-8")
s))) s))
([chars start end percent? plus?] :cljs (js/decodeURIComponent (subs path start end))))
(let [s (String. ^chars chars ^int start ^int (- end start))] ([path start end percent? plus?]
#?(:clj (let [s (String. ^chars path ^int start ^int (- end start))]
(if percent? (if percent?
(URLDecoder/decode (URLDecoder/decode
(if plus? (.replace ^String s "+" "%2B") s) (if plus? (.replace ^String s "+" "%2B") s)
"UTF-8") "UTF-8")
s)))) s))
:cljs (js/decodeURIComponent (subs path start end)))))
(defn data-matcher [data] (defn data-matcher [data]
#?(:cljx (Trie/dataMatcher data) #?(:clj (Trie/dataMatcher data)
:clj (let [match (->Match data nil)] :cljs (let [match (->Match data nil)]
(reify Matcher (reify Matcher
(match [_ i max _] (match [_ i max _]
(if (= i max) (if (= i max)
@ -144,31 +144,30 @@
(view [_] data) (view [_] data)
(depth [_] 1))))) (depth [_] 1)))))
(defn static-matcher [^String path matcher] (defn static-matcher [path matcher]
#?(:cljx (Trie/staticMatcher path matcher) #?(:clj (Trie/staticMatcher ^String path ^Trie$Matcher matcher)
:clj (let [^chars chars (.toCharArray path) :cljs (let [size (count path)]
size (alength chars)]
(reify Matcher (reify Matcher
(match [_ i max path] (match [_ i max p]
(if-not (< max (+ ^int i size)) (if-not (< max (+ i size))
(loop [j 0] (loop [j 0]
(if (= j size) (if (= j size)
(match matcher (+ ^int i size) max path) (match matcher (+ i size) max p)
(if (= ^char (aget ^chars path (+ ^int i j)) ^char (aget ^chars chars j)) (if (= (get p (+ i j)) (get path j))
(recur (inc j))))))) (recur (inc j)))))))
(view [_] [path (view matcher)]) (view [_] [path (view matcher)])
(depth [_] (inc (depth matcher))))))) (depth [_] (inc (depth matcher)))))))
(defn wild-matcher [key matcher] (defn wild-matcher [key matcher]
#?(:cljx (Trie/wildMatcher key matcher) #?(:clj (Trie/wildMatcher key matcher)
:clj (reify Matcher :cljs (reify Matcher
(match [_ i max path] (match [_ i max path]
(if (and (< ^int i ^int max) (not= ^char (aget ^chars path i) \/)) (if (and (< i max) (not= (get path i) \/))
(loop [percent? false, plus? false, ^int j ^int i] (loop [percent? false, plus? false, j i]
(if (= ^int max j) (if (= max j)
(if-let [match (match matcher max max path)] (if-let [match (match matcher max max path)]
(-assoc! match key (decode! path i max percent? plus?))) (-assoc! match key (decode! path i max percent? plus?)))
(let [c ^char (aget ^chars path j)] (let [c ^char (get path j)]
(case c (case c
\/ (if-let [match (match matcher j max path)] \/ (if-let [match (match matcher j max path)]
(-assoc! match key (decode! path i j percent? plus?))) (-assoc! match key (decode! path i j percent? plus?)))
@ -179,24 +178,24 @@
(depth [_] (inc (depth matcher)))))) (depth [_] (inc (depth matcher))))))
(defn catch-all-matcher [key data] (defn catch-all-matcher [key data]
#?(:cljx (Trie/catchAllMatcher key data) #?(:clj (Trie/catchAllMatcher key data)
:clj (let [match (->Match data nil)] :cljs (let [match (->Match data nil)]
(reify Matcher (reify Matcher
(match [_ i max path] (match [_ i max path]
(if (< ^int i max) (if (< i max)
(-assoc! match key (decode! path i max)))) (-assoc! match key (decode! path i max))))
(view [_] [key [data]]) (view [_] [key [data]])
(depth [_] 1))))) (depth [_] 1)))))
(defn linear-matcher [matchers] (defn linear-matcher [matchers]
#?(:cljx (Trie/linearMatcher matchers) #?(:clj (Trie/linearMatcher matchers)
:clj (let [matchers (.toArray ^java.util.List (reverse (sort-by depth matchers))) :cljs (let [matchers (vec (reverse (sort-by depth matchers)))
size (alength matchers)] size (count matchers)]
(reify Matcher (reify Matcher
(match [_ i max path] (match [_ i max path]
(loop [j 0] (loop [j 0]
(if (< j size) (if (< j size)
(or (match (aget matchers j) i max path) (or (match (get matchers j) i max path)
(recur (inc j)))))) (recur (inc j))))))
(view [_] (mapv view matchers)) (view [_] (mapv view matchers))
(depth [_] (apply max (map depth matchers))))))) (depth [_] (apply max (map depth matchers)))))))
@ -227,24 +226,24 @@
(first matchers)))) (first matchers))))
(defn pretty [matcher] (defn pretty [matcher]
#?(:cljx (-> matcher str read-string eval) #?(:clj (-> matcher str read-string eval)
:clj (view matcher))) :cljs (view matcher)))
(defn lookup [matcher ^String path] (defn lookup [matcher path]
#?(:cljx (if-let [match ^Trie$Match (Trie/lookup ^Trie$Matcher matcher ^String path)] #?(:clj (if-let [match ^Trie$Match (Trie/lookup ^Trie$Matcher matcher ^String path)]
(->Match (.data match) (.parameters match))) (->Match (.data match) (.parameters match)))
:clj (let [chars (.toCharArray path)] :cljs (if-let [match (match matcher 0 (count path) path)]
(if-let [match (match matcher 0 (alength chars) chars)]
(let [params (if-let [path-params (:path-params match)] (let [params (if-let [path-params (:path-params match)]
(persistent! path-params) (persistent! path-params)
{})] {})]
(assoc match :path-params params)))))) (assoc match :path-params params)))))
;; ;;
;; spike ;; spike
;; ;;
(-> (comment
(->
[["/v2/whoami" 1] [["/v2/whoami" 1]
["/v2/users/:user-id/datasets" 2] ["/v2/users/:user-id/datasets" 2]
["/v2/public/projects/:project-id/datasets" 3] ["/v2/public/projects/:project-id/datasets" 3]
@ -302,10 +301,13 @@
(compile) (compile)
(pretty)) (pretty))
(-> [["/kikka" 2] (-> [["/kikka" 2]
["/kikka/kakka/kukka" 3] ["/kikka/kakka/kukka" 3]
["/kikka/:kakka/kurkku" 4] ["/kikka/:kakka/kurkku" 4]
["/kikka/kuri/{user/doc}/html" 5]] ["/kikka/kuri/{user/doc}/html" 5]]
(insert) (insert)
(compile) (compile)
(pretty)) (pretty))
(map str (.toCharArray "\u2215\u0048\u0065\u006C\u006C\u006F"))
(count ["∕" "H" "e" "l" "l" "o" " " "W" "o" "r" "l" "d"]))