rnacos 0.8.3

Nacos server re-implemented in Rust.
Documentation 
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use crate::ldap::ldap_conn::LdapConnActorAddr;
use crate::ldap::ldap_msg_actor::LdapMsgActor;
use crate::ldap::model::actor_model::{LdapMsgReq, LdapMsgResult};
use crate::ldap::model::LdapConfig;
use crate::user::UserManager;
use actix::prelude::*;
use bean_factory::{bean, BeanFactory, FactoryData, Inject};
use std::sync::Arc;

#[bean(inject)]
pub struct LdapManager {
    ldap_config: Arc<LdapConfig>,
    ldap_conn_addr: Option<LdapConnActorAddr>,
    enable_ldap: bool,
    user_manager_addr: Option<Addr<UserManager>>,
}

impl LdapManager {
    pub fn new(ldap_config: Arc<LdapConfig>, enable_ldap: bool) -> Self {
        Self {
            ldap_config,
            ldap_conn_addr: None,
            enable_ldap,
            user_manager_addr: None,
        }
    }

    pub fn init(&mut self, ctx: &mut Context<Self>) {
        if !self.enable_ldap {
            return;
        }
        let ldap_config = self.ldap_config.clone();
        let user_manager_addr = self.user_manager_addr.clone();
        async move { LdapConnActorAddr::new_from_config(ldap_config, user_manager_addr).await }
            .into_actor(self)
            .map(|res, act, _ctx| match res {
                Ok(res) => {
                    act.ldap_conn_addr = Some(res);
                }
                Err(err) => {
                    log::error!("init ldap error:{}", err);
                }
            })
            .wait(ctx);
    }

    async fn handle_req(
        req: LdapMsgReq,
        ldap_msg_addr: Option<Addr<LdapMsgActor>>,
    ) -> anyhow::Result<LdapMsgResult> {
        let ldap_msg_addr = if let Some(v) = ldap_msg_addr {
            v
        } else {
            return Err(anyhow::anyhow!("ldap_msg_addr is None"));
        };
        ldap_msg_addr.send(req).await?
    }
}

impl Actor for LdapManager {
    type Context = Context<Self>;
    fn started(&mut self, _ctx: &mut Self::Context) {
        log::info!("LdapManager started");
    }
}

impl Inject for LdapManager {
    type Context = Context<Self>;

    fn inject(
        &mut self,
        factory_data: FactoryData,
        _factory: BeanFactory,
        ctx: &mut Self::Context,
    ) {
        self.user_manager_addr = factory_data.get_actor();
        self.init(ctx);
    }
}

impl Handler<LdapMsgReq> for LdapManager {
    type Result = ResponseActFuture<Self, anyhow::Result<LdapMsgResult>>;

    fn handle(&mut self, msg: LdapMsgReq, _ctx: &mut Self::Context) -> Self::Result {
        let msg_addr = self
            .ldap_conn_addr
            .as_ref()
            .map(|v| v.ldap_msg_actor.clone());
        let fut = Self::handle_req(msg, msg_addr)
            .into_actor(self)
            .map(|result, _act, _ctx| result);
        Box::pin(fut)
    }
}