Go語言中的事務(wù)怎么使用

本篇內(nèi)容主要講解“Go語言中的事務(wù)怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“Go語言中的事務(wù)怎么使用”吧!

事務(wù)實(shí)踐

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

• 開啟數(shù)據(jù)庫事務(wù)

• 通過 ID 獲取數(shù)據(jù)記錄

• 確認(rèn)是否可以進(jìn)行更新操作

• 如果可以更新操作就更新記錄

• 提交事務(wù)

• 如果遇到錯誤，就回滾事務(wù)

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

事務(wù)說明

從上面的源碼整體看起來沒什么問題。在進(jìn)行相關(guān)的操作時只要正常刪除從db 中刪除數(shù)據(jù)后就完成提交事務(wù)，但是如果在期間如果發(fā)生問題就會返回error就會引發(fā) rollback 操作。

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

tx.releaseConn(err)

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

優(yōu)化方案

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

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

但是沒有任何更改也進(jìn)行提交，然后只有發(fā)生錯誤才進(jìn)行回滾可能會影響代碼的可讀性。在開啟事務(wù)的方法中你會看到在調(diào)用 beginDC 的方法中有使用 context 服務(wù)上下文進(jìn)行回滾事務(wù)。所以還有一個方案就是通過取消上下文來讓事務(wù)結(jié)束從而釋放鎖。

// 方法 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 就會通知事務(wù)已完成，然后就會關(guān)閉事務(wù)。

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語言中的事務(wù)怎么使用”有了更深的了解，不妨來實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！