ns_game 0.2.26

ns_game lib.
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

use crate::config::ClickHouseConfig;
use crate::static_def::BASE_CONFIG;
use anyhow::Result;
use clickhouse::{Client, RowOwned, RowRead};

/// Clickhouse 查询
pub struct ClickHouseDB {
    client: Client,
}

impl ClickHouseDB {
    pub fn new(config: &ClickHouseConfig) -> Self {
        let client = Client::default()
            .with_user(&config.username)
            .with_password(&config.password)
            .with_database(&config.database)
            .with_url(&config.url);
        log::debug!("ClickHouseDB connected to {}", &config.url);
        Self { client }
    }

    /// 获取期牌数据
    #[inline]
    pub async fn get_drawing_data<T: RowOwned + RowRead>(
        &self,
        tag: i32,
        drawing: i32,
        index: i64,
        count: i64,
    ) -> Result<Vec<T>> {
        Ok(self
            .client
            .query(&format!(
                "select * from {} where tag=? and drawing=? and index>? order by index limit ?",
                BASE_CONFIG.drawing_table
            ))
            .bind(tag)
            .bind(drawing)
            .bind(index)
            .bind(count)
            .fetch_all()
            .await?)
    }

    /// 获取期牌数据
    #[inline]
    pub async fn get_drawing_data_by_index<T: RowOwned + RowRead>(
        &self,
        tag: i32,
        drawing: i32,
        index: i64,
    ) -> Result<Option<T>> {
        Ok(self
            .client
            .query(&format!(
                "select * from {} where tag=? and drawing=? and index=?",
                BASE_CONFIG.drawing_table
            ))
            .bind(tag)
            .bind(drawing)
            .bind(index)
            .fetch_optional()
            .await?)
    }

    /// 获取期牌期数集合
    #[inline]
    pub async fn get_drawing_ids(&self, tag: i32) -> Result<Vec<i32>> {
        Ok(self
            .client
            .query(&format!(
                "SELECT drawing FROM {} WHERE tag = ? GROUP BY drawing, drawing_per order by drawing",BASE_CONFIG.drawing_table
            ))
            .bind(tag)
            .fetch_all()
            .await?)
    }

    /// 获取期牌数量
    #[inline]
    pub async fn get_drawing_per(&self, tag: i32, drawing: i32) -> Result<i64> {
        Ok(self
            .client
            .query(&format!(
                "SELECT drawing_per FROM {} WHERE tag = ? and drawing = ? limit 1",
                BASE_CONFIG.drawing_table
            ))
            .bind(tag)
            .bind(drawing)
            .fetch_one()
            .await?)
    }
}