webbase 0.0.18

webbase
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

use super::influxclient::get_influxdb;
use crate::time_util;
use anyhow::anyhow;
#[allow(unused_imports)]
use chrono::{DateTime, FixedOffset, Utc};
use influxdb::InfluxDbWriteable;
use influxdb::{ReadQuery, WriteQuery};
use tracing::error;

static SYS_STATUS_ID: &str = "system_status";

#[derive(InfluxDbWriteable)]
pub struct TsSysStatus {
    pub time: DateTime<Utc>,
    pub data: String,
}

impl TsSysStatus {
    pub async fn write(data: String) {
        let client = get_influxdb();
        let tsdata = TsSysStatus {
            time: time_util::get_current_time(),
            data: data,
        }
        .into_query(SYS_STATUS_ID);
        let result = client.query(tsdata).await;
        if result.is_err() {
            error!("write ts data error: {:?}", result.err());
        }
    }

    pub async fn write_batch(data: Vec<String>) {
        let client = get_influxdb();
        let tsdata: Vec<WriteQuery> = data
            .into_iter()
            .map(|v| {
                TsSysStatus {
                    time: time_util::get_current_time(),
                    data: v,
                }
                .into_query(SYS_STATUS_ID)
            })
            .collect();
        let result = client.query(tsdata).await;
        if result.is_err() {
            error!("write ts data error: {:?}", result.err());
        }
    }

    pub async fn read() -> anyhow::Result<()> {
        let client = get_influxdb();
        let read_query = ReadQuery::new(format!("SELECT * FROM {}", SYS_STATUS_ID));
        let read_result = client.query(read_query).await;
        if read_result.is_err() {
            return Err(anyhow!("write ts data error: {:?}", read_result.err()));
        }
        let tsdata = read_result.unwrap();
        println!("{:?}", tsdata);
        anyhow::Ok(())
    }
}