rust/hyperswitch
1
0
Fork 0
mirror of https://github.com/juspay/hyperswitch.git synced 2025-10-30 01:27:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(payment_attempt): add kv for find last successful attempt (#2206)

Browse Source
This commit is contained in:
Kartikeya Hegde
2023-09-25 16:30:12 +05:30
committed by GitHub
parent 5d66156132
commit d3157f0bd6
5 changed files with 56 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
crates/storage_impl/src/payments/payment_attempt.rs
View File

@ -26,6 +26,7 @@ use redis_interface::HsetnxReply;
use router_env::{instrument, tracing};
use crate::{
diesel_error_to_data_error,
lookup::ReverseLookupInterface,
redis::kv_store::{PartitionKey, RedisConnInterface},
utils::{
@ -49,7 +50,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
.insert(&conn)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -67,7 +68,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
.update_with_attempt_id(&conn, payment_attempt.to_storage_model())
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -89,7 +90,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -109,7 +110,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -129,7 +130,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -153,7 +154,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -174,7 +175,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
DieselPaymentAttempt::get_filters_for_payments(&conn, intents.as_slice(), merchant_id)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(
@ -202,7 +203,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -218,7 +219,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
DieselPaymentAttempt::find_by_merchant_id_payment_id(&conn, merchant_id, payment_id)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(|a| {
@ -239,7 +240,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
DieselPaymentAttempt::find_by_merchant_id_attempt_id(&conn, merchant_id, attempt_id)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
.map(PaymentAttempt::from_storage_model)
@ -275,7 +276,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for RouterStore<T> {
)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
}
@ -380,7 +381,7 @@ impl<T: DatabaseStore> PaymentAttemptInterface for KVRouterStore<T> {
.insert(&conn)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})?;
@ -570,13 +571,41 @@ impl<T: DatabaseStore> PaymentAttemptInterface for KVRouterStore<T> {
merchant_id: &str,
storage_scheme: MerchantStorageScheme,
) -> error_stack::Result<PaymentAttempt, errors::StorageError> {
self.router_store
.find_payment_attempt_last_successful_attempt_by_payment_id_merchant_id(
payment_id,
merchant_id,
storage_scheme,
)
.await
let database_call = || {
self.router_store
.find_payment_attempt_last_successful_attempt_by_payment_id_merchant_id(
payment_id,
merchant_id,
storage_scheme,
)
};
match storage_scheme {
MerchantStorageScheme::PostgresOnly => database_call().await,
MerchantStorageScheme::RedisKv => {
let key = format!("{merchant_id}_{payment_id}");
let pattern = "pa_*";
let redis_conn = self
.get_redis_conn()
.change_context(errors::StorageError::KVError)?;
let redis_fut = async {
redis_conn
.hscan_and_deserialize::<PaymentAttempt>(&key, pattern, None)
.await
.and_then(|mut payment_attempts| {
payment_attempts.sort_by(|a, b| b.modified_at.cmp(&a.modified_at));
payment_attempts
.iter()
.find(|&pa| pa.status == api_models::enums::AttemptStatus::Charged)
.cloned()
.ok_or(error_stack::report!(
redis_interface::errors::RedisError::NotFound
))
})
};
try_redis_get_else_try_database_get(redis_fut, database_call).await
}
}
}
async fn find_payment_attempt_by_merchant_id_connector_txn_id(
@ -1463,7 +1492,7 @@ async fn add_connector_txn_id_to_reverse_lookup<T: DatabaseStore>(
.insert(&conn)
.await
.map_err(|err| {
let new_err = crate::diesel_error_to_data_error(err.current_context());
let new_err = diesel_error_to_data_error(err.current_context());
err.change_context(new_err)
})
}
@ -1487,7 +1516,7 @@ async fn add_preprocessing_id_to_reverse_lookup<T: DatabaseStore>(
.insert(&conn)
.await
.map_err(|er| {
let new_err = crate::diesel_error_to_data_error(er.current_context());
let new_err = diesel_error_to_data_error(er.current_context());
er.change_context(new_err)
})
}