Go語言中的WaitGroup是一種用于等待一組goroutine完成執行的機制。在高并發數據庫訪問中，可以使用WaitGroup來等待所有數據庫操作完成后再進行后續處理，以確保并發操作的正確性。

使用WaitGroup需要以下三個步驟：

1. 創建WaitGroup對象：使用sync包中的WaitGroup類型創建一個WaitGroup對象。

var wg sync.WaitGroup

2. 增加等待的goroutine數量：在每個要等待的goroutine之前，使用Add方法增加WaitGroup的計數器。每個增加的數量應與要等待的goroutine數量相等。

wg.Add(1)

3. goroutine完成后減少計數器：在每個goroutine執行完成后，使用Done方法減少WaitGroup的計數器。

wg.Done()

4. 等待所有goroutine完成：使用Wait方法來等待所有的goroutine完成執行。Wait方法會阻塞調用的goroutine，直到計數器歸零。

wg.Wait()

下面是一個使用WaitGroup實現高并發數據庫訪問的示例代碼：

package main
import (
"database/sql"
"fmt"
"sync"
_ "github.com/go-sql-driver/mysql"
)
func main() {
db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database")
if err != nil {
fmt.Println("Failed to connect to database:", err)
return
}
defer db.Close()
// 并發查詢數量
concurrency := 10
// 創建WaitGroup對象
var wg sync.WaitGroup
// 增加等待的goroutine數量
wg.Add(concurrency)
for i := 0; i < concurrency; i++ {
go func(id int) {
defer wg.Done()
// 執行數據庫查詢操作
rows, err := db.Query("SELECT * FROM table")
if err != nil {
fmt.Println("Failed to query database:", err)
return
}
defer rows.Close()
// 處理查詢結果
for rows.Next() {
// ...
}
}(i)
}
// 等待所有goroutine完成
wg.Wait()
fmt.Println("All queries completed.")
}

在上述代碼中，首先創建了一個WaitGroup對象，并設置了并發查詢的數量。然后，通過增加等待的goroutine數量來增加WaitGroup的計數器。在每個goroutine中執行數據庫查詢操作，并在查詢完成后使用Done方法減少計數器。最后，使用Wait方法來等待所有的goroutine完成執行。

通過使用WaitGroup，可以確保所有的數據庫查詢操作都完成后再進行后續處理，從而實現高并發數據庫訪問。