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
}