mll-axum-utils 0.2.1

一个 Axum 的工具库
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use std::{
    collections::HashSet,
    net::{IpAddr, SocketAddr},
};

use axum::{
    body::Body,
    extract::ConnectInfo,
    http::Request,
    response::{IntoResponse, Response},
};

use super::Intercept;
use crate::res::Res;

#[derive(Clone)]
pub struct Blacklist {
    pub ips: HashSet<IpAddr>,
}

impl Intercept for Blacklist {
    fn before(&self, req: Request<Body>) -> Result<Request<Body>, Response> {
        if let Some(info) = req.extensions().get::<ConnectInfo<SocketAddr>>() {
            if self.ips.contains(&info.0.ip()) {
                return Err(Res::<()>::error("黑名单用户禁止访问").into_response());
            }
        }

        Ok(req)
    }
}