Go语言结合GORM实现数据库事务处理的示例代码

package main

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
)

type User struct {
	ID      uint   `gorm:"primaryKey"`
	Name    string `gorm:"size:100"`
	Balance int    // 账户余额
}

var db *gorm.DB

func initDB() {
	dsn := "root:123456@tcp(127.0.0.1:3306)/testdb?charset=utf8mb4&parseTime=True&loc=Local"
	var err error
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatal("数据库连接失败:", err)
	}
	_ = db.AutoMigrate(&User{})
}

// Transfer 转账函数：fromID -> toID，金额 amount
func Transfer(fromID, toID uint, amount int) error {
	return db.Transaction(func(tx *gorm.DB) error {
		var from, to User

		// 查询转出账户
		if err := tx.First(&from, fromID).Error; err != nil {
			return err
		}
		if from.Balance < amount {
			return fmt.Errorf("余额不足")
		}

		// 扣减余额
		if err := tx.Model(&from).Update("balance", from.Balance-amount).Error; err != nil {
			return err
		}

		// 查询转入账户
		if err := tx.First(&to, toID).Error; err != nil {
			return err
		}

		// 增加余额
		if err := tx.Model(&to).Update("balance", to.Balance+amount).Error; err != nil {
			return err
		}

		// 所有操作成功 -> 提交事务
		return nil
	})
}

func main() {
	initDB()

	// 初始化两位用户
	db.Create(&User{Name: "Alice", Balance: 100})
	db.Create(&User{Name: "Bob", Balance: 50})

	// 转账：Alice -> Bob 30元
	err := Transfer(1, 2, 30)
	if err != nil {
		log.Println("转账失败:", err)
	} else {
		log.Println("转账成功")
	}

	var users []User
	db.Find(&users)
	log.Println("当前用户余额:", users)
}