clickhouse_data_value/
date.rs

1use core::{
2    ops::{Deref, DerefMut},
3    str::FromStr,
4};
5
6use chrono::NaiveDate as ChronoNaiveDate;
7use serde::{
8    de::{self, Visitor},
9    Deserialize, Deserializer,
10};
11
12#[derive(PartialEq, Debug, Clone)]
13pub struct NaiveDate(pub ChronoNaiveDate);
14impl From<ChronoNaiveDate> for NaiveDate {
15    fn from(inner: ChronoNaiveDate) -> Self {
16        Self(inner)
17    }
18}
19impl Deref for NaiveDate {
20    type Target = ChronoNaiveDate;
21
22    fn deref(&self) -> &Self::Target {
23        &self.0
24    }
25}
26impl DerefMut for NaiveDate {
27    fn deref_mut(&mut self) -> &mut Self::Target {
28        &mut self.0
29    }
30}
31
32pub type ParseError = chrono::ParseError;
33impl FromStr for NaiveDate {
34    type Err = ParseError;
35
36    fn from_str(s: &str) -> Result<Self, Self::Err> {
37        ChronoNaiveDate::parse_from_str(s, "%Y-%m-%d").map(Into::into)
38    }
39}
40
41struct NaiveDateVisitor;
42impl<'de> Visitor<'de> for NaiveDateVisitor {
43    type Value = NaiveDate;
44
45    fn expecting(&self, formatter: &mut core::fmt::Formatter) -> core::fmt::Result {
46        formatter.write_str("format %Y-%m-%d")
47    }
48
49    fn visit_str<E>(self, string: &str) -> Result<Self::Value, E>
50    where
51        E: de::Error,
52    {
53        string
54            .parse()
55            .map_err(|err: ParseError| de::Error::custom(err.to_string()))
56    }
57}
58impl<'de> Deserialize<'de> for NaiveDate {
59    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
60    where
61        D: Deserializer<'de>,
62    {
63        deserializer.deserialize_str(NaiveDateVisitor)
64    }
65}
66pub fn deserialize<'de, D>(d: D) -> Result<ChronoNaiveDate, D::Error>
67where
68    D: de::Deserializer<'de>,
69{
70    d.deserialize_str(NaiveDateVisitor).map(|x| x.0)
71}
72
73#[cfg(test)]
74mod tests {
75    use super::*;
76
77    use chrono::NaiveDate;
78
79    #[test]
80    fn test_parse() -> Result<(), Box<dyn std::error::Error>> {
81        assert_eq!(
82            "2021-03-01"
83                .parse::<NaiveDate>()
84                .map_err(|err| err.to_string())?,
85            NaiveDate::from_ymd_opt(2021, 3, 1).expect("")
86        );
87
88        Ok(())
89    }
90
91    #[derive(Deserialize)]
92    struct Row {
93        #[serde(deserialize_with = "crate::date::deserialize")]
94        date: chrono::NaiveDate,
95    }
96
97    #[test]
98    fn test_de() -> Result<(), Box<dyn std::error::Error>> {
99        let deserializer = de::IntoDeserializer::<de::value::Error>::into_deserializer;
100        assert_eq!(
101            super::deserialize(deserializer("2021-03-01")).unwrap(),
102            NaiveDate::from_ymd_opt(2021, 3, 1).expect("")
103        );
104
105        let content = include_str!("../tests/files/date.txt");
106        let line = content.lines().next().unwrap();
107
108        let Row { date } = serde_json::from_str(line)?;
109        assert_eq!(date, NaiveDate::from_ymd_opt(2021, 3, 1).expect(""));
110
111        Ok(())
112    }
113}
clickhouse_data_value/date.rs

clickhouse_data_value/
date.rs
