share timeout const

Author: icecrasher321

apps/sim/lib/billing/constants.ts

@@ -34,6 +34,11 @@ export const SEARCH_TOOL_COST = 0.01
  */
 export const DEFAULT_OVERAGE_THRESHOLD = 100
 
+/**
+ * Maximum time to wait on billing coordination row locks before retrying later.
+ */
+export const BILLING_LOCK_TIMEOUT_MS = 5_000
+
 /**
  * Available credit tiers. Each tier maps a credit amount to the underlying dollar cost.
  * 1 credit = $0.005, so credits = dollars * 200.

apps/sim/lib/billing/threshold-billing.ts

@@ -2,7 +2,7 @@ import { db } from '@sim/db'
 import { member, organization, subscription, userStats } from '@sim/db/schema'
 import { createLogger } from '@sim/logger'
 import { and, eq, sql } from 'drizzle-orm'
-import { DEFAULT_OVERAGE_THRESHOLD } from '@/lib/billing/constants'
+import { BILLING_LOCK_TIMEOUT_MS, DEFAULT_OVERAGE_THRESHOLD } from '@/lib/billing/constants'
 import { getEffectiveBillingStatus, isOrganizationBillingBlocked } from '@/lib/billing/core/access'
 import { calculateSubscriptionOverage, computeOrgOverageAmount } from '@/lib/billing/core/billing'
 import {
@@ -22,7 +22,6 @@ import { enqueueOutboxEvent } from '@/lib/core/outbox/service'
 const logger = createLogger('ThresholdBilling')
 
 const OVERAGE_THRESHOLD = envNumber(env.OVERAGE_THRESHOLD_DOLLARS, DEFAULT_OVERAGE_THRESHOLD)
-const USER_STATS_LOCK_TIMEOUT_MS = 5_000
 const USAGE_TOTAL_EPSILON = 0.000001
 
 interface PersonalUsageSnapshot {
@@ -119,7 +118,7 @@ export async function checkAndBillOverageThreshold(userId: string): Promise<void
     const totalOverageCents = Math.round(currentOverage * 100)
 
     await db.transaction(async (tx) => {
-      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${USER_STATS_LOCK_TIMEOUT_MS}ms'`))
+      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${BILLING_LOCK_TIMEOUT_MS}ms'`))
 
       const statsRecords = await tx
         .select()
@@ -363,7 +362,7 @@ async function checkAndBillOrganizationOverageThreshold(organizationId: string):
     const totalOverageCents = Math.round(currentOverage * 100)
 
     await db.transaction(async (tx) => {
-      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${USER_STATS_LOCK_TIMEOUT_MS}ms'`))
+      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${BILLING_LOCK_TIMEOUT_MS}ms'`))
 
       const lockedOwnerRows = await tx
         .select({ userId: member.userId })

apps/sim/lib/billing/webhooks/invoices.ts

@@ -11,6 +11,7 @@ import { createLogger } from '@sim/logger'
 import { and, eq, inArray, isNull, ne, or, sql } from 'drizzle-orm'
 import type Stripe from 'stripe'
 import { getEmailSubject, PaymentFailedEmail, renderCreditPurchaseEmail } from '@/components/emails'
+import { BILLING_LOCK_TIMEOUT_MS } from '@/lib/billing/constants'
 import { calculateSubscriptionOverage, isSubscriptionOrgScoped } from '@/lib/billing/core/billing'
 import { addCredits, getCreditBalanceForEntity } from '@/lib/billing/credits/balance'
 import { setUsageLimitForCredits } from '@/lib/billing/credits/purchase'
@@ -389,6 +390,8 @@ export async function getBilledOverageForSubscription(sub: {
 export async function resetUsageForSubscription(sub: { plan: string | null; referenceId: string }) {
   if (await isSubscriptionOrgScoped(sub)) {
     await db.transaction(async (tx) => {
+      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${BILLING_LOCK_TIMEOUT_MS}ms'`))
+
       const ownerRows = await tx
         .select({ userId: member.userId })
         .from(member)
@@ -907,6 +910,8 @@ export async function handleInvoiceFinalized(event: Stripe.Event) {
         // then lock the tracker row so `billedOverageThisPeriod` is serialized
         // against threshold billing, resets, owner transfers, and retries.
         const phase1 = await db.transaction(async (tx) => {
+          await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${BILLING_LOCK_TIMEOUT_MS}ms'`))
+
           let trackerUserId = entityId
           if (entityType === 'organization') {
             const ownerRows = await tx