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
}
Cảm ơn các bạn đã đọc. Có gì liên quan đến golang thì liên hệ mình:
Contact
- facebook: https://www.facebook.com/phucducdev/
- gmail: ducnp09081998@gmail.com
- linkedin: https://www.linkedin.com/in/phucducktpm/
- hashnode: https://hashnode.com/@OpenDev
- telegram: https://t.me/OpenDevGolang