Function consumeOrganizationCredits

packages/billing/src/org-billing.ts:271–323

(params: {
  organizationId: string
  creditsToConsume: number
  logger: Logger
})

 * Uses advisory locks to serialize credit operations per organization.
 */
export async function consumeOrganizationCredits(params: {
  organizationId: string
  creditsToConsume: number
  logger: Logger
}): Promise<CreditConsumptionResult> {
  const { organizationId, creditsToConsume, logger } = params

  const { result, lockWaitMs } = await withAdvisoryLockTransaction({
    callback: async (tx) => {
      const now = new Date()
      const activeGrants = await getOrderedActiveOrganizationGrants({
        ...params,
        now,
        conn: tx,
      })

      if (activeGrants.length === 0) {
        logger.error(
          { organizationId, creditsToConsume },
          'No active organization grants found to consume credits from',
        )
        throw new Error('No active organization grants found')
      }

      const consumeResult = await consumeFromOrderedGrants({
        userId: organizationId,
        creditsToConsume,
        grants: activeGrants,
        tx,
        logger,
      })

      return consumeResult
    },
    lockKey: `org:${organizationId}`,
    context: { organizationId, creditsToConsume },
    logger,
  })

  // Log successful organization credit consumption with lock timing
  logger.info(
    {
      organizationId,
      creditsConsumed: result.consumed,
      creditsRequested: creditsToConsume,
      fromPurchased: result.fromPurchased,
      lockWaitMs,
    },
    'Organization credits consumed',
  )

  return result
}

/**
 * Grants credits to an organization.