背景

近期看到一篇文章，真的感嘆作者的洞察力，在開發(fā)時有可能就會犯這樣的錯誤，所以一定要多學習，多實踐。其問題就是你在提交事務時，如果中間有其他業(yè)務就取消操作，那么事務也關閉了嗎？

事務實踐

服務端在進行和數(shù)據(jù)庫交互時，對于一些場景我們可能會使用事務來保證數(shù)據(jù)的冪等性。比如在一個更新的場景時基本操作流程時如下：

• 開啟數(shù)據(jù)庫事務
• 通過 ID 獲取數(shù)據(jù)記錄
• 確認是否可以進行更新操作
• 如果可以更新操作就更新記錄
• 提交事務
• 如果遇到錯誤，就回滾事務

在從數(shù)據(jù)庫中獲取數(shù)據(jù)時，可以通過鎖行的方式防止其他服務或者程序也對這條記錄進行操作，比如使用 select ... for update 方式獲取數(shù)據(jù)并鎖定該記錄。以下是簡單的使用事務操作數(shù)據(jù)的的方法：

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 中刪除數(shù)據(jù)后就完成提交事務，但是如果在期間如果發(fā)生問題就會返回error就會引發(fā) rollback 操作。

但還有一個需要注意的點，當獲取到的數(shù)據(jù)時，判斷到該記錄已經被刪除時，就會結束操作，但是結束操作卻沒有對事務進行釋放操作，所以就會造成一個很大的問題：數(shù)據(jù)量大的時候就會造成整個后續(xù)所有的請求都超時，導致所有的請求都不能完成操作。

tx.releaseConn(err)

可以看下事務實現(xiàn)的源碼，無論在 rollback 還是 commit 都會有 releaseConn 釋放連接，所以之前的例子中 defer 函數(shù)僅在出現(xiàn)錯誤的時才調用回滾，如果不提交也不回滾就會導致事務一直處于活躍的狀態(tài)，就會一直持有該事務，其請求再過來時達到最大值時就會造成事務超時。

優(yōu)化方案

解決問題有一個很簡單的的方案就是每個判斷 error 的條件下都進行回滾。也可以直接在 defer 函數(shù)改成回滾事務，提交事務后再執(zhí)行回滾也不會執(zhí)行任何操作。

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

但是沒有任何更改也進行提交，然后只有發(fā)生錯誤才進行回滾可能會影響代碼的可讀性。在開啟事務的方法中你會看到在調用 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()

所以我們可以直接使用取消上下文的方法，可以先創(chuàng)建一個新的取消上下文，如果沒有回滾或者提交時，最后執(zhí)行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()
        }
    }()

......
}

總結

所以在使用事務處理業(yè)務的時候，一定要注意業(yè)務邏輯，如果在業(yè)務邏輯中出現(xiàn)某些條件場景不進行操作數(shù)據(jù)庫時，結束這次業(yè)務處理時也要記得關閉事務。要么就關閉事務，要么就是無論有沒有處理都提交事務。這可能只是一個小小的問題，但是如果在交易的場景中如果沒有注意就可能造成很大的問題。

到此這篇關于一文帶你深入了解Go語言中的事務的文章就介紹到這了,更多相關Go語言事務內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

最新評論