sqlwrapper/db_conn_pool.go at master · beefllama/sqlwrapper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package sqlwrapper

import (
	"context"
	"database/sql"

	"github.com/pkg/errors"
	"go.uber.org/multierr"
)

func OpenDBConnection(ctx context.Context, driverName, dsn string) (DBConnPool, error) {
	conn, err := sql.Open(driverName, dsn)
	if err != nil {
		return nil, errors.Wrap(err, "failed to open database connection")
	}

	if pingErr := conn.PingContext(ctx); pingErr != nil {
		return nil, errors.Wrap(err, "failed to ping database")
	}

	return &dbConnPool{
		connPool: conn,
	}, nil
}

type dbConnPool struct {
	connPool *sql.DB
}

func (p *dbConnPool) Close() error {
	return p.connPool.Close()
}

func (p *dbConnPool) BeginTx(ctx context.Context, opts *sql.TxOptions) (Tx, error) {
	if opts == nil {
		opts = &sql.TxOptions{
			Isolation: sql.LevelDefault,
			ReadOnly:  false,
		}
	}

	txSql, err := p.connPool.BeginTx(ctx, opts)
	if err != nil {
		return nil, err
	}

	return &tx{txSql: txSql}, nil
}

func (p *dbConnPool) RunInTx(ctx context.Context, callback func(ctx context.Context, tx Tx) error) error {
	tx, err := p.BeginTx(ctx, &sql.TxOptions{
		Isolation: sql.LevelDefault,
		ReadOnly:  false,
	})
	if err != nil {
		return errors.Wrap(err, "failed to  begin transaction")
	}

	err = callback(ctx, tx)
	if err != nil {
		if rollbackErr := tx.Rollback(); rollbackErr != nil {
			return multierr.Combine(err, rollbackErr)
		}
		return err
	}

	if commitErr := tx.Commit(); commitErr != nil {
		return errors.Wrap(commitErr, "failed to  commit transaction")
	}

	return nil
}

func (p *dbConnPool) Exec(ctx context.Context, queryBuilder SQLBuilder) (sql.Result, error) {
	return exec(ctx, p.connPool, queryBuilder)
}

func (p *dbConnPool) Get(ctx context.Context, dest any, queryBuilder SQLBuilder) error {
	return get(ctx, p.connPool, dest, queryBuilder)
}

func (p *dbConnPool) GetOne(ctx context.Context, dest any, queryBuilder SQLBuilder) error {
	return getOne(ctx, p.connPool, dest, queryBuilder)
}

func (p *dbConnPool) ExecRaw(ctx context.Context, query string, args ...any) (sql.Result, error) {
	return execRaw(ctx, p.connPool, query, args...)
}
func (p *dbConnPool) GetRaw(ctx context.Context, dest any, query string, args ...any) error {
	return getRaw(ctx, p.connPool, dest, query, args...)
}
func (p *dbConnPool) GetOneRaw(ctx context.Context, dest any, query string, args ...any) error {
	return getOneRaw(ctx, p.connPool, dest, query, args...)
}