refactor(storage): update crate name to diesel models (#1685)

crates/diesel_models/src/lib.rs

@@ -0,0 +1,95 @@
+pub mod address;
+pub mod api_keys;
+pub mod cards_info;
+pub mod configs;
+pub mod connector_response;
+pub mod customers;
+pub mod dispute;
+pub mod encryption;
+pub mod enums;
+pub mod ephemeral_key;
+pub mod errors;
+pub mod events;
+pub mod file;
+#[cfg(feature = "kv_store")]
+pub mod kv;
+pub mod locker_mock_up;
+pub mod mandate;
+pub mod merchant_account;
+pub mod merchant_connector_account;
+pub mod merchant_key_store;
+pub mod payment_attempt;
+pub mod payment_intent;
+pub mod payment_method;
+pub mod process_tracker;
+pub mod query;
+pub mod refund;
+pub mod reverse_lookup;
+#[allow(unused_qualifications)]
+pub mod schema;
+
+use diesel_impl::{DieselArray, OptionalDieselArray};
+
+pub type StorageResult<T> = error_stack::Result<T, errors::DatabaseError>;
+pub type PgPooledConn = async_bb8_diesel::Connection<diesel::PgConnection>;
+
+/// The types and implementations provided by this module are required for the schema generated by
+/// `diesel_cli` 2.0 to work with the types defined in Rust code. This is because
+/// [`diesel`][diesel] 2.0 [changed the nullability of array elements][diesel-2.0-array-nullability],
+/// which causes [`diesel`][diesel] to deserialize arrays as `Vec<Option<T>>`. To prevent declaring
+/// array elements as `Option<T>`, this module provides types and implementations to deserialize
+/// arrays as `Vec<T>`, considering only non-null values (`Some(T)`) among the deserialized
+/// `Option<T>` values.
+///
+/// [diesel-2.0-array-nullability]: https://diesel.rs/guides/migration_guide.html#2-0-0-nullability-of-array-elements
+
+#[doc(hidden)]
+pub(crate) mod diesel_impl {
+    use diesel::{
+        deserialize::FromSql,
+        pg::Pg,
+        sql_types::{Array, Nullable},
+        Queryable,
+    };
+
+    pub struct DieselArray<T>(Vec<Option<T>>);
+
+    impl<T> From<DieselArray<T>> for Vec<T> {
+        fn from(array: DieselArray<T>) -> Self {
+            array.0.into_iter().flatten().collect()
+        }
+    }
+
+    impl<T, U> Queryable<Array<Nullable<U>>, Pg> for DieselArray<T>
+    where
+        T: FromSql<U, Pg>,
+        Vec<Option<T>>: FromSql<Array<Nullable<U>>, Pg>,
+    {
+        type Row = Vec<Option<T>>;
+
+        fn build(row: Self::Row) -> diesel::deserialize::Result<Self> {
+            Ok(Self(row))
+        }
+    }
+
+    pub struct OptionalDieselArray<T>(Option<Vec<Option<T>>>);
+
+    impl<T> From<OptionalDieselArray<T>> for Option<Vec<T>> {
+        fn from(option_array: OptionalDieselArray<T>) -> Self {
+            option_array
+                .0
+                .map(|array| array.into_iter().flatten().collect())
+        }
+    }
+
+    impl<T, U> Queryable<Nullable<Array<Nullable<U>>>, Pg> for OptionalDieselArray<T>
+    where
+        Option<Vec<Option<T>>>: FromSql<Nullable<Array<Nullable<U>>>, Pg>,
+    {
+        type Row = Option<Vec<Option<T>>>;
+
+        fn build(row: Self::Row) -> diesel::deserialize::Result<Self> {
+            Ok(Self(row))
+        }
+    }
+}