feat(#8): add db transaction in repository layer

Author: chud-lori

.env

@@ -1,4 +1,4 @@
-DATABASE_URL=postgres://postgres:root@localhost/go_boilerplate?sslmode=disable
+DATABASE_URL=postgres://postgres:root@localhost/gosyah?sslmode=disable
 CACHE_URL=redis://localhost:6379
 LOGGER_LEVEL=debug
 CONTEXT_TIMEOUT=60

repository/pgsql/pgsql_todo.go

@@ -12,28 +12,66 @@ type pgsqlTodoRepository struct {
 	db *sql.DB
 }
 
-// NewPgsqlTodoRepository will create new an todoRepository object representation of TodoRepository interface
+// NewPgsqlTodoRepository will create a new todoRepository object representation of TodoRepository interface
 func NewPgsqlTodoRepository(db *sql.DB) *pgsqlTodoRepository {
 	return &pgsqlTodoRepository{
 		db: db,
 	}
 }
 
 func (r *pgsqlTodoRepository) Create(ctx context.Context, todo *domain.Todo) (err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		err = fmt.Errorf("failed to begin transaction: %v", err)
+		return
+	}
+	defer tx.Rollback()
+
 	query := "INSERT INTO todos (name, created_at, updated_at) VALUES ($1, $2, $3)"
-	_, err = r.db.ExecContext(ctx, query, todo.Name, todo.CreatedAt, todo.UpdatedAt)
+	_, err = tx.ExecContext(ctx, query, todo.Name, todo.CreatedAt, todo.UpdatedAt)
+
+	if err != nil {
+		err = fmt.Errorf("failed to insert todo: %v", err)
+		return
+	}
+
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }
 
 func (r *pgsqlTodoRepository) GetByID(ctx context.Context, id int64) (todo domain.Todo, err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		err = fmt.Errorf("failed to begin transaction: %v", err)
+		return
+	}
+	defer tx.Rollback()
+
 	query := "SELECT id, name, created_at, updated_at FROM todos WHERE id = $1"
-	err = r.db.QueryRowContext(ctx, query, id).Scan(&todo.ID, &todo.Name, &todo.CreatedAt, &todo.UpdatedAt)
+	err = tx.QueryRowContext(ctx, query, id).Scan(&todo.ID, &todo.Name, &todo.CreatedAt, &todo.UpdatedAt)
+
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }
 
 func (r *pgsqlTodoRepository) Fetch(ctx context.Context) (todos []domain.Todo, err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		err = fmt.Errorf("failed to begin transaction: %v", err)
+		return
+	}
+
 	query := "SELECT id, name, created_at, updated_at FROM todos"
-	rows, err := r.db.QueryContext(ctx, query)
+	rows, err := tx.QueryContext(ctx, query)
+
 	if err != nil {
 		return todos, err
 	}
@@ -50,12 +88,23 @@ func (r *pgsqlTodoRepository) Fetch(ctx context.Context) (todos []domain.Todo, e
 		todos = append(todos, todo)
 	}
 
+	if err = tx.Commit(); err != nil {
+		return todos, fmt.Errorf("failed to commit transaction: %v", err)
+	}
+
 	return todos, nil
 }
 
 func (r *pgsqlTodoRepository) Update(ctx context.Context, todo *domain.Todo) (err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		err = fmt.Errorf("failed to begin transaction: %v", err)
+		return
+	}
+	defer tx.Rollback()
+
 	query := "UPDATE todos SET name = $1, updated_at = $2 WHERE id = $3"
-	res, err := r.db.ExecContext(ctx, query, todo.Name, todo.UpdatedAt, todo.ID)
+	res, err := tx.ExecContext(ctx, query, todo.Name, todo.UpdatedAt, todo.ID)
 	if err != nil {
 		return
 	}
@@ -69,12 +118,24 @@ func (r *pgsqlTodoRepository) Update(ctx context.Context, todo *domain.Todo) (er
 		err = fmt.Errorf("weird behavior, total affected: %d", affect)
 	}
 
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }
 
 func (r *pgsqlTodoRepository) Delete(ctx context.Context, id int64) (err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		err = fmt.Errorf("failed to begin transaction: %v", err)
+		return
+	}
+	defer tx.Rollback()
+
 	query := "DELETE FROM todos WHERE id = $1"
-	res, err := r.db.ExecContext(ctx, query, id)
+	res, err := tx.ExecContext(ctx, query, id)
 	if err != nil {
 		return
 	}
@@ -88,5 +149,10 @@ func (r *pgsqlTodoRepository) Delete(ctx context.Context, id int64) (err error)
 		err = fmt.Errorf("weird behavior, total affected: %d", affect)
 	}
 
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }

repository/pgsql/pgsql_user.go

@@ -3,6 +3,7 @@ package pgsql
 import (
 	"context"
 	"database/sql"
+	"fmt"
 
 	"github.com/syahidfrd/go-boilerplate/domain"
 )
@@ -18,13 +19,37 @@ func NewPgsqlUserRepository(db *sql.DB) *pgsqlUserRepository {
 }
 
 func (r *pgsqlUserRepository) Create(ctx context.Context, user *domain.User) (err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		return
+	}
+	defer tx.Rollback()
+
 	query := "INSERT INTO users (email, password, created_at, updated_at) VALUES ($1, $2, $3, $4)"
-	_, err = r.db.ExecContext(ctx, query, user.Email, user.Password, user.CreatedAt, user.UpdatedAt)
+	_, err = tx.ExecContext(ctx, query, user.Email, user.Password, user.CreatedAt, user.UpdatedAt)
+
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }
 
 func (r *pgsqlUserRepository) GetByEmail(ctx context.Context, email string) (user domain.User, err error) {
+	tx, err := r.db.BeginTx(ctx, nil)
+	if err != nil {
+		return
+	}
+	defer tx.Rollback()
+
 	query := "SELECT id, email, password, created_at, updated_at FROM users WHERE email = $1"
-	err = r.db.QueryRowContext(ctx, query, email).Scan(&user.ID, &user.Email, &user.Password, &user.CreatedAt, &user.UpdatedAt)
+	err = tx.QueryRowContext(ctx, query, email).Scan(&user.ID, &user.Email, &user.Password, &user.CreatedAt, &user.UpdatedAt)
+
+	if err = tx.Commit(); err != nil {
+		err = fmt.Errorf("failed to commit transaction: %v", err)
+		return
+	}
+
 	return
 }