qshare 0.1.4

量化数据：股票、期货等
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
115
116
117
118
119
120
121
122
123
124
125
126
127

use std::{fs, io::Error, path::Path};

use crate::{const_vars, DataResult, DataResultFormat};
use mime::Mime;
use polars::export::chrono::Local;
use polars::frame::DataFrame;
use reqwest::{header, Request, Response};

///
/// http 请求工具类
///
pub struct HttpClient;

impl HttpClient {
    ///
    /// 执行http请求，调用call_back_body方法处理响应数据，返回DataResult
    ///
    pub(crate) async fn _exec(
        request: Request,
        call_back_body: fn(body: String) -> DataResult<DataFrame>,
    ) -> Result<DataResult<DataFrame>, anyhow::Error> {
        tracing::debug!("request url: {:?}", request);

        let http_client = reqwest::Client::new();

        let response = http_client.execute(request).await?;

        let body = response.text().await?;

        Ok(call_back_body(body))
    }

    ///
    /// 使用 DataResultFormat 对结果进行处理
    ///
    pub async fn exec_by_format(
        request: Request,
        format: impl DataResultFormat,
    ) -> Result<DataResult<DataFrame>, anyhow::Error> {
        tracing::debug!("request url: {:?}", request);

        let http_client = reqwest::Client::new();

        let response = http_client.execute(request).await?;

        match response.text().await {
            Ok(body) => {
                //1. 处理body
                if let Ok(data_frame) = format.to_dataframe(Some(body)) {
                    // 2. 格式化df
                    return Ok(format.format(data_frame.data));
                }
            }
            Err(e) => {
                tracing::warn!("http response text error:{}", e);
            }
        }

        Ok(DataResult::default())
    }

    /// 将服务器返回的 content-type 解析成 Mime 类型
    fn _get_content_type(resp: &Response) -> Option<Mime> {
        resp.headers()
            .get(header::CONTENT_TYPE)
            .map(|v| v.to_str().unwrap().parse().unwrap())
    }

    ///
    /// TODO 扩展根据响应content-type类型处理解析方法
    ///
    fn _process_body_to_dataframe(
        m: Option<Mime>,
        body: &String,
    ) -> Result<DataResult<DataFrame>, anyhow::Error> {
        match m {
            None => Ok(DataResult::<DataFrame>::from(body.to_string())),
            Some(v) if v == mime::APPLICATION_JSON => {
                Ok(DataResult::<DataFrame>::from(body.to_string()))
            }
            _ => Ok(DataResult::<DataFrame>::from(body.to_string())),
        }
    }
}

///
/// 环境工具
///
pub struct Envs;
impl Envs {
    ///
    /// 本地缓存目录
    ///
    pub fn cache_temp_home() -> String {
        dotenvy::var(const_vars::CACHE_TEMP_HOME).unwrap()
    }
}

pub struct DateUtils;

impl DateUtils {
    ///
    /// 年-月-日格式的日期
    ///
    pub fn now_fmt_ymd() -> String {
        let now = Local::now();
        now.format("%Y-%m-%d").to_string()
    }
}

pub struct IoUtils;

impl IoUtils {
    ///
    /// mkdir -p
    ///
    pub fn create_dir_recursive(path: &Path) -> Result<(), Error> {
        if path.exists() {
            Ok(())
        } else {
            match path.parent() {
                Some(parent) => IoUtils::create_dir_recursive(parent),
                None => fs::create_dir(path),
            }
        }
    }
}