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

use tardis::{
    basic::{error::TardisError, result::TardisResult},
    log,
};

use crate::config::gateway_dto::SgGateway;

use self::http_route_dto::SgHttpRoute;

#[cfg(feature = "k8s")]
pub mod config_by_k8s;
#[cfg(feature = "local")]
pub mod config_by_local;
#[cfg(feature = "cache")]
pub mod config_by_redis;
pub mod gateway_dto;
pub mod http_route_dto;
pub mod plugin_filter_dto;

pub async fn init(k8s_mode: bool, namespace_or_conf_uri: Option<String>, check_interval_sec: Option<u64>) -> TardisResult<Vec<(SgGateway, Vec<SgHttpRoute>)>> {
    log::info!(
        "[SG.Config] Config initialization mode: {}",
        if k8s_mode {
            format!("kubernetes, with namespaces: {namespace_or_conf_uri:?}")
        } else {
            format!("non-kubernetes, with uri: {namespace_or_conf_uri:?}")
        }
    );
    if k8s_mode {
        #[cfg(feature = "k8s")]
        {
            config_by_k8s::init(namespace_or_conf_uri).await
        }
        #[cfg(not(feature = "k8s"))]
        {
            Err(tardis::basic::error::TardisError::not_found(
                "[SG.Config] The current compilation mode does not support k8s",
                "",
            ))
        }
    } else {
        let conf_uri = namespace_or_conf_uri.ok_or_else(|| TardisError::not_found("[SG.Config] The configuration path must be specified in the current mode", ""))?;
        #[cfg(feature = "cache")]
        {
            return config_by_redis::init(&conf_uri, check_interval_sec.unwrap_or(10)).await;
        }
        #[cfg(feature = "local")]
        {
            return config_by_local::init(&conf_uri, check_interval_sec.unwrap_or(10)).await;
        }
        Err(tardis::basic::error::TardisError::not_found("[SG.Config] The current compilation mode does not exist", ""))
    }
}