Add Sub implementation that takes rhs by reference

chronotope · Jun 2, 2023 · ec731c2 · ec731c2
1 parent b271084
commit ec731c2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/datetime/mod.rs b/src/datetime/mod.rs
@@ -1045,6 +1045,15 @@ impl<Tz: TimeZone> Sub<DateTime<Tz>> for DateTime<Tz> {
     }
 }
 
+impl<Tz: TimeZone> Sub<&DateTime<Tz>> for DateTime<Tz> {
+    type Output = OldDuration;
+
+    #[inline]
+    fn sub(self, rhs: &DateTime<Tz>) -> OldDuration {
+        self.signed_duration_since(rhs)
+    }
+}
+
 impl<Tz: TimeZone> Add<Days> for DateTime<Tz> {
     type Output = DateTime<Tz>;
 

diff --git a/src/datetime/tests.rs b/src/datetime/tests.rs
@@ -320,6 +320,10 @@ fn signed_duration_since_autoref() {
     let diff1 = dt1.signed_duration_since(dt2); // Copy/consume
     let diff2 = dt2.signed_duration_since(&dt1); // Take by reference
     assert_eq!(diff1, -diff2);
+
+    let diff1 = dt1 - &dt2; // We can choose to substract rhs by reference
+    let diff2 = dt2 - dt1; // Or consume rhs
+    assert_eq!(diff1, -diff2);
 }
 
 #[test]