graph: Implement FromSql for a couple time related types

Author: lutter

graph/src/blockchain/types.rs

@@ -435,3 +435,9 @@ impl ToSql<Timestamptz, Pg> for BlockTime {
         <Timestamp as ToSql<Timestamptz, Pg>>::to_sql(&self.0, out)
     }
 }
+
+impl FromSql<Timestamptz, Pg> for BlockTime {
+    fn from_sql(bytes: diesel::pg::PgValue) -> diesel::deserialize::Result<Self> {
+        <Timestamp as FromSql<Timestamptz, Pg>>::from_sql(bytes).map(|ts| Self(ts))
+    }
+}

graph/src/data/store/scalar/timestamp.rs

@@ -2,6 +2,7 @@ use chrono::{DateTime, Utc};
 use diesel::deserialize::FromSql;
 use diesel::pg::PgValue;
 use diesel::serialize::ToSql;
+use diesel::sql_types::Timestamptz;
 use serde::{self, Deserialize, Serialize};
 use stable_hash::StableHash;
 
@@ -95,12 +96,19 @@ impl Display for Timestamp {
     }
 }
 
-impl ToSql<diesel::sql_types::Timestamptz, diesel::pg::Pg> for Timestamp {
+impl ToSql<Timestamptz, diesel::pg::Pg> for Timestamp {
     fn to_sql<'b>(
         &'b self,
         out: &mut diesel::serialize::Output<'b, '_, diesel::pg::Pg>,
     ) -> diesel::serialize::Result {
-        <_ as ToSql<diesel::sql_types::Timestamptz, _>>::to_sql(&self.0, &mut out.reborrow())
+        <_ as ToSql<Timestamptz, _>>::to_sql(&self.0, &mut out.reborrow())
+    }
+}
+
+impl FromSql<Timestamptz, diesel::pg::Pg> for Timestamp {
+    fn from_sql(bytes: PgValue) -> diesel::deserialize::Result<Self> {
+        let dt = <_ as FromSql<Timestamptz, _>>::from_sql(bytes)?;
+        Ok(Timestamp(dt))
     }
 }