http_pool/
route.rs

1use std::cmp::Ordering;
2use std::collections::HashMap;
3use std::fmt::Debug;
4
5/// 路由类型
6#[derive(Debug, Clone, Copy, PartialOrd, PartialEq, Ord, Eq)]
7pub enum RouteType {
8    /// 静态路径: /home/sub
9    Static,
10    /// 动态路径: /home/:page
11    Dynamic,
12    /// 通配符: /home/*
13    Wildcard,
14}
15
16/// 路由匹配结果
17#[derive(Clone)]
18pub enum RouteMatch<T: Clone> {
19    Matched {
20        handler: T,
21        params: HashMap<String, String>,
22    },
23    NotMatched,
24}
25
26/// 路由规则
27#[derive(Clone)]
28pub struct Route<T: Clone> {
29    path: String,
30    parts: Vec<String>,
31    handler: T,
32    route_type: RouteType,
33}
34
35/// 路由
36pub struct Router<T: Clone> {
37    routes: Vec<Route<T>>,
38}
39
40impl<T: Clone> Router<T> {
41    pub fn new() -> Self {
42        Router { routes: vec![] }
43    }
44
45    /// 添加路由规则
46    /// 支持以下路由规则
47    ///     1.静态路径匹配
48    ///     2.动态路径匹配，
49    ///     3.通配符匹配
50    pub fn add(&mut self, path: impl Into<String>, handler: T) {
51        let path = path.into();
52
53        let route_type = if path.contains("*") {
54            RouteType::Wildcard
55        } else if path.contains(":") {
56            RouteType::Dynamic
57        } else {
58            RouteType::Static
59        };
60
61        let parts = path
62            .trim_matches('/')
63            .split('/')
64            .map(|s| s.to_string())
65            .collect();
66
67        self.routes.push(Route {
68            path,
69            parts,
70            handler,
71            route_type,
72        });
73
74        // 按优先级排序（Static > Dynamic > Wildcard）
75        self.routes.sort_by(|a, b| a.route_type.cmp(&b.route_type))
76    }
77
78    /// 移除路由规则
79    pub fn remove(&mut self, path: &str) -> Option<Route<T>> {
80        for i in (0..self.routes.len()).rev() {
81            if path.cmp(&self.routes[i].path) == Ordering::Equal {
82                return Some(self.routes.remove(i));
83            }
84        }
85
86        None
87    }
88
89    /// 匹配路由规则
90    /// 支持以下路由规则
91    ///     1.静态路径匹配
92    ///     2.动态路径匹配，
93    ///     3.通配符匹配
94    pub fn go_match(&self, path: &str) -> RouteMatch<T> {
95        let parts: Vec<&str> = path.trim_matches('/').split('/').collect();
96
97        for route in &self.routes {
98            let route_parts: Vec<&str> = route.parts.iter().map(|s| s as &str).collect();
99            let mut params = HashMap::new();
100
101            if route_parts.len() != parts.len()
102                && !route_parts.iter().any(|part| part.starts_with('*'))
103            {
104                continue;
105            }
106
107            let mut matched = true;
108
109            for (i, route_part) in route_parts.iter().enumerate() {
110                if *route_part == "*" || route_part.starts_with('*') {
111                    // wildcard match
112                    let key = route_part.trim_start_matches('*');
113                    params.insert(key.to_string(), parts[i..].join("/"));
114                    break;
115                } else if route_part.starts_with(':') {
116                    // dynamic match
117                    let key = route_part.trim_start_matches(':');
118                    if let Some(val) = parts.get(i) {
119                        params.insert(key.to_string(), val.to_string());
120                    } else {
121                        matched = false;
122                        break;
123                    }
124                } else if let Some(req_part) = parts.get(i) {
125                    if route_part != req_part {
126                        matched = false;
127                        break;
128                    }
129                } else {
130                    matched = false;
131                    break;
132                }
133            }
134
135            if matched {
136                return RouteMatch::Matched {
137                    handler: route.handler.clone(),
138                    params,
139                };
140            }
141        }
142
143        RouteMatch::NotMatched
144    }
145}
http_pool/route.rs

http_pool/
route.rs
