pipebuilder_common 0.2.2

lib for pipebuilder components
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

use crate::{
    BaseConfig, HealthService, LeaseConfig, LeaseService, NodeConfig, NodeService, Register,
    RegisterConfig, Result,
};
use tokio::sync::oneshot::{self, Receiver};
use tracing::info;

pub async fn build_register(config: RegisterConfig) -> Result<Register> {
    Register::new(config).await
}

pub fn build_lease_service(config: LeaseConfig, lease_id: i64) -> LeaseService {
    LeaseService::new(config, lease_id)
}

pub fn build_node_service(config: NodeConfig, lease_id: i64) -> NodeService {
    NodeService::new(config, lease_id)
}

pub async fn bootstrap(
    config: BaseConfig,
) -> Result<(
    Register,
    NodeService,
    HealthService,
    LeaseService,
    Receiver<()>,
)> {
    info!("bootstrap base service");
    // build register
    let mut register = build_register(config.register).await?;
    // lease grant
    let ttl = config.lease.ttl;
    let resp = register.lease_grant(ttl as i64).await?;
    let lease_id = resp.id();
    // build services
    let lease_svc = build_lease_service(config.lease, lease_id);
    let lease_id = lease_svc.get_lease_id();
    let node_svc = build_node_service(config.node, lease_id);
    // run svc
    lease_svc.run(register.clone());
    let (shutdown_tx, shutdown_rx) = oneshot::channel::<()>();
    node_svc.run(register.clone(), shutdown_tx);
    let health_svc = HealthService::default();
    Ok((register, node_svc, health_svc, lease_svc, shutdown_rx))
}