Import rusqlite-0.23.1

Change-Id: Id1ca7bcaad7820f463bfcce6945d80fc1d6918f5

1 files changed, 79 insertions, 0 deletions

diff --git a/src/types/time.rs b/src/types/time.rs
new file mode 100644
index 0000000..097b22a
--- /dev/null
+++ b/src/types/time.rs
@@ -0,0 +1,79 @@
+use crate::types::{FromSql, FromSqlError, FromSqlResult, ToSql, ToSqlOutput, ValueRef};
+use crate::Result;
+
+const CURRENT_TIMESTAMP_FMT: &str = "%Y-%m-%d %H:%M:%S";
+const SQLITE_DATETIME_FMT: &str = "%Y-%m-%dT%H:%M:%S.%fZ";
+const SQLITE_DATETIME_FMT_LEGACY: &str = "%Y-%m-%d %H:%M:%S:%f %Z";
+
+impl ToSql for time::Timespec {
+    fn to_sql(&self) -> Result<ToSqlOutput<'_>> {
+        let time_string = time::at_utc(*self)
+            .strftime(SQLITE_DATETIME_FMT)
+            .unwrap()
+            .to_string();
+        Ok(ToSqlOutput::from(time_string))
+    }
+}
+
+impl FromSql for time::Timespec {
+    fn column_result(value: ValueRef<'_>) -> FromSqlResult<Self> {
+        value
+            .as_str()
+            .and_then(|s| {
+                match s.len() {
+                    19 => time::strptime(s, CURRENT_TIMESTAMP_FMT),
+                    _ => time::strptime(s, SQLITE_DATETIME_FMT).or_else(|err| {
+                        time::strptime(s, SQLITE_DATETIME_FMT_LEGACY).or_else(|_| Err(err))
+                    }),
+                }
+                .or_else(|err| Err(FromSqlError::Other(Box::new(err))))
+            })
+            .map(|tm| tm.to_timespec())
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use crate::{Connection, Result, NO_PARAMS};
+
+    fn checked_memory_handle() -> Connection {
+        let db = Connection::open_in_memory().unwrap();
+        db.execute_batch("CREATE TABLE foo (t TEXT, i INTEGER, f FLOAT)")
+            .unwrap();
+        db
+    }
+
+    #[test]
+    fn test_timespec() {
+        let db = checked_memory_handle();
+
+        let mut ts_vec = vec![];
+
+        ts_vec.push(time::Timespec::new(10_000, 0)); //January 1, 1970 2:46:40 AM
+        ts_vec.push(time::Timespec::new(10_000, 1000)); //January 1, 1970 2:46:40 AM (and one microsecond)
+        ts_vec.push(time::Timespec::new(1_500_391_124, 1_000_000)); //July 18, 2017
+        ts_vec.push(time::Timespec::new(2_000_000_000, 2_000_000)); //May 18, 2033
+        ts_vec.push(time::Timespec::new(3_000_000_000, 999_999_999)); //January 24, 2065
+        ts_vec.push(time::Timespec::new(10_000_000_000, 0)); //November 20, 2286
+
+        for ts in ts_vec {
+            db.execute("INSERT INTO foo(t) VALUES (?)", &[&ts]).unwrap();
+
+            let from: time::Timespec = db
+                .query_row("SELECT t FROM foo", NO_PARAMS, |r| r.get(0))
+                .unwrap();
+
+            db.execute("DELETE FROM foo", NO_PARAMS).unwrap();
+
+            assert_eq!(from, ts);
+        }
+    }
+
+    #[test]
+    fn test_sqlite_functions() {
+        let db = checked_memory_handle();
+        let result: Result<time::Timespec> =
+            db.query_row("SELECT CURRENT_TIMESTAMP", NO_PARAMS, |r| r.get(0));
+        assert!(result.is_ok());
+    }
+}