[Golang] Lưu Ý Khi Dùng Transaction Database

Mình hy vọng sau khi đọc bài viết này, bạn có thể tránh được những lỗi tương tự trong dự án của chính mình. Mặc dù chỉ là một lỗi rất nhỏ nhưng nó có thể khiến bạn sẽ phải hối hận.

Chúng ta sẽ đối mặt là một transaction database đơn giản. Và hầu như tất cả các develop điều biết transaction , nếu vẫn thắc mắc về nó thì kham khảo here.

Trong service sẽ làm chức năng quản lý user trong database, get user bằng id sau đó cập nhật full-name

Thứ tự các bước như sau:

bắt đầu 1 transaction
lấy thông tin user bằng id
Xác nhận user vẫn còn trạng thái active
nếu hợp lệ thì cập nhật fullname
commit transaction
nếu không hợp lệ hoặc error thì revert transaction

| Mình sẽ dùng gorm để truy cập database

đoạn code repository get user bằng id:

func (r *userService) GetUserById(tx *gorm.DB, id int) (model.User, error) {
	var (
		user model.User
		err  error
	)
	err = tx.Table(user.Table()).Where("id = ?", id).First(&user).Error
	if err != nil {
		return user, err
	}
	return user, nil
}

code logic:

func (s *service) UpdateUserId(ctx context.Context, id int) (error) {
    
	tx := r.gormDB.Begin()

    defer func() {
        if err != nil {
            tx.Rollback()
        }
    }()
    
    user, err := s.userService.GetUserById(tx, id)
    if err != nil {
        return  err
    }

    if user.Status != "active" {
        return  nil
    }
	
    user.UpdatedAt = time.Now()

    err = s.userService.UpdateName(tx, name)
    if err != nil {
        return  err
    }

    err = tx.Commit()
    if err != nil {
        return  err
    }

    return nil
}

Vấn Đề

tất cả các request từ client sẽ bị time-out, database sẽ thêm rất nhiều connection vào hệ thống.
không có bất cứ 1 request nào có thể hoàn thành và kết thúc.

Tại Sao vấn đề xảy ra đó là khi user có status khác active điều này có nghĩa là sẽ return nil và transaction sẽ không release lock trên record của user đó.

mặc dù trong function defer có gọi rollback() nhưng chỉ xảy ra khi có error, user khác active sẽ return nil và record sẽ bị lock cho đến khi transaction bị timeout.

Fix Để fix vấn đề trên là cực kì đơn giản, thêm rollback().

if user.Status != "active" {
    tx.Rollback()
    return  nil
}

còn nếu bạn vẫn muốn rollback() trong defer thì sẽ return error.

var ErrUserNotActive = errors.New("user not active")
func (s *service) UpdateUserId(ctx context.Context, id int) (error) {
    
	tx := r.gormDB.Begin()

    defer func() {
        if err != nil {
            tx.Rollback()
        }
    }()
    
    user, err := s.userService.GetUserById(tx, id)
    if err != nil {
        return  err
    }

    if user.Status != "active" {
        return ErrUserNotActive
    }
	
    user.UpdatedAt = time.Now()

    err = s.userService.UpdateName(tx, name)
    if err != nil {
        return  err
    }

    err = tx.Commit()
    if err != nil {
        return  err
    }

    return nil
}