feat(pkg): add db transaction

Author: AfandyW

app/users/repository/get_users.go

@@ -8,7 +8,7 @@ import (
 )
 
 func (repo *repository) GetUsers(ctx context.Context) (users []domain.User, err error) {
-	err = repo.db.Find(&users).Error
+	err = repo.db.DB(ctx).Find(&users).Error
 	if err != nil {
 		logrus.Error("repo.GetUsers: failed to get users")
 		return

app/users/repository/repository.go

@@ -2,16 +2,16 @@ package repository
 
 import (
 	"github.com/hammer-code/lms-be/app/users"
-	"gorm.io/gorm"
+	pkgDB "github.com/hammer-code/lms-be/pkg/db"
 )
 
 type (
 	repository struct {
-		db *gorm.DB
+		db pkgDB.DatabaseTransaction
 	}
 )
 
-func NewRepository(db *gorm.DB) users.UserRepository {
+func NewRepository(db pkgDB.DatabaseTransaction) users.UserRepository {
 	return &repository{
 		db,
 	}

cmd/serve_http.go

@@ -14,6 +14,7 @@ import (
 	users_usecase "github.com/hammer-code/lms-be/app/users/usecase"
 	"github.com/hammer-code/lms-be/config"
 	"github.com/hammer-code/lms-be/domain"
+	pkgDB "github.com/hammer-code/lms-be/pkg/db"
 	"github.com/hammer-code/lms-be/utils"
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
@@ -39,11 +40,13 @@ var serveHttpCmd = &cobra.Command{
 				DSN: cfg.DB_POSTGRES_DSN,
 			}})
 
+		dbTx := pkgDB.NewDBTransaction(db)
+
 		// repository
-		userRepo := users_repo.NewRepository(db)
+		userRepo := users_repo.NewRepository(dbTx)
 
 		// usecase
-		userUsecase := users_usecase.NewUsecase(userRepo)
+		userUsecase := users_usecase.NewUsecase(userRepo, dbTx)
 
 		// handler
 		userHandler := users_handler.NewHandler(userUsecase)

domain/context.go

@@ -0,0 +1,5 @@
+package domain
+
+const (
+	ContextDatabaseTransaction = "database-transaction"
+)

pkg/db/db.go

@@ -13,6 +13,7 @@ type (
 
 	DatabaseTransaction interface {
 		StartTransaction(ctx context.Context, fn func(txCtx context.Context) error) error
+		DB(ctx context.Context) *gorm.DB
 	}
 )
 

pkg/db/get_database_transaction.go

@@ -0,0 +1,17 @@
+package db
+
+import (
+	"context"
+
+	"github.com/hammer-code/lms-be/domain"
+	"gorm.io/gorm"
+)
+
+// get db transaction or repo db(without transaction)
+func (d *dbTX) DB(ctx context.Context) *gorm.DB {
+	tx, ok := ctx.Value(domain.ContextDatabaseTransaction).(*gorm.DB)
+	if !ok {
+		return d.db
+	}
+	return tx
+}

pkg/db/start_transaction.go

@@ -3,6 +3,7 @@ package db
 import (
 	"context"
 
+	"github.com/hammer-code/lms-be/domain"
 	"github.com/sirupsen/logrus"
 )
 
@@ -11,7 +12,9 @@ func (d *dbTX) StartTransaction(ctx context.Context, fn func(txCtx context.Conte
 
 	defer tx.Rollback()
 
-	if err := fn(ctx); err != nil {
+	txCtx := context.WithValue(ctx, domain.ContextDatabaseTransaction, tx)
+
+	if err := fn(txCtx); err != nil {
 		logrus.Error("transaction database return err: ", err)
 		return err
 	}