rust/hyperswitch
1
0
Fork 0
mirror of https://github.com/juspay/hyperswitch.git synced 2025-11-03 21:37:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(router): add total count for payments list (#1912)

Browse Source
This commit is contained in:
Apoorv Dixit
2023-08-29 16:48:35 +05:30
committed by GitHub
parent 784702d9c5
commit 7a5c8413cf
10 changed files with 292 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
crates/router/src/db/payment_attempt.rs
View File

@ -1,3 +1,5 @@
use api_models::enums::{Connector, PaymentMethod};
use super::MockDb;
use crate::{
core::errors::{self, CustomResult},
@ -76,10 +78,20 @@ pub trait PaymentAttemptInterface {
merchant_id: &str,
storage_scheme: enums::MerchantStorageScheme,
) -> CustomResult<diesel_models::payment_attempt::PaymentListFilters, errors::StorageError>;
async fn get_total_count_of_filtered_payment_attempts(
&self,
merchant_id: &str,
active_attempt_ids: &[String],
connector: Option<Vec<Connector>>,
payment_methods: Option<Vec<PaymentMethod>>,
storage_scheme: enums::MerchantStorageScheme,
) -> CustomResult<i64, errors::StorageError>;
}
#[cfg(not(feature = "kv_store"))]
mod storage {
use api_models::enums::{Connector, PaymentMethod};
use error_stack::IntoReport;
use storage_impl::DataModelExt;
@ -211,6 +223,37 @@ mod storage {
.into_report()
}
async fn get_total_count_of_filtered_payment_attempts(
&self,
merchant_id: &str,
active_attempt_ids: &[String],
connector: Option<Vec<Connector>>,
payment_methods: Option<Vec<PaymentMethod>>,
_storage_scheme: enums::MerchantStorageScheme,
) -> CustomResult<i64, errors::StorageError> {
let conn = connection::pg_connection_read(self).await?;
let connector_strings = if let Some(connector_vec) = &connector {
Some(
connector_vec
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>(),
)
} else {
None
};
PaymentAttempt::get_total_count_of_attempts(
&conn,
merchant_id,
active_attempt_ids,
connector_strings,
payment_methods,
)
.await
.map_err(Into::into)
.into_report()
}
async fn find_payment_attempt_by_preprocessing_id_merchant_id(
&self,
preprocessing_id: &str,
@ -281,6 +324,17 @@ impl PaymentAttemptInterface for MockDb {
Err(errors::StorageError::MockDbError)?
}
async fn get_total_count_of_filtered_payment_attempts(
&self,
_merchant_id: &str,
_active_attempt_ids: &[String],
_connector: Option<Vec<Connector>>,
_payment_methods: Option<Vec<PaymentMethod>>,
_storage_scheme: enums::MerchantStorageScheme,
) -> CustomResult<i64, errors::StorageError> {
Err(errors::StorageError::MockDbError)?
}
async fn find_payment_attempt_by_attempt_id_merchant_id(
&self,
_attempt_id: &str,
@ -920,6 +974,35 @@ mod storage {
.map_err(Into::into)
.into_report()
}
async fn get_total_count_of_filtered_payment_attempts(
&self,
merchant_id: &str,
active_attempt_ids: &[String],
connector: Option<Vec<api_models::enums::Connector>>,
payment_methods: Option<Vec<api_models::enums::PaymentMethod>>,
_storage_scheme: enums::MerchantStorageScheme,
) -> CustomResult<i64, errors::StorageError> {
let conn = connection::pg_connection_read(self).await?;
let connector_strings = connector.as_ref().map(|connector_vec| {
connector_vec
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>()
});
PaymentAttempt::get_total_count_of_attempts(
&conn,
merchant_id,
active_attempt_ids,
connector_strings,
payment_methods,
)
.await
.map_err(Into::into)
.into_report()
}
}
#[inline]

11
crates/router/src/db/payment_intent.rs
View File

@ -48,6 +48,17 @@ impl PaymentIntentInterface for MockDb {
Err(errors::DataStorageError::MockDbError)?
}
#[cfg(feature = "olap")]
async fn get_filtered_active_attempt_ids_for_total_count(
&self,
_merchant_id: &str,
_constraints: &PaymentIntentFetchConstraints,
_storage_scheme: enums::MerchantStorageScheme,
) -> error_stack::Result<Vec<String>, errors::DataStorageError> {
// [#172]: Implement function for `MockDb`
Err(errors::DataStorageError::MockDbError)?
}
#[allow(clippy::panic)]
async fn insert_payment_intent(
&self,