gorm-V2-08 事务

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
}