archive derive of PartialEq for rkyv - 0.4.x (#959)

.github/workflows/test.yml

@@ -75,7 +75,7 @@ jobs:
  - uses: taiki-e/install-action@cargo-hack
  - uses: Swatinem/rust-cache@v2
  - run: |
- cargo hack check --feature-powerset --optional-deps serde,rkyv \
+ cargo hack check --feature-powerset --optional-deps serde \
  --skip __internal_bench,iana-time-zone,pure-rust-locales,libc,winapi \
  --all-targets
  # run using `bash` on all platforms for consistent

Cargo.toml

@@ -26,14 +26,15 @@ clock = ["std", "winapi", "iana-time-zone", "android-tzdata"]
 oldtime = []
 wasmbind = ["wasm-bindgen", "js-sys"]
 unstable-locales = ["pure-rust-locales"]
+rkyv-validation = ["rkyv/validation"]
 __internal_bench = []
 
 [dependencies]
 num-traits = { version = "0.2", default-features = false }
 rustc-serialize = { version = "0.3.20", optional = true }
 serde = { version = "1.0.99", default-features = false, optional = true }
 pure-rust-locales = { version = "0.7", optional = true }
-rkyv = { version = "0.7", optional = true }
+rkyv = { version = "0.7.41", optional = true }
 arbitrary = { version = "1.0.0", features = ["derive"], optional = true }
 
 [target.'cfg(all(target_arch = "wasm32", not(any(target_os = "emscripten", target_os = "wasi"))))'.dependencies]

src/date.rs

@@ -479,7 +479,7 @@ impl<Tz: TimeZone> Eq for Date<Tz> {}
 
 impl<Tz: TimeZone> PartialOrd for Date<Tz> {
  fn partial_cmp(&self, other: &Date<Tz>) -> Option<Ordering> {
- self.date.partial_cmp(&other.date)
+ Some(self.cmp(other))
  }
 }
 

src/datetime/mod.rs

@@ -51,6 +51,8 @@ mod tests;
 /// [`TimeZone`](./offset/trait.TimeZone.html) implementations.
 #[derive(Clone)]
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
+#[cfg_attr(feature = "rkyv", archive(compare(PartialEq, PartialOrd)))]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct DateTime<Tz: TimeZone> {
  datetime: NaiveDateTime,
  offset: Tz::Offset,
@@ -1504,6 +1506,31 @@ impl<Tz: TimeZone> fmt::Debug for DateTime<Tz> {
  }
 }
 
+// `fmt::Debug` is hand implemented for the `rkyv::Archive` variant of `DateTime` because
+// deriving a trait recursively does not propagate trait defined associated types with their own
+// constraints:
+// In our case `<<Tz as offset::TimeZone>::Offset as Archive>::Archived`
+// cannot be formatted using `{:?}` because it doesn't implement `Debug`.
+// See below for further discussion:
+// * https://github.com/rust-lang/rust/issues/26925
+// * https://github.com/rkyv/rkyv/issues/333
+// * https://github.com/dtolnay/syn/issues/370
+#[cfg(feature = "rkyv-validation")]
+impl<Tz: TimeZone> fmt::Debug for ArchivedDateTime<Tz>
+where
+ Tz: Archive,
+ <Tz as Archive>::Archived: fmt::Debug,
+ <<Tz as TimeZone>::Offset as Archive>::Archived: fmt::Debug,
+ <Tz as TimeZone>::Offset: fmt::Debug + Archive,
+{
+ fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+ f.debug_struct("ArchivedDateTime")
+ .field("datetime", &self.datetime)
+ .field("offset", &self.offset)
+ .finish()
+ }
+}
+
 impl<Tz: TimeZone> fmt::Display for DateTime<Tz>
 where
  Tz::Offset: fmt::Display,

src/duration.rs

@@ -54,8 +54,10 @@ macro_rules! try_opt {
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq, PartialOrd)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct Duration {
  secs: i64,
  nanos: i32, // Always 0 <= nanos < NANOS_PER_SEC
@@ -795,4 +797,12 @@ mod tests {
  Err(OutOfRangeError(()))
  );
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let duration = Duration::seconds(1);
+ let bytes = rkyv::to_bytes::<_, 16>(&duration).unwrap();
+ assert_eq!(rkyv::from_bytes::<Duration>(&bytes).unwrap(), duration);
+ }
 }

src/format/scan.rs

@@ -274,7 +274,7 @@ where
  };
  s = match s.len() {
  len if len >= 2 => &s[2..],
- len if len == 0 => s,
+ 0 => s,
  _ => return Err(TOO_SHORT),
  };
 

