Function updateBalance

database/balance.go:881–913

updateBalance updates a balance entry in the database. This function handles the logic of updating all balance-related fields while ensuring data consistency using optimistic locking. The version field is incremented after a successful update to maintain control over concurrent modifications. Param

(ctx context.Context, tx *sql.Tx, balance *model.Balance)

// Returns:
// - error: Returns an error if the update operation fails at any point, including issues with metadata marshalling, query execution, or optimistic locking.
func updateBalance(ctx context.Context, tx *sql.Tx, balance *model.Balance) error {
	// SQL query to update the balance
	query := `
        UPDATE ledgerforge.balances
        SET balance = $2, credit_balance = $3, debit_balance = $4, inflight_balance = $5, inflight_credit_balance = $6, inflight_debit_balance = $7, currency = $8, currency_multiplier = $9, ledger_id = $10, created_at = $11, version = version + 1
        WHERE balance_id = $1 AND version = $12
    `

	// Execute the update query within the provided transaction context
	result, err := tx.ExecContext(ctx, query, balance.BalanceID, balance.Balance.String(), balance.CreditBalance.String(), balance.DebitBalance.String(), balance.InflightBalance.String(), balance.InflightCreditBalance.String(), balance.InflightDebitBalance.String(), balance.Currency, balance.CurrencyMultiplier, balance.LedgerID, balance.CreatedAt, balance.Version)
	if err != nil {
		// Return an error if the query execution fails
		return apierror.NewAPIError(apierror.ErrInternalServer, "Failed to update balance", err)
	}

	// Check if any rows were affected by the update
	rowsAffected, err := result.RowsAffected()
	if err != nil {
		// Return an error if unable to get affected rows
		return apierror.NewAPIError(apierror.ErrInternalServer, "Failed to get rows affected", err)
	}

	// If no rows were updated, return an optimistic locking error
	if rowsAffected == 0 {
		return apierror.NewAPIError(apierror.ErrConflict, fmt.Sprintf("Optimistic locking failure: balance with ID '%s' may have been updated or deleted by another transaction", balance.BalanceID), nil)
	}

	// Increment the version number after a successful update
	balance.Version++

	// Return nil indicating a successful update
	return nil
}

func updateBalanceSet(ctx context.Context, tx *sql.Tx, balances []*model.Balance) error {
	seen := make(map[string]struct{}, len(balances))