-
Notifications
You must be signed in to change notification settings - Fork 0
/
store.go
108 lines (86 loc) · 2.68 KB
/
store.go
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package store
import (
"context"
"time"
sq "github.com/Masterminds/squirrel"
"github.com/georgysavva/scany/v2/pgxscan"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgconn"
"github.com/jackc/pgx/v5/pgxpool"
"github.com/l-orlov/slim-fairy/bot/internal/config"
"github.com/pkg/errors"
)
const (
//defaultTimeout = 3 * time.Second
defaultTimeout = 30 * time.Minute // TODO: fix
)
type Storage struct {
pool *pgxpool.Pool
}
func New(ctx context.Context) (*Storage, error) {
pgDSN := config.Get().PgDSN
pool, err := pgxpool.New(ctx, pgDSN)
if err != nil {
return nil, errors.Wrap(err, "pgxpool.New")
}
return &Storage{
pool: pool,
}, nil
}
func (s *Storage) Close() {
s.pool.Close()
}
// Postgres specific squirrel builder
func psql() sq.StatementBuilderType {
return sq.StatementBuilder.PlaceholderFormat(sq.Dollar)
}
// Select executes select query
func Select(ctx context.Context, db pgxscan.Querier, dst interface{}, query string, args ...interface{}) error {
dbCtx, cancel := context.WithTimeout(ctx, defaultTimeout)
defer cancel()
return pgxscan.Select(dbCtx, db, dst, query, args...)
}
// Selectx executes select query with squirrel.Sqlizer
func Selectx(ctx context.Context, db pgxscan.Querier, dst interface{}, sqlizer sq.Sqlizer) error {
stmt, args, err := sqlizer.ToSql()
if err != nil {
return err
}
return Select(ctx, db, dst, stmt, args...)
}
// Querier can execute sql query and get the pgx.Rows
type Querier interface {
Query(ctx context.Context, query string, args ...interface{}) (pgx.Rows, error)
}
// Get executes get query
func Get(ctx context.Context, db Querier, dst interface{}, query string, args ...interface{}) error {
dbCtx, cancel := context.WithTimeout(ctx, defaultTimeout)
defer cancel()
return pgxscan.Get(dbCtx, db, dst, query, args...)
}
// Getx executes get query with squirrel.Sqlizer
func Getx(ctx context.Context, db Querier, dst interface{}, sqlizer sq.Sqlizer) error {
stmt, args, err := sqlizer.ToSql()
if err != nil {
return err
}
return Get(ctx, db, dst, stmt, args...)
}
// Executor can execute sql query
type Executor interface {
Exec(ctx context.Context, sql string, arguments ...any) (pgconn.CommandTag, error)
}
// Exec executes query
func Exec(ctx context.Context, db Executor, query string, args ...interface{}) (pgconn.CommandTag, error) {
dbCtx, cancel := context.WithTimeout(ctx, defaultTimeout)
defer cancel()
return db.Exec(dbCtx, query, args...)
}
// Execx executes query with squirrel.Sqlizer
func Execx(ctx context.Context, db Executor, sqlizer sq.Sqlizer) (pgconn.CommandTag, error) {
stmt, args, err := sqlizer.ToSql()
if err != nil {
return pgconn.CommandTag{}, err
}
return Exec(ctx, db, stmt, args...)
}