krx-rs 0.1.0

KRX Open API를 위한 Rust 클라이언트
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

use crate::error::{Error, Result};
use chrono::{Duration, Local};

/// 날짜 문자열이 "YYYYMMDD" 형식인지 검증합니다.
pub fn is_valid_date_format(date: &str) -> bool {
    date.len() == 8 && date.chars().all(|c| c.is_numeric())
}

/// `Option<String>` 형태의 날짜를 검증하고, 유효하면 `String`을 반환합니다.
///
/// # Errors
/// - 날짜가 `None`일 경우 `Error::InvalidInput` 반환.
/// - 날짜 형식이 유효하지 않을 경우 `Error::InvalidInput` 반환.
pub fn validate_base_date(base_date: Option<String>) -> Result<String> {
    let date = base_date.ok_or_else(|| {
        Error::InvalidInput("date is required, call `date()` or `latest()`".to_string())
    })?;

    if !is_valid_date_format(&date) {
        return Err(Error::InvalidInput(
            "date must be in YYYYMMDD format".to_string(),
        ));
    }

    Ok(date)
}

/// KRX API 조회를 위한 가장 최신 거래일(보통 전일)의 날짜 문자열을 생성합니다.
pub fn latest_workday_string() -> String {
    let yesterday = Local::now() - Duration::days(1);
    yesterday.format("%Y%m%d").to_string()
}

/// 기본 빌더 매크로 - 공통 날짜 설정 메서드들을 생성
#[macro_export]
macro_rules! impl_date_builder_methods {
    ($builder_type:ty) => {
        impl<'a> $builder_type {
            /// 조회 기준일자 설정 (YYYYMMDD).
            ///
            /// KRX 데이터는 2010년 이후부터 조회일 기준 전일까지만 제공됩니다.
            ///
            /// # Example
            /// ```
            /// .date("20240105")
            /// ```
            pub fn date(mut self, date: impl Into<String>) -> Self {
                self.base_date = Some(date.into());
                self
            }

            /// 가장 최신 거래일(보통 전일)의 데이터로 설정합니다.
            pub fn latest(mut self) -> Self {
                self.base_date = Some($crate::api::common::latest_workday_string());
                self
            }
        }
    };
}