feat: add features

This commit is contained in:
Rogee
2024-11-29 20:28:37 +08:00
parent 3d7ee40a7a
commit d1a2a80859
8 changed files with 337 additions and 22 deletions

17
backend/pkg/db/db.go Normal file
View File

@@ -0,0 +1,17 @@
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
}