zishuo/nunu-layout-advanced
1
0
Fork 0
mirror of https://github.com/go-nunu/nunu-layout-advanced.git synced 2026-04-22 15:37:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #22 from RenaLio/main

Browse Source 
fix(repository): use r.DB(ctx).Transaction to preserve tx context and nested tx semantics
This commit is contained in:
Chris
2026-03-20 16:14:34 +08:00
committed by GitHub
parent 1ffe9e4a84 6ce3545fcd
commit 64ef73f0bc
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/repository/repository.go
+3 -2
View File
@@ -3,6 +3,8 @@ package repository
import (
"context"
"fmt"
"time"
"github.com/glebarez/sqlite"
"github.com/go-nunu/nunu-layout-advanced/pkg/log"
"github.com/go-nunu/nunu-layout-advanced/pkg/zapgorm2"
@@ -13,7 +15,6 @@ import (
"gorm.io/driver/mysql"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"time"
)
const ctxTxKey = "TxKey"
@@ -61,7 +62,7 @@ func (r *Repository) DB(ctx context.Context) *gorm.DB {
}
func (r *Repository) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
return r.DB(ctx).Transaction(func(tx *gorm.DB) error {
ctx = context.WithValue(ctx, ctxTxKey, tx)
return fn(ctx)
})