payments/internal/storage/error.go

package storage

import (
	"context"
	"fmt"
	"strings"

	"github.com/hanzo-fi/go-libs/v5/pkg/storage/postgres"

	"github.com/hanzo-fi/go-libs/v5/pkg/observe/log"
	"github.com/pkg/errors"
	"github.com/uptrace/bun"
)

var (
	ErrValidation        = errors.New("validation error")
	ErrNotFound          = errors.New("not found")
	ErrDuplicateKeyValue = errors.New("object already exists")
	// Don't want to expose the internal that is a foreign key violation to the
	// client through the API
	ErrForeignKeyViolation = errors.New("value not found")
)

var FKViolationColumn = []string{
	"connector_id",
	"bank_account_id",
	"payment_id",
	"pool_id",
	"schedule_id",
	"payment_initiation_id",
	"payment_initiation_reversal_id",
}

func e(msg string, err error) error {
	if err == nil {
		return nil
	}

	err = postgres.ResolveError(err)
	var failedUniquenessConstraintErr postgres.ErrConstraintsFailed
	if errors.As(err, &failedUniquenessConstraintErr) {
		return ErrDuplicateKeyValue
	}

	var fkConstraintErr postgres.ErrFKConstraintFailed
	if errors.As(err, &fkConstraintErr) {
		for _, column := range FKViolationColumn {
			if strings.Contains(fkConstraintErr.GetConstraint(), column) {
				return fmt.Errorf("%s: %w", column, ErrForeignKeyViolation)
			}
		}

		return ErrForeignKeyViolation
	}

	var validationErr postgres.ErrValidationFailed
	if errors.As(err, &validationErr) {
		return fmt.Errorf("%s: %w", validationErr.Message(), ErrValidation)
	}

	if errors.Is(err, postgres.ErrNotFound) {
		return ErrNotFound
	}

	return fmt.Errorf("%s: %w", msg, err)
}

// meant to be called in defer block
func rollbackOnTxError(ctx context.Context, tx *bun.Tx, err error) {
	if err == nil {
		return
	}

	if rollbackErr := tx.Rollback(); rollbackErr != nil {
		logging.FromContext(ctx).WithField("original_error", err.Error()).Errorf("failed to rollback transaction: %w", rollbackErr)
	}
}