Go語言中的事務怎么使用

本篇內容主要講解“Go語言中的事務怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“Go語言中的事務怎么使用”吧!

事務實踐

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

• 開啟數據庫事務

• 通過 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語言中的事務怎么使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！