golang并发查询数据库怎么实现

   2025-02-15 9530
核心提示:在Go语言中实现并发查询数据库可以使用goroutine和channel来实现。以下是一个简单的示例代码:package mainimport (database/sql

在Go语言中实现并发查询数据库可以使用goroutine和channel来实现。以下是一个简单的示例代码:

package mainimport (    "database/sql"    "fmt"    "log"    "sync"    _ "github.com/go-sql-driver/mysql")type Result struct {    ID   int    Name string}func dbQuery(db *sql.DB, query string, resultChan chan<- Result, wg *sync.WaitGroup) {    defer wg.Done()    rows, err := db.Query(query)    if err != nil {        log.Printf("Error executing query: %s\n", err)        return    }    defer rows.Close()    for rows.Next() {        var result Result        err := rows.Scan(&result.ID, &result.Name)        if err != nil {            log.Printf("Error scanning result: %s\n", err)            return        }        resultChan <- result    }}func main() {    db, err := sql.Open("mysql", "user:password@tcp(localhost:3306)/database")    if err != nil {        log.Fatalf("Error connecting to database: %s\n", err)    }    defer db.Close()    queries := []string{"SELECT id, name FROM table1", "SELECT id, name FROM table2"}    resultChan := make(chan Result)    var wg sync.WaitGroup    for _, query := range queries {        wg.Add(1)        go dbQuery(db, query, resultChan, &wg)    }    go func() {        wg.Wait()        close(resultChan)    }()    for result := range resultChan {        fmt.Printf("ID: %d, Name: %s\n", result.ID, result.Name)    }}

上述代码中,我们首先建立数据库连接,然后定义一个Query函数,该函数接受一个SQL查询语句和一个结果通道,通过查询语句从数据库中获取数据,并将结果发送到结果通道中。

在主函数中,我们定义了一个查询语句的切片,然后创建了一个结果通道和一个等待组。接着,我们遍历查询语句切片,为每个查询语句启动一个goroutine来执行查询操作。在goroutine中,我们调用Query函数来执行数据库查询,并将结果发送到结果通道中。

最后,我们在主函数中启动一个goroutine来等待所有的查询操作完成,并在所有结果都被处理后关闭结果通道。然后遍历结果通道,输出查询结果。

这样我们就实现了并发查询数据库的功能。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言