axol 0.2.0

Axol Web Framework
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

use std::{net::SocketAddr, ops::Deref};

use anyhow::anyhow;
use axol_http::request::RequestPartsRef;

use crate::{Error, FromRequestParts, Result};

#[derive(Debug, Clone, Copy)]
pub struct ConnectInfo(pub SocketAddr);

impl Deref for ConnectInfo {
    type Target = SocketAddr;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

#[async_trait::async_trait]
impl<'a> FromRequestParts<'a> for ConnectInfo {
    async fn from_request_parts(request: RequestPartsRef<'a>) -> Result<Self> {
        let info = request
            .extensions
            .get::<ConnectInfo>()
            .ok_or_else(|| Error::internal(anyhow!("missing ConnectInfo extension")))?;
        Ok(*info)
    }
}