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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use crate::NewWithDb;
use axum::{async_trait, extract::FromRequestParts};
use mongodb::Database;
use std::convert::Infallible;

/**

MongoDbServer
一个结构体，用于存储数据库和集合，一个axum State，为其实现了FromRequestParts，
从而可以通过MongodbServer extract获取数据库和集合
可以使用[`crate::inject`]以及[`crate::preload::DBServers`]简化extract的使用

*/
#[derive(Debug, Clone)]
pub struct MongoDbServer<T>
where
    T: Clone,
{
    pub db: Database,
    pub servers: T,
}

#[async_trait]
impl<T> NewWithDb for MongoDbServer<T>
where
    T: Clone + NewWithDb,
{
    async fn new(db: Database) -> Self {
        Self {
            servers: T::new(db.clone()).await,
            db,
        }
    }
}

#[async_trait]
impl<S, T> FromRequestParts<S> for MongoDbServer<T>
where
    S: Send + Sync,
    T: Clone + Send + Sync + 'static,
{
    type Rejection = Infallible;
    async fn from_request_parts(
        parts: &mut axum::http::request::Parts,
        _state: &S,
    ) -> Result<Self, Self::Rejection> {
        let dbs = parts
            .extensions
            .get::<Self>()
            .expect("can not get MongoDbServer");

        Ok(dbs.clone())
    }
}