b160e13f20
This PR improves our Stripe event handling by keeping track of the last event we've seen for each record. The `billing_customers` and `billing_subscriptions` tables both have a new `last_stripe_event_id` column. When we apply an event to one of these records, we store the event ID that was applied. Then, when we are going through events we can ignore any event that has an ID that came before the `last_stripe_event_id` (based on the lexicographical ordering of the IDs). Release Notes: - N/A
86 lines
2.7 KiB
Rust
86 lines
2.7 KiB
Rust
use super::*;
|
|
|
|
#[derive(Debug)]
|
|
pub struct CreateBillingCustomerParams {
|
|
pub user_id: UserId,
|
|
pub stripe_customer_id: String,
|
|
pub last_stripe_event_id: Option<String>,
|
|
}
|
|
|
|
#[derive(Debug, Default)]
|
|
pub struct UpdateBillingCustomerParams {
|
|
pub user_id: ActiveValue<UserId>,
|
|
pub stripe_customer_id: ActiveValue<String>,
|
|
pub last_stripe_event_id: ActiveValue<Option<String>>,
|
|
}
|
|
|
|
impl Database {
|
|
/// Creates a new billing customer.
|
|
pub async fn create_billing_customer(
|
|
&self,
|
|
params: &CreateBillingCustomerParams,
|
|
) -> Result<billing_customer::Model> {
|
|
self.transaction(|tx| async move {
|
|
let customer = billing_customer::Entity::insert(billing_customer::ActiveModel {
|
|
user_id: ActiveValue::set(params.user_id),
|
|
stripe_customer_id: ActiveValue::set(params.stripe_customer_id.clone()),
|
|
..Default::default()
|
|
})
|
|
.exec_with_returning(&*tx)
|
|
.await?;
|
|
|
|
Ok(customer)
|
|
})
|
|
.await
|
|
}
|
|
|
|
/// Updates the specified billing customer.
|
|
pub async fn update_billing_customer(
|
|
&self,
|
|
id: BillingCustomerId,
|
|
params: &UpdateBillingCustomerParams,
|
|
) -> Result<()> {
|
|
self.transaction(|tx| async move {
|
|
billing_customer::Entity::update(billing_customer::ActiveModel {
|
|
id: ActiveValue::set(id),
|
|
user_id: params.user_id.clone(),
|
|
stripe_customer_id: params.stripe_customer_id.clone(),
|
|
last_stripe_event_id: params.last_stripe_event_id.clone(),
|
|
..Default::default()
|
|
})
|
|
.exec(&*tx)
|
|
.await?;
|
|
|
|
Ok(())
|
|
})
|
|
.await
|
|
}
|
|
|
|
/// Returns the billing customer for the user with the specified ID.
|
|
pub async fn get_billing_customer_by_user_id(
|
|
&self,
|
|
user_id: UserId,
|
|
) -> Result<Option<billing_customer::Model>> {
|
|
self.transaction(|tx| async move {
|
|
Ok(billing_customer::Entity::find()
|
|
.filter(billing_customer::Column::UserId.eq(user_id))
|
|
.one(&*tx)
|
|
.await?)
|
|
})
|
|
.await
|
|
}
|
|
|
|
/// Returns the billing customer for the user with the specified Stripe customer ID.
|
|
pub async fn get_billing_customer_by_stripe_customer_id(
|
|
&self,
|
|
stripe_customer_id: &str,
|
|
) -> Result<Option<billing_customer::Model>> {
|
|
self.transaction(|tx| async move {
|
|
Ok(billing_customer::Entity::find()
|
|
.filter(billing_customer::Column::StripeCustomerId.eq(stripe_customer_id))
|
|
.one(&*tx)
|
|
.await?)
|
|
})
|
|
.await
|
|
}
|
|
}
|