rusqlite/types/
time.rs

1//! [`ToSql`] and [`FromSql`] implementation for [`time::OffsetDateTime`].
2use crate::types::{FromSql, FromSqlError, FromSqlResult, ToSql, ToSqlOutput, ValueRef};
3use crate::Result;
4use time::{OffsetDateTime, PrimitiveDateTime, UtcOffset};
5
6const CURRENT_TIMESTAMP_FMT: &str = "%Y-%m-%d %H:%M:%S";
7const SQLITE_DATETIME_FMT: &str = "%Y-%m-%dT%H:%M:%S.%NZ";
8const SQLITE_DATETIME_FMT_LEGACY: &str = "%Y-%m-%d %H:%M:%S:%N %z";
9
10impl ToSql for OffsetDateTime {
11    #[inline]
12    fn to_sql(&self) -> Result<ToSqlOutput<'_>> {
13        let time_string = self.to_offset(UtcOffset::UTC).format(SQLITE_DATETIME_FMT);
14        Ok(ToSqlOutput::from(time_string))
15    }
16}
17
18impl FromSql for OffsetDateTime {
19    fn column_result(value: ValueRef<'_>) -> FromSqlResult<Self> {
20        value.as_str().and_then(|s| {
21            match s.len() {
22                19 => PrimitiveDateTime::parse(s, CURRENT_TIMESTAMP_FMT).map(|d| d.assume_utc()),
23                _ => PrimitiveDateTime::parse(s, SQLITE_DATETIME_FMT)
24                    .map(|d| d.assume_utc())
25                    .or_else(|err| {
26                        OffsetDateTime::parse(s, SQLITE_DATETIME_FMT_LEGACY).map_err(|_| err)
27                    }),
28            }
29            .map_err(|err| FromSqlError::Other(Box::new(err)))
30        })
31    }
32}
33
34#[cfg(test)]
35mod test {
36    use crate::{Connection, Result};
37    use std::time::Duration;
38    use time::OffsetDateTime;
39
40    fn checked_memory_handle() -> Result<Connection> {
41        let db = Connection::open_in_memory()?;
42        db.execute_batch("CREATE TABLE foo (t TEXT, i INTEGER, f FLOAT)")?;
43        Ok(db)
44    }
45
46    #[test]
47    fn test_offset_date_time() -> Result<()> {
48        let db = checked_memory_handle()?;
49
50        let mut ts_vec = vec![];
51
52        let make_datetime =
53            |secs, nanos| OffsetDateTime::from_unix_timestamp(secs) + Duration::from_nanos(nanos);
54
55        ts_vec.push(make_datetime(10_000, 0)); //January 1, 1970 2:46:40 AM
56        ts_vec.push(make_datetime(10_000, 1000)); //January 1, 1970 2:46:40 AM (and one microsecond)
57        ts_vec.push(make_datetime(1_500_391_124, 1_000_000)); //July 18, 2017
58        ts_vec.push(make_datetime(2_000_000_000, 2_000_000)); //May 18, 2033
59        ts_vec.push(make_datetime(3_000_000_000, 999_999_999)); //January 24, 2065
60        ts_vec.push(make_datetime(10_000_000_000, 0)); //November 20, 2286
61
62        for ts in ts_vec {
63            db.execute("INSERT INTO foo(t) VALUES (?)", &[&ts])?;
64
65            let from: OffsetDateTime = db.query_row("SELECT t FROM foo", [], |r| r.get(0))?;
66
67            db.execute("DELETE FROM foo", [])?;
68
69            assert_eq!(from, ts);
70        }
71        Ok(())
72    }
73
74    #[test]
75    fn test_sqlite_functions() -> Result<()> {
76        let db = checked_memory_handle()?;
77        let result: Result<OffsetDateTime> =
78            db.query_row("SELECT CURRENT_TIMESTAMP", [], |r| r.get(0));
79        assert!(result.is_ok());
80        Ok(())
81    }
82}
rusqlite/types/time.rs

rusqlite/types/
time.rs