src/month.rs

@@ -33,8 +33,10 @@ use crate::OutOfRange;
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 #[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub enum Month {
  /// January
@@ -417,4 +419,12 @@ mod tests {
  from_str::<Month>(string).unwrap_err();
  }
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let month = Month::January;
+ let bytes = rkyv::to_bytes::<_, 1>(&month).unwrap();
+ assert_eq!(rkyv::from_bytes::<Month>(&bytes).unwrap(), month);
+ }
 }

src/naive/date.rs

@@ -192,8 +192,10 @@ impl Days {
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq, PartialOrd)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct NaiveDate {
  ymdf: DateImpl, // (year << 13) | of
 }
@@ -3324,6 +3326,18 @@ mod tests {
  }
  }
 
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let date_min = NaiveDate::MIN;
+ let bytes = rkyv::to_bytes::<_, 4>(&date_min).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveDate>(&bytes).unwrap(), date_min);
+
+ let date_max = NaiveDate::MAX;
+ let bytes = rkyv::to_bytes::<_, 4>(&date_max).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveDate>(&bytes).unwrap(), date_max);
+ }
+
  // MAX_YEAR-12-31 minus 0000-01-01
  // = (MAX_YEAR-12-31 minus 0000-12-31) + (0000-12-31 - 0000-01-01)
  // = MAX_YEAR * 365 + (# of leap years from 0001 to MAX_YEAR) + 365

src/naive/datetime/mod.rs

@@ -77,8 +77,10 @@ pub const MAX_DATETIME: NaiveDateTime = NaiveDateTime::MAX;
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq, PartialOrd)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 #[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct NaiveDateTime {
  date: NaiveDate,

src/naive/datetime/tests.rs

@@ -536,3 +536,15 @@ fn test_and_timezone_min_max_dates() {
  }
  }
 }
+
+#[test]
+#[cfg(feature = "rkyv-validation")]
+fn test_rkyv_validation() {
+ let dt_min = NaiveDateTime::MIN;
+ let bytes = rkyv::to_bytes::<_, 12>(&dt_min).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveDateTime>(&bytes).unwrap(), dt_min);
+
+ let dt_max = NaiveDateTime::MAX;
+ let bytes = rkyv::to_bytes::<_, 12>(&dt_max).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveDateTime>(&bytes).unwrap(), dt_max);
+}

src/naive/isoweek.rs

@@ -20,8 +20,10 @@ use rkyv::{Archive, Deserialize, Serialize};
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct IsoWeek {
  // note that this allows for larger year range than `NaiveDate`.
  // this is crucial because we have an edge case for the first and last week supported,
@@ -151,6 +153,8 @@ impl fmt::Debug for IsoWeek {
 
 #[cfg(test)]
 mod tests {
+ #[cfg(feature = "rkyv-validation")]
+ use super::IsoWeek;
  use crate::naive::{internals, NaiveDate};
  use crate::Datelike;
 
@@ -205,4 +209,16 @@ mod tests {
  assert!(monday.iso_week() >= friday.iso_week());
  assert!(monday.iso_week() <= friday.iso_week());
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let minweek = NaiveDate::MIN.iso_week();
+ let bytes = rkyv::to_bytes::<_, 4>(&minweek).unwrap();
+ assert_eq!(rkyv::from_bytes::<IsoWeek>(&bytes).unwrap(), minweek);
+
+ let maxweek = NaiveDate::MAX.iso_week();
+ let bytes = rkyv::to_bytes::<_, 4>(&maxweek).unwrap();
+ assert_eq!(rkyv::from_bytes::<IsoWeek>(&bytes).unwrap(), maxweek);
+ }
 }

src/naive/time/mod.rs

@@ -206,8 +206,10 @@ mod tests;
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
 #[cfg_attr(
  feature = "rkyv",
+ archive(compare(PartialEq, PartialOrd)),
  archive_attr(derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Debug, Hash))
 )]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct NaiveTime {
  secs: u32,
  frac: u32,

src/naive/time/tests.rs

@@ -376,3 +376,15 @@ fn test_overflowing_offset() {
  assert_eq!(t.overflowing_add_offset(positive_offset).0, t + positive_offset);
  assert_eq!(t.overflowing_sub_offset(positive_offset).0, t - positive_offset);
 }
+
+#[test]
+#[cfg(feature = "rkyv-validation")]
+fn test_rkyv_validation() {
+ let t_min = NaiveTime::MIN;
+ let bytes = rkyv::to_bytes::<_, 8>(&t_min).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveTime>(&bytes).unwrap(), t_min);
+
+ let t_max = NaiveTime::MAX;
+ let bytes = rkyv::to_bytes::<_, 8>(&t_max).unwrap();
+ assert_eq!(rkyv::from_bytes::<NaiveTime>(&bytes).unwrap(), t_max);
+}

