18 lines
254 B
Go
18 lines
254 B
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"backend/common/consts"
|
|
|
|
"github.com/go-jet/jet/v2/qrm"
|
|
)
|
|
|
|
func FromContext(ctx context.Context, db *sql.DB) qrm.DB {
|
|
if tx, ok := ctx.Value(consts.CtxKeyTx).(*sql.Tx); ok {
|
|
return tx
|
|
}
|
|
return db
|
|
}
|