dorea 0.4.0

A key-value storage system
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//! web server 启动器程序

use axum::extract::Extension;
use axum::routing::{get, post};
use axum::Router;

use doson::DataValue;
use std::collections::HashMap;
use std::path::Path;
use std::sync::Arc;

use self::routes::socket_handler;

pub mod db;
pub mod routes;
pub mod secret;
pub mod tools;

pub struct ShareState {
    pub(crate) config: (
        crate::configure::DoreaFileConfig,
        crate::configure::RestConfig,
    ),
    pub(crate) client_addr: (&'static str, u16),
}

pub async fn startup(addr: (&'static str, u16), document_path: &Path) -> crate::Result<()> {
    let hostname = addr.0;
    let dorea_port = addr.1;

    // 读取 rest-service path
    let rest_config = crate::configure::load_rest_config(document_path)?;

    if !rest_config.switch {
        return Ok(());
    }

    // 全局共享状态数据
    let share_state = Arc::new(ShareState {
        config: (
            crate::configure::load_config(document_path).unwrap(),
            rest_config.clone(),
        ),
        client_addr: (hostname, dorea_port),
    });

    let rest_port = rest_config.port;
    tokio::task::spawn(async move {
        // 测试数据库连接，并初始化必须数据：
        match crate::client::DoreaClient::connect(
            (hostname, dorea_port),
            &share_state.config.0.connection.connection_password,
        )
        .await
        {
            Ok(mut c) => {
                init_service_system_db(&mut c).await.unwrap();
            }
            Err(err) => {
                panic!("{}", err);
            }
        };

        let app = Router::new()
            .route("/", get(routes::index).post(routes::index))
            .route("/auth", post(routes::auth))
            .route("/ping", post(routes::ping))
            .route("/:group/:operation", post(routes::controller))
            .route("/_ws/", get(socket_handler))
            .layer(Extension(share_state));

        let addr = format!("{}:{}", hostname, rest_port);

        log::info!("⍹ >> Web-Service Running at: http://{}/", addr);

        hyper::Server::bind(&addr.parse().unwrap())
            .serve(app.into_make_service())
            .await
            .unwrap();
    });

    Ok(())
}

// this function will init the system data.
pub async fn init_service_system_db(client: &mut crate::client::DoreaClient) -> crate::Result<()> {
    client.select("system").await?;

    if client.get("service@accounts").await.is_none() {
        client
            .setex("service@accounts", DataValue::Dict(HashMap::new()), 0)
            .await?;
    }

    if client.get("service@acc-checker").await.is_none() {
        client
            .setex(
                "service@acc-checker",
                DataValue::String(crate::tool::rand_str()),
                0,
            )
            .await?;
    }

    client
        .setex(
            "service@startup-time",
            DataValue::Number(chrono::Local::now().timestamp() as f64),
            0,
        )
        .await?;

    Ok(())
}