src/offset/fixed.rs

@@ -21,7 +21,12 @@ use crate::naive::{NaiveDate, NaiveDateTime};
 /// [`west_opt`](#method.west_opt) methods for examples.
 #[derive(PartialEq, Eq, Hash, Copy, Clone)]
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
-#[cfg_attr(feature = "rkyv", archive_attr(derive(Clone, Copy, PartialEq, Eq, Hash, Debug)))]
+#[cfg_attr(
+ feature = "rkyv",
+ archive(compare(PartialEq)),
+ archive_attr(derive(Clone, Copy, PartialEq, Eq, Hash, Debug))
+)]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 pub struct FixedOffset {
  local_minus_utc: i32,
 }
@@ -222,4 +227,12 @@ mod tests {
  let offset = FixedOffset::from_str("+06:30").unwrap();
  assert_eq!(offset.local_minus_utc, (6 * 3600) + 1800);
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let offset = FixedOffset::from_str("-0500").unwrap();
+ let bytes = rkyv::to_bytes::<_, 4>(&offset).unwrap();
+ assert_eq!(rkyv::from_bytes::<FixedOffset>(&bytes).unwrap(), offset);
+ }
 }

src/offset/local/mod.rs

@@ -105,7 +105,8 @@ mod tz_info;
 /// ```
 #[derive(Copy, Clone, Debug)]
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
-#[cfg_attr(feature = "rkyv", archive_attr(derive(Clone, Copy, Debug)))]
+#[cfg_attr(feature = "rkyv", archive(compare(PartialEq)), archive_attr(derive(Clone, Copy, Debug)))]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 #[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct Local;
 
@@ -258,4 +259,17 @@ mod tests {
  );
  }
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let local = Local;
+ // Local is a ZST and serializes to 0 bytes
+ let bytes = rkyv::to_bytes::<_, 0>(&local).unwrap();
+ assert_eq!(bytes.len(), 0);
+
+ // but is deserialized to an archived variant without a
+ // wrapping object
+ assert_eq!(rkyv::from_bytes::<Local>(&bytes).unwrap(), super::ArchivedLocal);
+ }
 }

src/offset/utc.rs

@@ -42,7 +42,12 @@ use crate::{Date, DateTime};
 /// ```
 #[derive(Copy, Clone, PartialEq, Eq, Hash)]
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
-#[cfg_attr(feature = "rkyv", archive_attr(derive(Clone, Copy, PartialEq, Eq, Debug, Hash)))]
+#[cfg_attr(
+ feature = "rkyv",
+ archive(compare(PartialEq)),
+ archive_attr(derive(Clone, Copy, PartialEq, Eq, Debug, Hash))
+)]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 #[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct Utc;
 

src/weekday.rs

@@ -32,7 +32,12 @@ use crate::OutOfRange;
 #[derive(PartialEq, Eq, Copy, Clone, Debug, Hash)]
 #[cfg_attr(feature = "rustc-serialize", derive(RustcEncodable, RustcDecodable))]
 #[cfg_attr(feature = "rkyv", derive(Archive, Deserialize, Serialize))]
-#[cfg_attr(feature = "rkyv", archive_attr(derive(Clone, Copy, PartialEq, Eq, Debug, Hash)))]
+#[cfg_attr(
+ feature = "rkyv",
+ archive(compare(PartialEq)),
+ archive_attr(derive(Clone, Copy, PartialEq, Eq, Debug, Hash))
+)]
+#[cfg_attr(feature = "rkyv-validation", archive(check_bytes))]
 #[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub enum Weekday {
  /// Monday.
@@ -381,4 +386,13 @@ mod tests {
  from_str::<Weekday>(str).unwrap_err();
  }
  }
+
+ #[test]
+ #[cfg(feature = "rkyv-validation")]
+ fn test_rkyv_validation() {
+ let mon = Weekday::Mon;
+ let bytes = rkyv::to_bytes::<_, 1>(&mon).unwrap();
+
+ assert_eq!(rkyv::from_bytes::<Weekday>(&bytes).unwrap(), mon);
+ }
 }