onasty: internal/store/psqlutil/psqlutil.go (babda7d822d93649597a4892e0dada0724e215b7)

onasty/internal/store/psqlutil/psqlutil.go (view raw)

Smirnov Oleksandr refactor: dont register uuid in pgx (#86), 1 year ago

package psqlutil

import (
	"context"
	"errors"

	"github.com/jackc/pgx/v5/pgconn"
	"github.com/jackc/pgx/v5/pgxpool"
)

type DB struct{ *pgxpool.Pool }

func Connect(ctx context.Context, dsn string) (*DB, error) {
	db, err := pgxpool.New(ctx, dsn)
	if err != nil {
		return nil, err
	}

	if err := db.Ping(ctx); err != nil {
		return nil, err
	}

	return &DB{
		Pool: db,
	}, nil
}

func (db *DB) Close() error {
	db.Pool.Close()
	return nil
}

// IsDuplicateErr function that checks if the error is a duplicate key violation.
func IsDuplicateErr(err error, constraintName ...string) bool {
	var pgErr *pgconn.PgError
	if errors.As(err, &pgErr) {
		if len(constraintName) == 0 || len(constraintName) == 1 {
			return pgErr.Code == "23505" && // unique_violation
				pgErr.ConstraintName == constraintName[0]
		}
		return pgErr.Code == "23505" // unique_violation
	}
	return false
}

1	package psqlutil
2
3	import (
4	"context"
5	"errors"
6
7	"github.com/jackc/pgx/v5/pgconn"
8	"github.com/jackc/pgx/v5/pgxpool"
9	)
10
11	type DB struct{ *pgxpool.Pool }
12
13	func Connect(ctx context.Context, dsn string) (*DB, error) {
14	db, err := pgxpool.New(ctx, dsn)
15	if err != nil {
16	return nil, err
17	}
18
19	if err := db.Ping(ctx); err != nil {
20	return nil, err
21	}
22
23	return &DB{
24	Pool: db,
25	}, nil
26	}
27
28	func (db *DB) Close() error {
29	db.Pool.Close()
30	return nil
31	}
32
33	// IsDuplicateErr function that checks if the error is a duplicate key violation.
34	func IsDuplicateErr(err error, constraintName ...string) bool {
35	var pgErr *pgconn.PgError
36	if errors.As(err, &pgErr) {
37	if len(constraintName) == 0 \|\| len(constraintName) == 1 {
38	return pgErr.Code == "23505" && // unique_violation
39	pgErr.ConstraintName == constraintName[0]
40	}
41	return pgErr.Code == "23505" // unique_violation
42	}
43	return false
44	}