Go 语言实现 MySQL 数据库事务
Go 实现 MySQL 数据库事务
一、MySQL事务
MySQL事务是指一组数据库操作，它们被视为一个逻辑单元，并且要么全部成功执行，要么全部回滚（撤销）。事务是数据库管理系统提供的一种机制，用于确保数据的一致性和完整性。
事务具有以下特性（通常由ACID原则定义）：
- 原子性（Atomicity）：事务中的所有操作要么全部成功执行，要么全部回滚，不存在部分执行的情况。如果事务中的任何一个操作失败，则所有操作都会被回滚到事务开始之前的状态，保持数据的一致性。
- 一致性（Consistency）：事务的执行使数据库从一个一致的状态转换到另一个一致的状态。这意味着在事务开始和结束时，数据必须满足预定义的完整性约束。
- 隔离性（Isolation）：事务的执行是相互隔离的，即一个事务的操作在提交之前对其他事务是不可见的。并发事务之间的相互影响被隔离，以避免数据损坏和不一致的结果。
- 持久性（Durability）：一旦事务提交成功，其对数据库的更改将永久保存，即使在系统故障或重启之后也能保持数据的持久性。
在MySQL中，使用以下语句来开始一个事务：
START TRANSACTION;
在事务中，可以执行一系列的数据库操作，如插入、更新和删除等。最后，使用以下语句来提交事务或回滚事务：
提交事务：
COMMIT;
回滚事务：
ROLLBACK;
通过使用事务，可以确保数据库操作的一致性和完整性，尤其在处理涉及多个相关操作的复杂业务逻辑时非常有用。
二、MySQL 事务 示例
以下是一个示例，演示如何在MySQL中使用事务：
START TRANSACTION;
-- 在事务中执行一系列数据库操作
INSERT INTO users (name, age) VALUES ('Alice', 25);
UPDATE accounts SET balance = balance - 100 WHERE user_id = 1;
DELETE FROM logs WHERE user_id = 1;
-- 如果一切正常，提交事务
COMMIT;
在上述示例中，我们开始了一个事务，并在事务中执行了一系列数据库操作。
首先，我们向users表插入了一条新记录，
然后更新了accounts表中用户ID为1的账户余额，
最后删除了logs表中与用户ID为1相关的日志条目。
如果在事务执行的过程中出现任何错误或异常情况，可以使用ROLLBACK语句回滚事务，使所有操作都被撤销，数据库恢复到事务开始之前的状态：
START TRANSACTION;
-- 在事务中执行一系列数据库操作
INSERT INTO users (name, age) VALUES ('Bob', 30);
UPDATE accounts SET balance = balance - 200 WHERE user_id = 2;
-- 发生错误或异常，回滚事务
ROLLBACK;
在上述示例中，如果在更新accounts表的操作中发生错误，整个事务将被回滚，插入的用户记录和更新的账户余额将被撤销。
事务的关键在于将多个相关的数据库操作组织在一起，并以原子性和一致性的方式进行提交或回滚。这确保了数据的完整性和一致性，同时也提供了灵活性和错误恢复机制。
三、MySQL 事务引擎
MySQL提供了多个事务引擎，每个引擎都具有不同的特性和适用场景。以下是MySQL中常见的事务引擎：
- InnoDB：InnoDB是MySQL默认的事务引擎。它支持事务、行级锁定、外键约束和崩溃恢复等功能。InnoDB适用于需要强调数据完整性和并发性能的应用程序。
- MyISAM：MyISAM是MySQL的另一个常见的事务引擎。它不支持事务和行级锁定，但具有较高的插入和查询性能。MyISAM适用于读密集型应用程序，例如日志记录和全文搜索。
- NDB Cluster：NDB Cluster是MySQL的集群事务引擎，适用于需要高可用性和可扩展性的分布式应用程序。它具有自动分片、数据冗余和故障恢复等功能。
- Memory：Memory（也称为Heap）引擎将表数据存储在内存中，提供非常高的插入和查询性能。但由于数据存储在内存中，因此在数据库重新启动时数据会丢失。Memory引擎适用于临时数据或缓存数据的存储。
除了以上列出的常见事务引擎之外，MySQL还支持其他一些事务引擎，例如Archive、Blackhole等。每个引擎都有其独特的特性和适用场景，选择合适的事务引擎需要根据应用程序的需求和性能要求进行评估。
在创建表时，可以指定所需的事务引擎。例如，使用以下语句创建一个使用InnoDB引擎的表：
CREATE TABLE mytable (
id INT PRIMARY KEY,
name VARCHAR(50)
) ENGINE=InnoDB;
需要注意的是，不同的事务引擎可能会有不同的配置和限制，因此在选择和使用特定的事务引擎时，建议参考MySQL文档以了解详细信息和最佳实践。
四、事务实例
开启事务 Begin 源码：
// BeginTx starts a transaction.
//
// The provided context is used until the transaction is committed or rolled back.
// If the context is canceled, the sql package will roll back
// the transaction. Tx.Commit will return an error if the context provided to
// BeginTx is canceled.
//
// The provided TxOptions is optional and may be nil if defaults should be used.
// If a non-default isolation level is used that the driver doesn't support,
// an error will be returned.
func (db *DB) BeginTx(ctx context.Context, opts *TxOptions) (*Tx, error) {
var tx *Tx
var err error
err = db.retry(func(strategy connReuseStrategy) error {
tx, err = db.begin(ctx, opts, strategy)
return err
})
return tx, err
}
// Begin starts a transaction. The default isolation level is dependent on
// the driver.
//
// Begin uses context.Background internally; to specify the context, use
// BeginTx.
func (db *DB) Begin() (*Tx, error) {
return db.BeginTx(context.Background(), nil)
}
中止事务 Rollback 源码：
// rollback aborts the transaction and optionally forces the pool to discard
// the connection.
func (tx *Tx) rollback(discardConn bool) error {
if !tx.done.CompareAndSwap(false, true) {
return ErrTxDone
}
if rollbackHook != nil {
rollbackHook()
}
// Cancel the Tx to release any active R-closemu locks.
// This is safe to do because tx.done has already transitioned
// from 0 to 1. Hold the W-closemu lock prior to rollback
// to ensure no other connection has an active query.
tx.cancel()
tx.closemu.Lock()
tx.closemu.Unlock()
var err error
withLock(tx.dc, func() {
err = tx.txi.Rollback()
})
if !errors.Is(err, driver.ErrBadConn) {
tx.closePrepared()
}
if discardConn {
err = driver.ErrBadConn
}
tx.close(err)
return err
}
// Rollback aborts the transaction.
func (tx *Tx) Rollback() error {
return tx.rollback(false)
}
提交事务 Commit 源码：
// Commit commits the transaction.
func (tx *Tx) Commit() error {
// Check context first to avoid transaction leak.
// If put it behind tx.done CompareAndSwap statement, we can't ensure
// the consistency between tx.done and the real COMMIT operation.
select {
default:
case <-tx.ctx.Done():
if tx.done.Load() {
return ErrTxDone
}
return tx.ctx.Err()
}
if !tx.done.CompareAndSwap(false, true) {
return ErrTxDone
}
// Cancel the Tx to release any active R-closemu locks.
// This is safe to do because tx.done has already transitioned
// from 0 to 1. Hold the W-closemu lock prior to rollback
// to ensure no other connection has an active query.
tx.cancel()
tx.closemu.Lock()
tx.closemu.Unlock()
var err error
withLock(tx.dc, func() {
err = tx.txi.Commit()
})
if !errors.Is(err, driver.ErrBadConn) {
tx.closePrepared()
}
tx.close(err)
return err
}
例子
package main
import (
"database/sql"
"fmt"
"time"
_ "github.com/go-sql-driver/mysql" // 匿名导入 自动执行 init()
)
var db *sql.DB
func initMySQL() (err error) {
//DSN (Data Source Name)
dsn := "root:12345678@tcp(127.0.0.1:3306)/sql_test"
// 注意：要初始化全局的 db 对象，不要新声明一个 db 变量
db, err = sql.Open("mysql", dsn) // 只对格式进行校验，并不会真正连接数据库
if err != nil {
return err
}
// Ping 验证与数据库的连接是否仍处于活动状态，并在必要时建立连接。
err = db.Ping()
if err != nil {
fmt.Printf("connect to db failed, err: %v\n", err)
return err
}
// 数值需要根据业务具体情况来确定
db.SetConnMaxLifetime(time.Second * 10) // 设置可以重用连接的最长时间
db.SetConnMaxIdleTime(time.Second * 5) // 设置连接可能处于空闲状态的最长时间
db.SetMaxOpenConns(200) // 设置与数据库的最大打开连接数
db.SetMaxIdleConns(10) // 设置空闲连接池中的最大连接数
return nil
}
type user struct {
id int
age int
name string
}
// 事务操作
func transactionDemo() {
// 启动事务。默认隔离级别取决于驱动程序。
tx, err := db.Begin() // 开启事务
if err != nil {
if tx != nil {
tx.Rollback() // 回滚 中止事务
}
fmt.Printf("begin trans failed, err:%v\n", err)
return
}
sqlStr1 := "UPDATE user SET age=? WHERE id=?"
ret1, err := tx.Exec(sqlStr1, 22, 2)
if err != nil {
tx.Rollback() // 回滚
fmt.Printf("exec sql1 failed, err: %v\n", err)
return
}
// RowsAffected 返回受更新、插入或删除影响的行数。并非每个数据库或数据库驱动程序都支持此功能。
affRow1, err := ret1.RowsAffected()
if err != nil {
tx.Rollback() // 回滚
fmt.Printf("exec ret1.RowsAffected() failed, err: %v\n", err)
return
}
sqlStr2 := "UPDATE user SET age=? WHERE id=?"
ret2, err := tx.Exec(sqlStr2, 100, 5)
if err != nil {
tx.Rollback() // 回滚
fmt.Printf("exec sql2 failed, err: %v\n", err)
return
}
affRow2, err := ret2.RowsAffected()
if err != nil {
tx.Rollback() // 回滚
fmt.Printf("exec ret2.RowsAffected() failed, err: %v\n", err)
return
}
fmt.Println(affRow1, affRow2)
if affRow1 == 1 && affRow2 == 1 {
fmt.Println("事务提交...")
tx.Commit() // 提交事务
} else {
tx.Rollback()
fmt.Println("事务回滚...")
}
fmt.Println("exec trans success!")
}
func main() {
if err := initMySQL(); err != nil {
fmt.Printf("connect to db failed, err: %v\n", err)
}
// 检查完错误之后执行，确保 db 不为 nil
// Close() 用来释放数据库连接相关的资源
// Close 将关闭数据库并阻止启动新查询。关闭，然后等待服务器上已开始处理的所有查询完成。
defer db.Close()
fmt.Println("connect to database success")
// db.xx() 去使用数据库操作...
// 事务
transactionDemo()
}
运行
Code/go/mysql_demo via 🐹 v1.20.3 via 🅒 base
➜ go run main.go
connect to database success
1 0
事务回滚...
exec trans success!
Code/go/mysql_demo via 🐹 v1.20.3 via 🅒 base
➜ go run main.go
connect to database success
1 1
事务提交...
exec trans success!
Code/go/mysql_demo via 🐹 v1.20.3 via 🅒 base
➜
SQL 查询结果
mysql> select * from user;
+----+--------+------+
| id | name | age |
+----+--------+------+
| 1 | 小乔 | 16 |
| 2 | 小乔 | 12 |
| 5 | 昭君 | 12 |
| 6 | 黛玉 | 16 |
+----+--------+------+
4 rows in set (0.00 sec)
mysql> select * from user; # 第一次执行后查询
+----+--------+------+
| id | name | age |
+----+--------+------+
| 1 | 小乔 | 16 |
| 2 | 小乔 | 12 |
| 5 | 昭君 | 12 |
| 6 | 黛玉 | 16 |
+----+--------+------+
4 rows in set (0.00 sec)
mysql> select * from user; # 第二次执行后查询
+----+--------+------+
| id | name | age |
+----+--------+------+
| 1 | 小乔 | 16 |
| 2 | 小乔 | 22 |
| 5 | 昭君 | 100 |
| 6 | 黛玉 | 16 |
+----+--------+------+
4 rows in set (0.00 sec)
mysql>
在上例中，第一次执行SQL更新中的sqlStr2值为：age = 100， id = 3。因没有id为3 的数据，故事务回滚了，即使sqlStr1执行成功，也会回滚到事务开始的状态。所以第一次执行后查询数据库，数据未发生改变。第二次执行SQL更新中的sqlStr2值为：age = 100， id = 5。数据库中存在 id 为 5 的数据，事务提交成功，SQL语句执行成功。故第二次执行后查询数据发生改变，更新成功。
本文来自博客园，作者：寻月隐君，转载请注明原文链接：https://www.cnblogs.com/QiaoPengjun/p/17473494.html