mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 01:40:21 +08:00
refactor: 拆分 gateway 单体为独立 crate，新增 systemd 部署方案
将 gateway 内部的 model-fetch、provider-transport、scheduler-core、 usage-runtime、video-tasks-core 模块提取为独立 crate；重构 gateway 内部模块结构（state/router/cache/data/query 等）；移除大量遗留模块 文件；新增 systemd 二进制部署骨架及相关文档；更新前端 usage 相关 API 和组件。
This commit is contained in:
fawney19
@@ -4,7 +4,17 @@ use std::sync::RwLock;
|
||||
use async_trait::async_trait;
|
||||
|
||||
use super::types::{
|
||||
StoredUsageSettlement, StoredWalletSnapshot, UsageSettlementInput, WalletLookupKey,
|
||||
AdjustWalletBalanceInput, AdminPaymentOrderListQuery, AdminWalletLedgerQuery,
|
||||
AdminWalletListQuery, AdminWalletRefundRequestListQuery, CompleteAdminWalletRefundInput,
|
||||
CreateManualWalletRechargeInput, CreateWalletRechargeOrderInput,
|
||||
CreateWalletRechargeOrderOutcome, CreateWalletRefundRequestInput,
|
||||
CreateWalletRefundRequestOutcome, CreditAdminPaymentOrderInput, FailAdminWalletRefundInput,
|
||||
ProcessAdminWalletRefundInput, ProcessPaymentCallbackInput, ProcessPaymentCallbackOutcome,
|
||||
StoredAdminPaymentCallbackPage, StoredAdminPaymentOrder, StoredAdminPaymentOrderPage,
|
||||
StoredAdminWalletLedgerPage, StoredAdminWalletListItem, StoredAdminWalletListPage,
|
||||
StoredAdminWalletRefundPage, StoredAdminWalletRefundRequestPage,
|
||||
StoredAdminWalletTransactionPage, StoredWalletDailyUsageLedger,
|
||||
StoredWalletDailyUsageLedgerPage, StoredWalletSnapshot, WalletLookupKey, WalletMutationOutcome,
|
||||
WalletReadRepository, WalletWriteRepository,
|
||||
};
|
||||
use crate::DataLayerError;
|
||||
@@ -12,7 +22,6 @@ use crate::DataLayerError;
|
||||
#[derive(Debug, Default)]
|
||||
pub struct InMemoryWalletRepository {
|
||||
wallets_by_id: RwLock<BTreeMap<String, StoredWalletSnapshot>>,
|
||||
provider_monthly_used: RwLock<BTreeMap<String, f64>>,
|
||||
}
|
||||
|
||||
impl InMemoryWalletRepository {
|
||||
@@ -26,9 +35,16 @@ impl InMemoryWalletRepository {
|
||||
}
|
||||
Self {
|
||||
wallets_by_id: RwLock::new(wallets_by_id),
|
||||
provider_monthly_used: RwLock::new(BTreeMap::new()),
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn with_wallets_mut<R>(
|
||||
&self,
|
||||
f: impl FnOnce(&mut BTreeMap<String, StoredWalletSnapshot>) -> R,
|
||||
) -> R {
|
||||
let mut wallets = self.wallets_by_id.write().expect("wallet repo lock");
|
||||
f(&mut wallets)
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
@@ -96,112 +112,329 @@ impl WalletReadRepository for InMemoryWalletRepository {
|
||||
.cloned()
|
||||
.collect())
|
||||
}
|
||||
|
||||
async fn list_admin_wallets(
|
||||
&self,
|
||||
query: &AdminWalletListQuery,
|
||||
) -> Result<StoredAdminWalletListPage, DataLayerError> {
|
||||
let wallets = self.wallets_by_id.read().expect("wallet repo lock");
|
||||
let mut items = wallets
|
||||
.values()
|
||||
.filter(|wallet| {
|
||||
query
|
||||
.status
|
||||
.as_deref()
|
||||
.is_none_or(|expected| wallet.status == expected)
|
||||
})
|
||||
.filter(|wallet| match query.owner_type.as_deref() {
|
||||
Some("user") => wallet.user_id.is_some(),
|
||||
Some("api_key") => wallet.api_key_id.is_some(),
|
||||
_ => true,
|
||||
})
|
||||
.map(|wallet| StoredAdminWalletListItem {
|
||||
id: wallet.id.clone(),
|
||||
user_id: wallet.user_id.clone(),
|
||||
api_key_id: wallet.api_key_id.clone(),
|
||||
balance: wallet.balance,
|
||||
gift_balance: wallet.gift_balance,
|
||||
limit_mode: wallet.limit_mode.clone(),
|
||||
currency: wallet.currency.clone(),
|
||||
status: wallet.status.clone(),
|
||||
total_recharged: wallet.total_recharged,
|
||||
total_consumed: wallet.total_consumed,
|
||||
total_refunded: wallet.total_refunded,
|
||||
total_adjusted: wallet.total_adjusted,
|
||||
user_name: None,
|
||||
api_key_name: None,
|
||||
created_at_unix_secs: None,
|
||||
updated_at_unix_secs: Some(wallet.updated_at_unix_secs),
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
items.sort_by(|left, right| {
|
||||
right
|
||||
.updated_at_unix_secs
|
||||
.cmp(&left.updated_at_unix_secs)
|
||||
.then_with(|| right.id.cmp(&left.id))
|
||||
});
|
||||
let total = items.len() as u64;
|
||||
let items = items
|
||||
.into_iter()
|
||||
.skip(query.offset)
|
||||
.take(query.limit)
|
||||
.collect::<Vec<_>>();
|
||||
Ok(StoredAdminWalletListPage { items, total })
|
||||
}
|
||||
|
||||
async fn list_admin_wallet_ledger(
|
||||
&self,
|
||||
_query: &AdminWalletLedgerQuery,
|
||||
) -> Result<StoredAdminWalletLedgerPage, DataLayerError> {
|
||||
Ok(StoredAdminWalletLedgerPage::default())
|
||||
}
|
||||
|
||||
async fn list_admin_wallet_refund_requests(
|
||||
&self,
|
||||
_query: &AdminWalletRefundRequestListQuery,
|
||||
) -> Result<StoredAdminWalletRefundRequestPage, DataLayerError> {
|
||||
Ok(StoredAdminWalletRefundRequestPage::default())
|
||||
}
|
||||
|
||||
async fn list_admin_wallet_transactions(
|
||||
&self,
|
||||
_wallet_id: &str,
|
||||
_limit: usize,
|
||||
_offset: usize,
|
||||
) -> Result<StoredAdminWalletTransactionPage, DataLayerError> {
|
||||
Ok(StoredAdminWalletTransactionPage::default())
|
||||
}
|
||||
|
||||
async fn find_wallet_today_usage(
|
||||
&self,
|
||||
_wallet_id: &str,
|
||||
_billing_timezone: &str,
|
||||
) -> Result<Option<StoredWalletDailyUsageLedger>, DataLayerError> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn list_wallet_daily_usage_history(
|
||||
&self,
|
||||
_wallet_id: &str,
|
||||
_billing_timezone: &str,
|
||||
_limit: usize,
|
||||
) -> Result<StoredWalletDailyUsageLedgerPage, DataLayerError> {
|
||||
Ok(StoredWalletDailyUsageLedgerPage::default())
|
||||
}
|
||||
|
||||
async fn list_admin_wallet_refunds(
|
||||
&self,
|
||||
_wallet_id: &str,
|
||||
_limit: usize,
|
||||
_offset: usize,
|
||||
) -> Result<StoredAdminWalletRefundPage, DataLayerError> {
|
||||
Ok(StoredAdminWalletRefundPage::default())
|
||||
}
|
||||
|
||||
async fn list_admin_payment_orders(
|
||||
&self,
|
||||
_query: &AdminPaymentOrderListQuery,
|
||||
) -> Result<StoredAdminPaymentOrderPage, DataLayerError> {
|
||||
Ok(StoredAdminPaymentOrderPage::default())
|
||||
}
|
||||
|
||||
async fn find_admin_payment_order(
|
||||
&self,
|
||||
_order_id: &str,
|
||||
) -> Result<Option<StoredAdminPaymentOrder>, DataLayerError> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn list_wallet_payment_orders_by_user_id(
|
||||
&self,
|
||||
_user_id: &str,
|
||||
_limit: usize,
|
||||
_offset: usize,
|
||||
) -> Result<StoredAdminPaymentOrderPage, DataLayerError> {
|
||||
Ok(StoredAdminPaymentOrderPage::default())
|
||||
}
|
||||
|
||||
async fn find_wallet_payment_order_by_user_id(
|
||||
&self,
|
||||
_user_id: &str,
|
||||
_order_id: &str,
|
||||
) -> Result<Option<StoredAdminPaymentOrder>, DataLayerError> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn find_wallet_refund(
|
||||
&self,
|
||||
_wallet_id: &str,
|
||||
_refund_id: &str,
|
||||
) -> Result<Option<super::types::StoredAdminWalletRefund>, DataLayerError> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn list_admin_payment_callbacks(
|
||||
&self,
|
||||
_payment_method: Option<&str>,
|
||||
_limit: usize,
|
||||
_offset: usize,
|
||||
) -> Result<StoredAdminPaymentCallbackPage, DataLayerError> {
|
||||
Ok(StoredAdminPaymentCallbackPage::default())
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl WalletWriteRepository for InMemoryWalletRepository {
|
||||
async fn settle_usage(
|
||||
async fn create_wallet_recharge_order(
|
||||
&self,
|
||||
input: UsageSettlementInput,
|
||||
) -> Result<Option<StoredUsageSettlement>, DataLayerError> {
|
||||
input.validate()?;
|
||||
if input.billing_status != "pending" {
|
||||
return Ok(Some(StoredUsageSettlement {
|
||||
request_id: input.request_id,
|
||||
wallet_id: None,
|
||||
billing_status: input.billing_status,
|
||||
wallet_balance_before: None,
|
||||
wallet_balance_after: None,
|
||||
wallet_recharge_balance_before: None,
|
||||
wallet_recharge_balance_after: None,
|
||||
wallet_gift_balance_before: None,
|
||||
wallet_gift_balance_after: None,
|
||||
provider_monthly_used_usd: None,
|
||||
finalized_at_unix_secs: input.finalized_at_unix_secs,
|
||||
}));
|
||||
}
|
||||
|
||||
input: CreateWalletRechargeOrderInput,
|
||||
) -> Result<CreateWalletRechargeOrderOutcome, DataLayerError> {
|
||||
let mut wallets = self.wallets_by_id.write().expect("wallet repo lock");
|
||||
let wallet_id = input
|
||||
.api_key_id
|
||||
.as_deref()
|
||||
.and_then(|api_key_id| {
|
||||
wallets
|
||||
.values()
|
||||
.find(|wallet| wallet.api_key_id.as_deref() == Some(api_key_id))
|
||||
.map(|wallet| wallet.id.clone())
|
||||
})
|
||||
.or_else(|| {
|
||||
input.user_id.as_deref().and_then(|user_id| {
|
||||
wallets
|
||||
.values()
|
||||
.find(|wallet| wallet.user_id.as_deref() == Some(user_id))
|
||||
.map(|wallet| wallet.id.clone())
|
||||
})
|
||||
});
|
||||
let wallet = wallet_id
|
||||
.as_deref()
|
||||
.and_then(|wallet_id| wallets.get_mut(wallet_id));
|
||||
|
||||
let final_billing_status = if input.status == "completed" {
|
||||
"settled"
|
||||
} else {
|
||||
"void"
|
||||
};
|
||||
|
||||
let mut settlement = StoredUsageSettlement {
|
||||
request_id: input.request_id,
|
||||
wallet_id: None,
|
||||
billing_status: final_billing_status.to_string(),
|
||||
wallet_balance_before: None,
|
||||
wallet_balance_after: None,
|
||||
wallet_recharge_balance_before: None,
|
||||
wallet_recharge_balance_after: None,
|
||||
wallet_gift_balance_before: None,
|
||||
wallet_gift_balance_after: None,
|
||||
provider_monthly_used_usd: None,
|
||||
finalized_at_unix_secs: input.finalized_at_unix_secs,
|
||||
};
|
||||
|
||||
if let Some(wallet) = wallet {
|
||||
let before_recharge = wallet.balance;
|
||||
let before_gift = wallet.gift_balance;
|
||||
let before_total = before_recharge + before_gift;
|
||||
settlement.wallet_id = Some(wallet.id.clone());
|
||||
settlement.wallet_balance_before = Some(before_total);
|
||||
settlement.wallet_recharge_balance_before = Some(before_recharge);
|
||||
settlement.wallet_gift_balance_before = Some(before_gift);
|
||||
|
||||
if final_billing_status == "settled" {
|
||||
if wallet.limit_mode.eq_ignore_ascii_case("unlimited") {
|
||||
wallet.total_consumed += input.total_cost_usd;
|
||||
} else {
|
||||
let gift_deduction = before_gift.max(0.0).min(input.total_cost_usd);
|
||||
let recharge_deduction = input.total_cost_usd - gift_deduction;
|
||||
wallet.gift_balance = before_gift - gift_deduction;
|
||||
wallet.balance = before_recharge - recharge_deduction;
|
||||
wallet.total_consumed += input.total_cost_usd;
|
||||
}
|
||||
}
|
||||
|
||||
settlement.wallet_recharge_balance_after = Some(wallet.balance);
|
||||
settlement.wallet_gift_balance_after = Some(wallet.gift_balance);
|
||||
settlement.wallet_balance_after = Some(wallet.balance + wallet.gift_balance);
|
||||
let wallet = wallets
|
||||
.values_mut()
|
||||
.find(|wallet| wallet.user_id.as_deref() == Some(input.user_id.as_str()));
|
||||
if wallet
|
||||
.as_ref()
|
||||
.is_some_and(|wallet| wallet.status != "active")
|
||||
{
|
||||
return Ok(CreateWalletRechargeOrderOutcome::WalletInactive);
|
||||
}
|
||||
let wallet_id = wallet
|
||||
.map(|wallet| wallet.id.clone())
|
||||
.or(input.preferred_wallet_id)
|
||||
.unwrap_or_else(|| "wallet-memory".to_string());
|
||||
Ok(CreateWalletRechargeOrderOutcome::Created(
|
||||
StoredAdminPaymentOrder {
|
||||
id: "payment-order-memory".to_string(),
|
||||
order_no: input.order_no,
|
||||
wallet_id,
|
||||
user_id: Some(input.user_id),
|
||||
amount_usd: input.amount_usd,
|
||||
pay_amount: input.pay_amount,
|
||||
pay_currency: input.pay_currency,
|
||||
exchange_rate: input.exchange_rate,
|
||||
refunded_amount_usd: 0.0,
|
||||
refundable_amount_usd: 0.0,
|
||||
payment_method: input.payment_method,
|
||||
gateway_order_id: Some(input.gateway_order_id),
|
||||
gateway_response: Some(input.gateway_response),
|
||||
status: "pending".to_string(),
|
||||
created_at_unix_secs: 0,
|
||||
paid_at_unix_secs: None,
|
||||
credited_at_unix_secs: None,
|
||||
expires_at_unix_secs: Some(input.expires_at_unix_secs),
|
||||
},
|
||||
))
|
||||
}
|
||||
|
||||
if final_billing_status == "settled" {
|
||||
if let Some(provider_id) = input.provider_id {
|
||||
let mut quotas = self
|
||||
.provider_monthly_used
|
||||
.write()
|
||||
.expect("provider quota lock");
|
||||
let value = quotas.entry(provider_id).or_insert(0.0);
|
||||
*value += input.actual_total_cost_usd;
|
||||
settlement.provider_monthly_used_usd = Some(*value);
|
||||
}
|
||||
async fn create_wallet_refund_request(
|
||||
&self,
|
||||
input: CreateWalletRefundRequestInput,
|
||||
) -> Result<CreateWalletRefundRequestOutcome, DataLayerError> {
|
||||
let wallets = self.wallets_by_id.read().expect("wallet repo lock");
|
||||
let Some(wallet) = wallets.get(&input.wallet_id) else {
|
||||
return Ok(CreateWalletRefundRequestOutcome::WalletMissing);
|
||||
};
|
||||
if input.amount_usd > wallet.balance {
|
||||
return Ok(CreateWalletRefundRequestOutcome::RefundAmountExceedsAvailableBalance);
|
||||
}
|
||||
Ok(CreateWalletRefundRequestOutcome::Created(
|
||||
super::types::StoredAdminWalletRefund {
|
||||
id: "refund-memory".to_string(),
|
||||
refund_no: input.refund_no,
|
||||
wallet_id: input.wallet_id,
|
||||
user_id: Some(input.user_id),
|
||||
payment_order_id: input.payment_order_id,
|
||||
source_type: input
|
||||
.source_type
|
||||
.unwrap_or_else(|| "wallet_balance".to_string()),
|
||||
source_id: input.source_id,
|
||||
refund_mode: input
|
||||
.refund_mode
|
||||
.unwrap_or_else(|| "offline_payout".to_string()),
|
||||
amount_usd: input.amount_usd,
|
||||
status: "pending_approval".to_string(),
|
||||
reason: input.reason,
|
||||
failure_reason: None,
|
||||
gateway_refund_id: None,
|
||||
payout_method: None,
|
||||
payout_reference: None,
|
||||
payout_proof: None,
|
||||
requested_by: None,
|
||||
approved_by: None,
|
||||
processed_by: None,
|
||||
created_at_unix_secs: 0,
|
||||
updated_at_unix_secs: 0,
|
||||
processed_at_unix_secs: None,
|
||||
completed_at_unix_secs: None,
|
||||
},
|
||||
))
|
||||
}
|
||||
|
||||
Ok(Some(settlement))
|
||||
async fn process_payment_callback(
|
||||
&self,
|
||||
_input: ProcessPaymentCallbackInput,
|
||||
) -> Result<ProcessPaymentCallbackOutcome, DataLayerError> {
|
||||
Ok(ProcessPaymentCallbackOutcome::Failed {
|
||||
duplicate: false,
|
||||
error: "payment callback is not supported in memory wallet repository".to_string(),
|
||||
})
|
||||
}
|
||||
|
||||
async fn adjust_wallet_balance(
|
||||
&self,
|
||||
_input: AdjustWalletBalanceInput,
|
||||
) -> Result<
|
||||
Option<(
|
||||
StoredWalletSnapshot,
|
||||
super::types::StoredAdminWalletTransaction,
|
||||
)>,
|
||||
DataLayerError,
|
||||
> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn create_manual_wallet_recharge(
|
||||
&self,
|
||||
_input: CreateManualWalletRechargeInput,
|
||||
) -> Result<Option<(StoredWalletSnapshot, StoredAdminPaymentOrder)>, DataLayerError> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
async fn process_admin_wallet_refund(
|
||||
&self,
|
||||
_input: ProcessAdminWalletRefundInput,
|
||||
) -> Result<
|
||||
WalletMutationOutcome<(
|
||||
StoredWalletSnapshot,
|
||||
super::types::StoredAdminWalletRefund,
|
||||
super::types::StoredAdminWalletTransaction,
|
||||
)>,
|
||||
DataLayerError,
|
||||
> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
|
||||
async fn complete_admin_wallet_refund(
|
||||
&self,
|
||||
_input: CompleteAdminWalletRefundInput,
|
||||
) -> Result<WalletMutationOutcome<super::types::StoredAdminWalletRefund>, DataLayerError> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
|
||||
async fn fail_admin_wallet_refund(
|
||||
&self,
|
||||
_input: FailAdminWalletRefundInput,
|
||||
) -> Result<
|
||||
WalletMutationOutcome<(
|
||||
StoredWalletSnapshot,
|
||||
super::types::StoredAdminWalletRefund,
|
||||
Option<super::types::StoredAdminWalletTransaction>,
|
||||
)>,
|
||||
DataLayerError,
|
||||
> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
|
||||
async fn expire_admin_payment_order(
|
||||
&self,
|
||||
_order_id: &str,
|
||||
) -> Result<WalletMutationOutcome<(StoredAdminPaymentOrder, bool)>, DataLayerError> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
|
||||
async fn fail_admin_payment_order(
|
||||
&self,
|
||||
_order_id: &str,
|
||||
) -> Result<WalletMutationOutcome<StoredAdminPaymentOrder>, DataLayerError> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
|
||||
async fn credit_admin_payment_order(
|
||||
&self,
|
||||
_input: CreditAdminPaymentOrderInput,
|
||||
) -> Result<WalletMutationOutcome<(StoredAdminPaymentOrder, bool)>, DataLayerError> {
|
||||
Ok(WalletMutationOutcome::NotFound)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -209,8 +442,7 @@ impl WalletWriteRepository for InMemoryWalletRepository {
|
||||
mod tests {
|
||||
use super::InMemoryWalletRepository;
|
||||
use crate::repository::wallet::{
|
||||
StoredWalletSnapshot, UsageSettlementInput, WalletLookupKey, WalletReadRepository,
|
||||
WalletWriteRepository,
|
||||
AdminWalletListQuery, StoredWalletSnapshot, WalletLookupKey, WalletReadRepository,
|
||||
};
|
||||
|
||||
fn sample_wallet() -> StoredWalletSnapshot {
|
||||
@@ -244,27 +476,73 @@ mod tests {
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn settles_usage_against_wallet_and_provider_quota() {
|
||||
let repository = InMemoryWalletRepository::seed(vec![sample_wallet()]);
|
||||
let settlement = repository
|
||||
.settle_usage(UsageSettlementInput {
|
||||
request_id: "req-1".to_string(),
|
||||
user_id: Some("user-1".to_string()),
|
||||
api_key_id: Some("key-1".to_string()),
|
||||
provider_id: Some("provider-1".to_string()),
|
||||
status: "completed".to_string(),
|
||||
billing_status: "pending".to_string(),
|
||||
total_cost_usd: 3.0,
|
||||
actual_total_cost_usd: 1.5,
|
||||
finalized_at_unix_secs: Some(200),
|
||||
async fn lists_admin_wallets_with_filters_and_pagination() {
|
||||
let repository = InMemoryWalletRepository::seed(vec![
|
||||
sample_wallet(),
|
||||
StoredWalletSnapshot::new(
|
||||
"wallet-2".to_string(),
|
||||
Some("user-2".to_string()),
|
||||
None,
|
||||
3.0,
|
||||
1.0,
|
||||
"finite".to_string(),
|
||||
"USD".to_string(),
|
||||
"inactive".to_string(),
|
||||
0.0,
|
||||
0.0,
|
||||
0.0,
|
||||
0.0,
|
||||
90,
|
||||
)
|
||||
.expect("wallet should build"),
|
||||
StoredWalletSnapshot::new(
|
||||
"wallet-3".to_string(),
|
||||
None,
|
||||
Some("key-3".to_string()),
|
||||
5.0,
|
||||
0.0,
|
||||
"unlimited".to_string(),
|
||||
"USD".to_string(),
|
||||
"active".to_string(),
|
||||
0.0,
|
||||
0.0,
|
||||
0.0,
|
||||
0.0,
|
||||
110,
|
||||
)
|
||||
.expect("wallet should build"),
|
||||
]);
|
||||
|
||||
let page = repository
|
||||
.list_admin_wallets(&AdminWalletListQuery {
|
||||
status: Some("active".to_string()),
|
||||
owner_type: Some("api_key".to_string()),
|
||||
limit: 1,
|
||||
offset: 0,
|
||||
})
|
||||
.await
|
||||
.expect("settlement should succeed")
|
||||
.expect("settlement should exist");
|
||||
.expect("list should succeed");
|
||||
|
||||
assert_eq!(settlement.billing_status, "settled");
|
||||
assert_eq!(settlement.wallet_balance_before, Some(12.0));
|
||||
assert_eq!(settlement.wallet_balance_after, Some(9.0));
|
||||
assert_eq!(settlement.provider_monthly_used_usd, Some(1.5));
|
||||
assert_eq!(page.total, 2);
|
||||
assert_eq!(page.items.len(), 1);
|
||||
assert_eq!(page.items[0].id, "wallet-3");
|
||||
assert_eq!(page.items[0].updated_at_unix_secs, Some(110));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn daily_usage_queries_default_to_empty_in_memory() {
|
||||
let repository = InMemoryWalletRepository::seed(vec![sample_wallet()]);
|
||||
let today = repository
|
||||
.find_wallet_today_usage("wallet-1", "Asia/Shanghai")
|
||||
.await
|
||||
.expect("lookup should succeed");
|
||||
let history = repository
|
||||
.list_wallet_daily_usage_history("wallet-1", "Asia/Shanghai", 20)
|
||||
.await
|
||||
.expect("history should succeed");
|
||||
|
||||
assert!(today.is_none());
|
||||
assert_eq!(history.total, 0);
|
||||
assert!(history.items.is_empty());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -5,6 +5,19 @@ mod types;
|
||||
pub use memory::InMemoryWalletRepository;
|
||||
pub use sql::SqlxWalletRepository;
|
||||
pub use types::{
|
||||
StoredUsageSettlement, StoredWalletSnapshot, UsageSettlementInput, WalletLookupKey,
|
||||
AdjustWalletBalanceInput, AdminPaymentCallbackRecord, AdminPaymentOrderListQuery,
|
||||
AdminWalletLedgerQuery, AdminWalletListQuery, AdminWalletPaymentOrderRecord,
|
||||
AdminWalletRefundRecord, AdminWalletRefundRequestListQuery, AdminWalletTransactionRecord,
|
||||
CompleteAdminWalletRefundInput, CreateManualWalletRechargeInput,
|
||||
CreateWalletRechargeOrderInput, CreateWalletRechargeOrderOutcome,
|
||||
CreateWalletRefundRequestInput, CreateWalletRefundRequestOutcome, CreditAdminPaymentOrderInput,
|
||||
FailAdminWalletRefundInput, ProcessAdminWalletRefundInput, ProcessPaymentCallbackInput,
|
||||
ProcessPaymentCallbackOutcome, StoredAdminPaymentCallback, StoredAdminPaymentCallbackPage,
|
||||
StoredAdminPaymentOrder, StoredAdminPaymentOrderPage, StoredAdminWalletLedgerItem,
|
||||
StoredAdminWalletLedgerPage, StoredAdminWalletListItem, StoredAdminWalletListPage,
|
||||
StoredAdminWalletRefund, StoredAdminWalletRefundPage, StoredAdminWalletRefundRequestItem,
|
||||
StoredAdminWalletRefundRequestPage, StoredAdminWalletTransaction,
|
||||
StoredAdminWalletTransactionPage, StoredWalletDailyUsageLedger,
|
||||
StoredWalletDailyUsageLedgerPage, StoredWalletSnapshot, WalletLookupKey, WalletMutationOutcome,
|
||||
WalletReadRepository, WalletRepository, WalletWriteRepository,
|
||||
};
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -94,53 +94,500 @@ impl StoredWalletSnapshot {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletListQuery {
|
||||
pub status: Option<String>,
|
||||
pub owner_type: Option<String>,
|
||||
pub limit: usize,
|
||||
pub offset: usize,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct UsageSettlementInput {
|
||||
pub request_id: String,
|
||||
pub struct StoredAdminWalletListItem {
|
||||
pub id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub api_key_id: Option<String>,
|
||||
pub provider_id: Option<String>,
|
||||
pub balance: f64,
|
||||
pub gift_balance: f64,
|
||||
pub limit_mode: String,
|
||||
pub currency: String,
|
||||
pub status: String,
|
||||
pub billing_status: String,
|
||||
pub total_cost_usd: f64,
|
||||
pub actual_total_cost_usd: f64,
|
||||
pub finalized_at_unix_secs: Option<u64>,
|
||||
pub total_recharged: f64,
|
||||
pub total_consumed: f64,
|
||||
pub total_refunded: f64,
|
||||
pub total_adjusted: f64,
|
||||
pub user_name: Option<String>,
|
||||
pub api_key_name: Option<String>,
|
||||
pub created_at_unix_secs: Option<u64>,
|
||||
pub updated_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
impl UsageSettlementInput {
|
||||
pub fn validate(&self) -> Result<(), crate::DataLayerError> {
|
||||
if self.request_id.trim().is_empty() {
|
||||
return Err(crate::DataLayerError::InvalidInput(
|
||||
"wallet settlement request_id cannot be empty".to_string(),
|
||||
));
|
||||
}
|
||||
if self.status.trim().is_empty() || self.billing_status.trim().is_empty() {
|
||||
return Err(crate::DataLayerError::InvalidInput(
|
||||
"wallet settlement status cannot be empty".to_string(),
|
||||
));
|
||||
}
|
||||
if !self.total_cost_usd.is_finite() || !self.actual_total_cost_usd.is_finite() {
|
||||
return Err(crate::DataLayerError::InvalidInput(
|
||||
"wallet settlement cost must be finite".to_string(),
|
||||
));
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletListPage {
|
||||
pub items: Vec<StoredAdminWalletListItem>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletLedgerQuery {
|
||||
pub category: Option<String>,
|
||||
pub reason_code: Option<String>,
|
||||
pub owner_type: Option<String>,
|
||||
pub limit: usize,
|
||||
pub offset: usize,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredUsageSettlement {
|
||||
pub request_id: String,
|
||||
pub wallet_id: Option<String>,
|
||||
pub billing_status: String,
|
||||
pub wallet_balance_before: Option<f64>,
|
||||
pub wallet_balance_after: Option<f64>,
|
||||
pub wallet_recharge_balance_before: Option<f64>,
|
||||
pub wallet_recharge_balance_after: Option<f64>,
|
||||
pub wallet_gift_balance_before: Option<f64>,
|
||||
pub wallet_gift_balance_after: Option<f64>,
|
||||
pub provider_monthly_used_usd: Option<f64>,
|
||||
pub finalized_at_unix_secs: Option<u64>,
|
||||
pub struct StoredAdminWalletLedgerItem {
|
||||
pub id: String,
|
||||
pub wallet_id: String,
|
||||
pub category: String,
|
||||
pub reason_code: String,
|
||||
pub amount: f64,
|
||||
pub balance_before: f64,
|
||||
pub balance_after: f64,
|
||||
pub recharge_balance_before: f64,
|
||||
pub recharge_balance_after: f64,
|
||||
pub gift_balance_before: f64,
|
||||
pub gift_balance_after: f64,
|
||||
pub link_type: Option<String>,
|
||||
pub link_id: Option<String>,
|
||||
pub operator_id: Option<String>,
|
||||
pub operator_name: Option<String>,
|
||||
pub operator_email: Option<String>,
|
||||
pub description: Option<String>,
|
||||
pub wallet_user_id: Option<String>,
|
||||
pub wallet_user_name: Option<String>,
|
||||
pub wallet_api_key_id: Option<String>,
|
||||
pub api_key_name: Option<String>,
|
||||
pub wallet_status: String,
|
||||
pub created_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletLedgerPage {
|
||||
pub items: Vec<StoredAdminWalletLedgerItem>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletRefundRequestListQuery {
|
||||
pub status: Option<String>,
|
||||
pub limit: usize,
|
||||
pub offset: usize,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletRefundRequestItem {
|
||||
pub id: String,
|
||||
pub refund_no: String,
|
||||
pub wallet_id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub source_type: String,
|
||||
pub source_id: Option<String>,
|
||||
pub refund_mode: String,
|
||||
pub amount_usd: f64,
|
||||
pub status: String,
|
||||
pub reason: Option<String>,
|
||||
pub failure_reason: Option<String>,
|
||||
pub gateway_refund_id: Option<String>,
|
||||
pub payout_method: Option<String>,
|
||||
pub payout_reference: Option<String>,
|
||||
pub payout_proof: Option<serde_json::Value>,
|
||||
pub requested_by: Option<String>,
|
||||
pub approved_by: Option<String>,
|
||||
pub processed_by: Option<String>,
|
||||
pub wallet_user_id: Option<String>,
|
||||
pub wallet_user_name: Option<String>,
|
||||
pub wallet_api_key_id: Option<String>,
|
||||
pub api_key_name: Option<String>,
|
||||
pub wallet_status: String,
|
||||
pub created_at_unix_secs: Option<u64>,
|
||||
pub updated_at_unix_secs: Option<u64>,
|
||||
pub processed_at_unix_secs: Option<u64>,
|
||||
pub completed_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletRefundRequestPage {
|
||||
pub items: Vec<StoredAdminWalletRefundRequestItem>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletTransaction {
|
||||
pub id: String,
|
||||
pub wallet_id: String,
|
||||
pub category: String,
|
||||
pub reason_code: String,
|
||||
pub amount: f64,
|
||||
pub balance_before: f64,
|
||||
pub balance_after: f64,
|
||||
pub recharge_balance_before: f64,
|
||||
pub recharge_balance_after: f64,
|
||||
pub gift_balance_before: f64,
|
||||
pub gift_balance_after: f64,
|
||||
pub link_type: Option<String>,
|
||||
pub link_id: Option<String>,
|
||||
pub operator_id: Option<String>,
|
||||
pub operator_name: Option<String>,
|
||||
pub operator_email: Option<String>,
|
||||
pub description: Option<String>,
|
||||
pub created_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletTransactionRecord {
|
||||
pub id: String,
|
||||
pub wallet_id: String,
|
||||
pub category: String,
|
||||
pub reason_code: String,
|
||||
pub amount: f64,
|
||||
pub balance_before: f64,
|
||||
pub balance_after: f64,
|
||||
pub recharge_balance_before: f64,
|
||||
pub recharge_balance_after: f64,
|
||||
pub gift_balance_before: f64,
|
||||
pub gift_balance_after: f64,
|
||||
pub link_type: Option<String>,
|
||||
pub link_id: Option<String>,
|
||||
pub operator_id: Option<String>,
|
||||
pub description: Option<String>,
|
||||
pub created_at_unix_secs: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletTransactionPage {
|
||||
pub items: Vec<StoredAdminWalletTransaction>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredWalletDailyUsageLedger {
|
||||
pub id: Option<String>,
|
||||
pub billing_date: String,
|
||||
pub billing_timezone: String,
|
||||
pub total_cost_usd: f64,
|
||||
pub total_requests: u64,
|
||||
pub input_tokens: u64,
|
||||
pub output_tokens: u64,
|
||||
pub cache_creation_tokens: u64,
|
||||
pub cache_read_tokens: u64,
|
||||
pub first_finalized_at_unix_secs: Option<u64>,
|
||||
pub last_finalized_at_unix_secs: Option<u64>,
|
||||
pub aggregated_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredWalletDailyUsageLedgerPage {
|
||||
pub items: Vec<StoredWalletDailyUsageLedger>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletRefund {
|
||||
pub id: String,
|
||||
pub refund_no: String,
|
||||
pub wallet_id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub source_type: String,
|
||||
pub source_id: Option<String>,
|
||||
pub refund_mode: String,
|
||||
pub amount_usd: f64,
|
||||
pub status: String,
|
||||
pub reason: Option<String>,
|
||||
pub failure_reason: Option<String>,
|
||||
pub gateway_refund_id: Option<String>,
|
||||
pub payout_method: Option<String>,
|
||||
pub payout_reference: Option<String>,
|
||||
pub payout_proof: Option<serde_json::Value>,
|
||||
pub requested_by: Option<String>,
|
||||
pub approved_by: Option<String>,
|
||||
pub processed_by: Option<String>,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub updated_at_unix_secs: u64,
|
||||
pub processed_at_unix_secs: Option<u64>,
|
||||
pub completed_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletRefundRecord {
|
||||
pub id: String,
|
||||
pub refund_no: String,
|
||||
pub wallet_id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub source_type: String,
|
||||
pub source_id: Option<String>,
|
||||
pub refund_mode: String,
|
||||
pub amount_usd: f64,
|
||||
pub status: String,
|
||||
pub reason: Option<String>,
|
||||
pub failure_reason: Option<String>,
|
||||
pub gateway_refund_id: Option<String>,
|
||||
pub payout_method: Option<String>,
|
||||
pub payout_reference: Option<String>,
|
||||
pub payout_proof: Option<serde_json::Value>,
|
||||
pub requested_by: Option<String>,
|
||||
pub approved_by: Option<String>,
|
||||
pub processed_by: Option<String>,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub updated_at_unix_secs: u64,
|
||||
pub processed_at_unix_secs: Option<u64>,
|
||||
pub completed_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminWalletRefundPage {
|
||||
pub items: Vec<StoredAdminWalletRefund>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Eq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminPaymentOrderListQuery {
|
||||
pub status: Option<String>,
|
||||
pub payment_method: Option<String>,
|
||||
pub limit: usize,
|
||||
pub offset: usize,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminPaymentOrder {
|
||||
pub id: String,
|
||||
pub order_no: String,
|
||||
pub wallet_id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub amount_usd: f64,
|
||||
pub pay_amount: Option<f64>,
|
||||
pub pay_currency: Option<String>,
|
||||
pub exchange_rate: Option<f64>,
|
||||
pub refunded_amount_usd: f64,
|
||||
pub refundable_amount_usd: f64,
|
||||
pub payment_method: String,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub gateway_response: Option<serde_json::Value>,
|
||||
pub status: String,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub paid_at_unix_secs: Option<u64>,
|
||||
pub credited_at_unix_secs: Option<u64>,
|
||||
pub expires_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminWalletPaymentOrderRecord {
|
||||
pub id: String,
|
||||
pub order_no: String,
|
||||
pub wallet_id: String,
|
||||
pub user_id: Option<String>,
|
||||
pub amount_usd: f64,
|
||||
pub pay_amount: Option<f64>,
|
||||
pub pay_currency: Option<String>,
|
||||
pub exchange_rate: Option<f64>,
|
||||
pub refunded_amount_usd: f64,
|
||||
pub refundable_amount_usd: f64,
|
||||
pub payment_method: String,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub status: String,
|
||||
pub gateway_response: Option<serde_json::Value>,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub paid_at_unix_secs: Option<u64>,
|
||||
pub credited_at_unix_secs: Option<u64>,
|
||||
pub expires_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminPaymentOrderPage {
|
||||
pub items: Vec<StoredAdminPaymentOrder>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminPaymentCallback {
|
||||
pub id: String,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub payment_method: String,
|
||||
pub callback_key: String,
|
||||
pub order_no: Option<String>,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub payload_hash: Option<String>,
|
||||
pub signature_valid: bool,
|
||||
pub status: String,
|
||||
pub payload: Option<serde_json::Value>,
|
||||
pub error_message: Option<String>,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub processed_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdminPaymentCallbackRecord {
|
||||
pub id: String,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub payment_method: String,
|
||||
pub callback_key: String,
|
||||
pub order_no: Option<String>,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub payload_hash: Option<String>,
|
||||
pub signature_valid: bool,
|
||||
pub status: String,
|
||||
pub payload: Option<serde_json::Value>,
|
||||
pub error_message: Option<String>,
|
||||
pub created_at_unix_secs: u64,
|
||||
pub processed_at_unix_secs: Option<u64>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, Default, serde::Serialize, serde::Deserialize)]
|
||||
pub struct StoredAdminPaymentCallbackPage {
|
||||
pub items: Vec<StoredAdminPaymentCallback>,
|
||||
pub total: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct CreateWalletRechargeOrderInput {
|
||||
pub preferred_wallet_id: Option<String>,
|
||||
pub user_id: String,
|
||||
pub amount_usd: f64,
|
||||
pub pay_amount: Option<f64>,
|
||||
pub pay_currency: Option<String>,
|
||||
pub exchange_rate: Option<f64>,
|
||||
pub payment_method: String,
|
||||
pub gateway_order_id: String,
|
||||
pub gateway_response: serde_json::Value,
|
||||
pub order_no: String,
|
||||
pub expires_at_unix_secs: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub enum CreateWalletRechargeOrderOutcome {
|
||||
Created(StoredAdminPaymentOrder),
|
||||
WalletInactive,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct CreateWalletRefundRequestInput {
|
||||
pub wallet_id: String,
|
||||
pub user_id: String,
|
||||
pub amount_usd: f64,
|
||||
pub payment_order_id: Option<String>,
|
||||
pub source_type: Option<String>,
|
||||
pub source_id: Option<String>,
|
||||
pub refund_mode: Option<String>,
|
||||
pub reason: Option<String>,
|
||||
pub idempotency_key: Option<String>,
|
||||
pub refund_no: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub enum CreateWalletRefundRequestOutcome {
|
||||
Created(StoredAdminWalletRefund),
|
||||
Duplicate(StoredAdminWalletRefund),
|
||||
WalletMissing,
|
||||
RefundAmountExceedsAvailableBalance,
|
||||
PaymentOrderNotFound,
|
||||
PaymentOrderNotRefundable,
|
||||
RefundAmountExceedsAvailableOrderAmount,
|
||||
DuplicateRejected,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct ProcessPaymentCallbackInput {
|
||||
pub payment_method: String,
|
||||
pub callback_key: String,
|
||||
pub order_no: Option<String>,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub amount_usd: f64,
|
||||
pub pay_amount: Option<f64>,
|
||||
pub pay_currency: Option<String>,
|
||||
pub exchange_rate: Option<f64>,
|
||||
pub payload_hash: String,
|
||||
pub payload: serde_json::Value,
|
||||
pub signature_valid: bool,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub enum ProcessPaymentCallbackOutcome {
|
||||
DuplicateProcessed {
|
||||
order_id: Option<String>,
|
||||
},
|
||||
Failed {
|
||||
duplicate: bool,
|
||||
error: String,
|
||||
},
|
||||
AlreadyCredited {
|
||||
duplicate: bool,
|
||||
order_id: String,
|
||||
order_no: String,
|
||||
wallet_id: String,
|
||||
},
|
||||
Applied {
|
||||
duplicate: bool,
|
||||
order_id: String,
|
||||
order_no: String,
|
||||
wallet_id: String,
|
||||
order: StoredAdminPaymentOrder,
|
||||
},
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq)]
|
||||
pub enum WalletMutationOutcome<T> {
|
||||
Applied(T),
|
||||
NotFound,
|
||||
Invalid(String),
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct AdjustWalletBalanceInput {
|
||||
pub wallet_id: String,
|
||||
pub amount_usd: f64,
|
||||
pub balance_type: String,
|
||||
pub operator_id: Option<String>,
|
||||
pub description: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct CreateManualWalletRechargeInput {
|
||||
pub wallet_id: String,
|
||||
pub amount_usd: f64,
|
||||
pub payment_method: String,
|
||||
pub operator_id: Option<String>,
|
||||
pub description: Option<String>,
|
||||
pub order_no: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct ProcessAdminWalletRefundInput {
|
||||
pub wallet_id: String,
|
||||
pub refund_id: String,
|
||||
pub operator_id: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct CompleteAdminWalletRefundInput {
|
||||
pub wallet_id: String,
|
||||
pub refund_id: String,
|
||||
pub gateway_refund_id: Option<String>,
|
||||
pub payout_reference: Option<String>,
|
||||
pub payout_proof: Option<serde_json::Value>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct FailAdminWalletRefundInput {
|
||||
pub wallet_id: String,
|
||||
pub refund_id: String,
|
||||
pub reason: String,
|
||||
pub operator_id: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||||
pub struct CreditAdminPaymentOrderInput {
|
||||
pub order_id: String,
|
||||
pub gateway_order_id: Option<String>,
|
||||
pub pay_amount: Option<f64>,
|
||||
pub pay_currency: Option<String>,
|
||||
pub exchange_rate: Option<f64>,
|
||||
pub gateway_response_patch: Option<serde_json::Value>,
|
||||
pub operator_id: Option<String>,
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
@@ -159,14 +606,156 @@ pub trait WalletReadRepository: Send + Sync {
|
||||
&self,
|
||||
api_key_ids: &[String],
|
||||
) -> Result<Vec<StoredWalletSnapshot>, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_wallets(
|
||||
&self,
|
||||
query: &AdminWalletListQuery,
|
||||
) -> Result<StoredAdminWalletListPage, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_wallet_ledger(
|
||||
&self,
|
||||
query: &AdminWalletLedgerQuery,
|
||||
) -> Result<StoredAdminWalletLedgerPage, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_wallet_refund_requests(
|
||||
&self,
|
||||
query: &AdminWalletRefundRequestListQuery,
|
||||
) -> Result<StoredAdminWalletRefundRequestPage, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_wallet_transactions(
|
||||
&self,
|
||||
wallet_id: &str,
|
||||
limit: usize,
|
||||
offset: usize,
|
||||
) -> Result<StoredAdminWalletTransactionPage, crate::DataLayerError>;
|
||||
|
||||
async fn find_wallet_today_usage(
|
||||
&self,
|
||||
wallet_id: &str,
|
||||
billing_timezone: &str,
|
||||
) -> Result<Option<StoredWalletDailyUsageLedger>, crate::DataLayerError>;
|
||||
|
||||
async fn list_wallet_daily_usage_history(
|
||||
&self,
|
||||
wallet_id: &str,
|
||||
billing_timezone: &str,
|
||||
limit: usize,
|
||||
) -> Result<StoredWalletDailyUsageLedgerPage, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_wallet_refunds(
|
||||
&self,
|
||||
wallet_id: &str,
|
||||
limit: usize,
|
||||
offset: usize,
|
||||
) -> Result<StoredAdminWalletRefundPage, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_payment_orders(
|
||||
&self,
|
||||
query: &AdminPaymentOrderListQuery,
|
||||
) -> Result<StoredAdminPaymentOrderPage, crate::DataLayerError>;
|
||||
|
||||
async fn find_admin_payment_order(
|
||||
&self,
|
||||
order_id: &str,
|
||||
) -> Result<Option<StoredAdminPaymentOrder>, crate::DataLayerError>;
|
||||
|
||||
async fn list_wallet_payment_orders_by_user_id(
|
||||
&self,
|
||||
user_id: &str,
|
||||
limit: usize,
|
||||
offset: usize,
|
||||
) -> Result<StoredAdminPaymentOrderPage, crate::DataLayerError>;
|
||||
|
||||
async fn find_wallet_payment_order_by_user_id(
|
||||
&self,
|
||||
user_id: &str,
|
||||
order_id: &str,
|
||||
) -> Result<Option<StoredAdminPaymentOrder>, crate::DataLayerError>;
|
||||
|
||||
async fn find_wallet_refund(
|
||||
&self,
|
||||
wallet_id: &str,
|
||||
refund_id: &str,
|
||||
) -> Result<Option<StoredAdminWalletRefund>, crate::DataLayerError>;
|
||||
|
||||
async fn list_admin_payment_callbacks(
|
||||
&self,
|
||||
payment_method: Option<&str>,
|
||||
limit: usize,
|
||||
offset: usize,
|
||||
) -> Result<StoredAdminPaymentCallbackPage, crate::DataLayerError>;
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
pub trait WalletWriteRepository: Send + Sync {
|
||||
async fn settle_usage(
|
||||
async fn create_wallet_recharge_order(
|
||||
&self,
|
||||
input: UsageSettlementInput,
|
||||
) -> Result<Option<StoredUsageSettlement>, crate::DataLayerError>;
|
||||
input: CreateWalletRechargeOrderInput,
|
||||
) -> Result<CreateWalletRechargeOrderOutcome, crate::DataLayerError>;
|
||||
|
||||
async fn create_wallet_refund_request(
|
||||
&self,
|
||||
input: CreateWalletRefundRequestInput,
|
||||
) -> Result<CreateWalletRefundRequestOutcome, crate::DataLayerError>;
|
||||
|
||||
async fn process_payment_callback(
|
||||
&self,
|
||||
input: ProcessPaymentCallbackInput,
|
||||
) -> Result<ProcessPaymentCallbackOutcome, crate::DataLayerError>;
|
||||
|
||||
async fn adjust_wallet_balance(
|
||||
&self,
|
||||
input: AdjustWalletBalanceInput,
|
||||
) -> Result<Option<(StoredWalletSnapshot, StoredAdminWalletTransaction)>, crate::DataLayerError>;
|
||||
|
||||
async fn create_manual_wallet_recharge(
|
||||
&self,
|
||||
input: CreateManualWalletRechargeInput,
|
||||
) -> Result<Option<(StoredWalletSnapshot, StoredAdminPaymentOrder)>, crate::DataLayerError>;
|
||||
|
||||
async fn process_admin_wallet_refund(
|
||||
&self,
|
||||
input: ProcessAdminWalletRefundInput,
|
||||
) -> Result<
|
||||
WalletMutationOutcome<(
|
||||
StoredWalletSnapshot,
|
||||
StoredAdminWalletRefund,
|
||||
StoredAdminWalletTransaction,
|
||||
)>,
|
||||
crate::DataLayerError,
|
||||
>;
|
||||
|
||||
async fn complete_admin_wallet_refund(
|
||||
&self,
|
||||
input: CompleteAdminWalletRefundInput,
|
||||
) -> Result<WalletMutationOutcome<StoredAdminWalletRefund>, crate::DataLayerError>;
|
||||
|
||||
async fn fail_admin_wallet_refund(
|
||||
&self,
|
||||
input: FailAdminWalletRefundInput,
|
||||
) -> Result<
|
||||
WalletMutationOutcome<(
|
||||
StoredWalletSnapshot,
|
||||
StoredAdminWalletRefund,
|
||||
Option<StoredAdminWalletTransaction>,
|
||||
)>,
|
||||
crate::DataLayerError,
|
||||
>;
|
||||
|
||||
async fn expire_admin_payment_order(
|
||||
&self,
|
||||
order_id: &str,
|
||||
) -> Result<WalletMutationOutcome<(StoredAdminPaymentOrder, bool)>, crate::DataLayerError>;
|
||||
|
||||
async fn fail_admin_payment_order(
|
||||
&self,
|
||||
order_id: &str,
|
||||
) -> Result<WalletMutationOutcome<StoredAdminPaymentOrder>, crate::DataLayerError>;
|
||||
|
||||
async fn credit_admin_payment_order(
|
||||
&self,
|
||||
input: CreditAdminPaymentOrderInput,
|
||||
) -> Result<WalletMutationOutcome<(StoredAdminPaymentOrder, bool)>, crate::DataLayerError>;
|
||||
}
|
||||
|
||||
pub trait WalletRepository: WalletReadRepository + WalletWriteRepository + Send + Sync {}
|
||||
@@ -175,7 +764,8 @@ impl<T> WalletRepository for T where T: WalletReadRepository + WalletWriteReposi
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{StoredWalletSnapshot, UsageSettlementInput};
|
||||
use super::StoredWalletSnapshot;
|
||||
use crate::repository::settlement::UsageSettlementInput;
|
||||
|
||||
#[test]
|
||||
fn rejects_invalid_wallet_snapshot() {
|
||||
|
||||
Reference in New Issue
Block a user