如何使用 GORM 确保数据库操作的并发安全？

2024-01-07 17:05:48

GORM 是一款流行的 Golang 对象关系映射器（ORM），它可以帮助我们轻松地操作数据库。但是，GORM 并没有承诺并发安全，这意味着如果我们不注意的话，可能会出现并发安全问题。本文将介绍如何使用 GORM 确保数据库操作的并发安全。

正文

GORM 提供了一套符合使用习惯的并发范式，来兼顾性能和并发安全。但是，如果我们的使用习惯不符合 GORM 预想的习惯，则可能会出现并发安全问题。

为了避免并发安全问题，我们可以采取以下措施：

使用事务：事务可以确保一组数据库操作要么全部成功，要么全部失败。在 GORM 中，我们可以使用 Transaction() 方法来开启一个事务。例如：

func main() {
	db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/gorm_test?charset=utf8mb4&parseTime=True&loc=Local")
	if err != nil {
		log.Fatal(err)
	}

	err = db.Transaction(func(tx *gorm.DB) error {
		// 在事务中执行数据库操作
		if err := tx.Create(&User{Name: "John Doe"}).Error; err != nil {
			return err
		}
		if err := tx.Create(&User{Name: "Jane Doe"}).Error; err != nil {
			return err
		}

		return nil
	})
	if err != nil {
		log.Fatal(err)
	}
}

使用乐观锁：乐观锁是一种并发控制技术，它假设在并发操作期间，数据不会被其他事务修改。在 GORM 中，我们可以使用 Set() 方法来设置乐观锁字段。例如：

func main() {
	db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/gorm_test?charset=utf8mb4&parseTime=True&loc=Local")
	if err != nil {
		log.Fatal(err)
	}

	user := User{Name: "John Doe"}
	if err := db.Create(&user).Error; err != nil {
		log.Fatal(err)
	}

	// 在另一个事务中修改 user 的 Name 字段
	if err := db.Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(&user).Set("name", "Jane Doe").Updates(&user).Error; err != nil {
			return err
		}

		return nil
	}); err != nil {
		log.Fatal(err)
	}

	// 尝试在第一个事务中再次修改 user 的 Name 字段
	if err := db.Model(&user).Set("name", "John Doe").Updates(&user).Error; err != nil {
		log.Fatal(err)
	}
}

使用悲观锁：悲观锁是一种并发控制技术，它假设在并发操作期间，数据可能会被其他事务修改。在 GORM 中，我们可以使用 Lock() 方法来设置悲观锁。例如：

func main() {
	db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/gorm_test?charset=utf8mb4&parseTime=True&loc=Local")
	if err != nil {
		log.Fatal(err)
	}

	user := User{Name: "John Doe"}
	if err := db.Create(&user).Error; err != nil {
		log.Fatal(err)
	}

	// 在另一个事务中修改 user 的 Name 字段
	if err := db.Transaction(func(tx *gorm.DB) error {
		if err := tx.Model(&user).Lock("UPDATE").Updates(&user).Error; err != nil {
			return err
		}

		return nil
	}); err != nil {
		log.Fatal(err)
	}

	// 尝试在第一个事务中再次修改 user 的 Name 字段
	if err := db.Model(&user).Updates(&user).Error; err != nil {
		log.Fatal(err)
	}
}