文章目录

• 1. 语法
• 2. 完整示例

1. 语法

• 开启事务

tx := db.Begin()

• 在事务中执行操作

从这里开始使用 ‘tx’，而不是 ‘db’

tx.Create(...)

400

• 回滚事务

tx.Rollback()

• 提交这个事务

tx.Commit()

2. 完整示例

package main

import (
	"database/sql"
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"time"
)

type User struct {
	ID int64
	Age int64
	Name string
	Email string
	Company string
	Deleted gorm.DeletedAt
}


func main() {
	db,sqlDB,_ := connect()
	defer sqlDB.Close()
	CreateAnimals(db)

}

func CreateAnimals(db *gorm.DB) (err error) {
	// 注意在事务中要使用 tx 作为数据库句柄
	tx := db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	if err := tx.Error;err != nil {
		return err
	}

	if err := tx.Create(&User{Name: "ZhaoYun",Age: 18}).Error; err != nil {
		tx.Rollback()
		return err
	}

	if err := tx.Create(&User{Name: "MaChao",Age: 20}).Error; err != nil {
		tx.Rollback()
		return err
	}

	return tx.Commit().Error
}

func connect() (db *gorm.DB,sqlDB *sql.DB,err error) {
	dsn := "root:40010355@tcp(127.0.0.1:3306)/crow?charset=utf8&parseTime=True&loc=Local"
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	sqlDB,_ = db.DB()
	if err != nil {
		fmt.Printf(err.Error())
		defer sqlDB.Close()
	}else {
		fmt.Printf("OK\n")
		sqlDB.SetMaxIdleConns(10)
		sqlDB.SetMaxOpenConns(100)
		sqlDB.SetConnMaxLifetime(time.Hour)
	}
	return
}