[英]How to avoid race conditions in GORM
我正在开发一个系统来启用具有增量队列号的患者注册。 我正在使用 Go、GORM 和 MySQL。
当多个患者同时注册时会出现问题,他们往往会获得相同的队列号,这是不应该发生的。
我尝试使用事务和钩子来实现这一点,但我仍然得到重复的队列号。 我还没有找到任何关于在事务发生时如何锁定数据库的资源。
func (r repository) CreatePatient(pat *model.Patient) error {
tx := r.db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
}
}()
err := tx.Error
if err != nil {
return err
}
// 1. get latest queue number and assign it to patient object
var queueNum int64
err = tx.Model(&model.Patient{}).Where("registration_id", pat.RegistrationID).Select("queue_number").Order("created_at desc").First(&queueNum).Error
if err != nil && err != gorm.ErrRecordNotFound {
tx.Rollback()
return err
}
pat.QueueNumber = queueNum + 1
// 2. write patient data into the db
err = tx.Create(pat).Error
if err != nil {
tx.Rollback()
return err
}
return tx.Commit().Error
}
正如@O所说。 琼斯,事务不会在这里保存您,因为您正在提取列的最大值,在数据库之外递增它,然后保存该新值。 从数据库的角度来看,更新的值不依赖于查询的值。
您可以尝试在单个查询中进行更新,这将使依赖性变得明显:
UPDATE patient AS p
JOIN (
SELECT max(queue_number) AS queue_number FROM patient WHERE registration_id = ?
) maxp
SET p.queue_number = maxp.queue_number + 1
WHERE id = ?
在 gorm 中,您无法运行这样的复杂更新,因此您需要使用Exec
。
我不是 100% 肯定上述方法会起作用,因为我不太熟悉 MySQL 事务隔离保证。
总的来说,保留一个队列表(通过reference_id)和一个你原子更新的计数器会更干净:
开始交易,然后
SELECT queue_number FROM queues WHERE registration_id = ? FOR UPDATE;
增加应用代码中的队列号,然后
UPDATE queues SET queue_number = ? WHERE registration_id = ?;
现在,您可以在事务提交之前在患者创建/更新中使用递增的队列号。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.