mirror/phylum
1
0
Fork 0
mirror of https://codeberg.org/shroff/phylum.git synced 2026-01-18 01:50:57 -06:00
Code Issues Packages Projects Releases Wiki Activity

[server] Fix nested transactions

Browse Source
This commit is contained in:
Abhishek Shroff
2024-08-13 22:53:23 +05:30
parent 9f8ba2211f
commit f16dfb4e07
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/internal/db/db_handler.go
View File

@@ -22,6 +22,9 @@ var latestSchemaVersion int
type DbHandler struct {
Queries
tx interface {
Begin(context.Context) (pgx.Tx, error)
}
pool *pgxpool.Pool
}
@@ -59,6 +62,7 @@ func NewDb(ctx context.Context, dsn string, trace bool) (*DbHandler, error) {
handler := &DbHandler{
Queries: *New(pool),
tx: pool,
pool: pool,
}
@@ -66,9 +70,10 @@ func NewDb(ctx context.Context, dsn string, trace bool) (*DbHandler, error) {
}
func (d DbHandler) WithTx(ctx context.Context, fn func(*DbHandler) error) error {
return pgx.BeginFunc(ctx, d.pool, func(tx pgx.Tx) error {
return pgx.BeginFunc(ctx, d.tx, func(tx pgx.Tx) error {
d := DbHandler{
Queries: *d.Queries.WithTx(tx),
tx: tx,
pool: d.pool,
}
return fn(&d)