一文带你深入了解Go语言中的事务

2023-05-16 14:05:37 事务 带你 一文

背景

近期看到一篇文章,真的感叹作者的洞察力,在开发时有可能就会犯这样的错误,所以一定要多学习,多实践。其问题就是你在提交事务时,如果中间有其他业务就取消操作,那么事务也关闭了吗?

事务实践

服务端在进行和数据库交互时,对于一些场景我们可能会使用事务来保证数据的幂等性。比如在一个更新的场景时基本操作流程时如下:

  • 开启数据库事务
  • 通过 ID 获取数据记录
  • 确认是否可以进行更新操作
  • 如果可以更新操作就更新记录
  • 提交事务
  • 如果遇到错误,就回滚事务

在从数据库中获取数据时,可以通过行的方式防止其他服务或者程序也对这条记录进行操作,比如使用 select ... for update 方式获取数据并锁定该记录。以下是简单的使用事务操作数据的的方法:

func (user *UserResp) DeleteUser(ctx context.Context, id string) error {
	tx, err := user.db.BeginTx(ctx, nil)
	if err != nil {
		return err
	}

	defer func() {
		if err != nil {
			tx.Rollback()
		}
	}()
	result, err := user.handler.getById(id)
	if err != nil {
		return err
	}
	if result.IsDeleted {
		return nil
	}
	if err = user.handler.Delete(id); err != nil {
		return err
	}
	if err = tx.Commit(); err != nil {
		return err
	}
	return nil
}

事务说明

从上面的源码整体看起来没什么问题。在进行相关的操作时只要正常删除从db 中删除数据后就完成提交事务,但是如果在期间如果发生问题就会返回error就会引发 rollback 操作。

但还有一个需要注意的点,当获取到的数据时,判断到该记录已经被删除时,就会结束操作,但是结束操作却没有对事务进行释放操作,所以就会造成一个很大的问题:数据量大的时候就会造成整个后续所有的请求都超时,导致所有的请求都不能完成操作。

tx.releaseConn(err)

可以看下事务实现的源码,无论在 rollback 还是 commit 都会有 releaseConn 释放连接,所以之前的例子中 defer 函数仅在出现错误的时才调用回滚,如果不提交也不回滚就会导致事务一直处于活跃的状态,就会一直持有该事务,其请求再过来时达到最大值时就会造成事务超时。

优化方案

解决问题有一个很简单的的方案就是每个判断 error 的条件下都进行回滚。也可以直接在 defer 函数改成回滚事务,提交事务后再执行回滚也不会执行任何操作。

	defer func() {
			tx.Rollback()
	}()

但是没有任何更改也进行提交,然后只有发生错误才进行回滚可能会影响代码的可读性。在开启事务的方法中你会看到在调用 beginDC 的方法中有使用 context 服务上下文进行回滚事务。所以还有一个方案就是通过取消上下文来让事务结束从而释放锁。

// 方法 beginDC 中的代码片段

ctx, cancel := context.WithCancel(ctx)
	tx = &Tx{
		db:                 db,
		dc:                 dc,
		releaseConn:        release,
		txi:                txi,
		cancel:             cancel,
		keepConnOnRollback: keepConnOnRollback,
		ctx:                ctx,
	}
	Go tx.awaitDone()

所以我们可以直接使用取消上下文的方法,可以先创建一个新的取消上下文,如果没有回滚或者提交时,最后执行cancel 就会通知事务已完成,然后就会关闭事务。

func (user *UserResp) DeleteUser(ctx context.Context, id string) error {
ctx, cancel := context.WithCancel(ctx)
    defer cancel()
    
    tx, err := s.db.BeginTxx(ctx, nil)
    if err != nil {
        return nil, err
    }
	
    defer func() {
        if err != nil {
            tx.Rollback()
        }
    }()

......
}

总结

所以在使用事务处理业务的时候,一定要注意业务逻辑,如果在业务逻辑中出现某些条件场景不进行操作数据库时,结束这次业务处理时也要记得关闭事务。要么就关闭事务,要么就是无论有没有处理都提交事务。这可能只是一个小小的问题,但是如果在交易的场景中如果没有注意就可能造成很大的问题。

到此这篇关于一文带你深入了解Go语言中的事务的文章就介绍到这了,更多相关Go语言事务内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

相关文章