feat: add functions for extracting timestamps

Some ID generation functions have timestamps embedded in them, and
being able to extract the timestamp is a useful feature. Timestamps
extracted from IDs can be used in logic checks, as
virtual/generated columns and have many other use cases.

This commit adds functions for extracting timestamps to the IDs that
support them (i.e. IDs that have a timestamp embedded somehow).

Signed-off-by: vados <[email protected]>

Cargo.toml

@@ -20,6 +20,8 @@ pg15 = [ "pgrx/pg15", "pgrx-tests/pg15" ]
 pg_test = []
 
 [dependencies]
+base36 = "0.0.1"
+chrono = { version = "0.4.31", default-features = false, features = [ "clock" ] }
 cuid = "1.3.1"
 cuid2 = "0.1.0"
 getrandom = "0.2.8"
@@ -34,6 +36,7 @@ uuid7 = "0.6.4"
 xid = "1.0.3"
 
 [dev-dependencies]
+miette = { version = "5.10.0", features = ["fancy"] }
 pgrx-tests = "=0.9.7"
 
 [profile.dev]

src/common.rs

@@ -0,0 +1,28 @@
+/// A trait that encapsulates things that can be converted to some type
+/// or to an error if conversion fails (ex. Result, Option, etc)
+pub(crate) trait OrPgxError<T> {
+    /// Convert the given type to a T, possibly failing
+    /// and calling [`pgrx::error`], with a given prefix if an error is returned
+    fn or_pgrx_error(self, prefix: impl AsRef<str>) -> T;
+}
+
+impl<T, E> OrPgxError<T> for Result<T, E>
+where
+    E: std::error::Error,
+{
+    fn or_pgrx_error(self, prefix: impl AsRef<str>) -> T {
+        match self {
+            Ok(v) => v,
+            Err(e) => pgrx::error!("{}: {e}", prefix.as_ref()),
+        }
+    }
+}
+
+impl<T> OrPgxError<T> for Option<T> {
+    fn or_pgrx_error(self, prefix: impl AsRef<str>) -> T {
+        match self {
+            Some(v) => v,
+            None => pgrx::error!("{}", prefix.as_ref()),
+        }
+    }
+}