27 lines
660 B
Go
27 lines
660 B
Go
package repository
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"git.sch9.ru/new_gate/ms-tester/pkg"
|
|
"github.com/jackc/pgerrcode"
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
)
|
|
|
|
func handlePgErr(err error, op string) error {
|
|
var pgErr *pgconn.PgError
|
|
if errors.As(err, &pgErr) {
|
|
if pgerrcode.IsIntegrityConstraintViolation(pgErr.Code) {
|
|
return pkg.Wrap(pkg.ErrBadInput, err, op, pgErr.Message)
|
|
}
|
|
if pgerrcode.IsNoData(pgErr.Code) {
|
|
return pkg.Wrap(pkg.ErrNotFound, err, op, pgErr.Message)
|
|
}
|
|
}
|
|
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return pkg.Wrap(pkg.ErrNotFound, err, op, "no rows found")
|
|
}
|
|
|
|
return pkg.Wrap(pkg.ErrUnhandled, err, op, "unexpected error")
|
|
}
|