117笔记问答Go语言中的WaitGroup是一种用于等待一组goroutine完成执行的机制。在高并发数据库访问中，可以使用WaitGroup来等待所有数据库操作完成后再进行后续处理，以确保并发操作的正确性。
使用WaitGroup需要以下三个步骤：
- 创建WaitGroup对象：使用sync包中的WaitGroup类型创建一个WaitGroup对象。
var wg sync.WaitGroup
- 增加等待的goroutine数量：在每个要等待的goroutine之前，使用Add方法增加WaitGroup的计数器。每个增加的数量应与要等待的goroutine数量相等。
wg.Add(1)
- goroutine完成后减少计数器：在每个goroutine执行完成后，使用Done方法减少WaitGroup的计数器。
wg.Done()
- 等待所有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，可以确保所有的数据库查询操作都完成后再进行后续处理，从而实现高并发数据库访问。