中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

golang并發查詢數據庫怎么實現

小億
143
2023-10-25 19:01:34
欄目: 編程語言

在Go語言中實現并發查詢數據庫可以使用goroutine和channel來實現。以下是一個簡單的示例代碼:

package main
import (
    "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來等待所有的查詢操作完成,并在所有結果都被處理后關閉結果通道。然后遍歷結果通道,輸出查詢結果。

這樣我們就實現了并發查詢數據庫的功能。

0
庆阳市| 达州市| 娄烦县| 平遥县| 徐汇区| 松潘县| 临澧县| 衡阳市| 洪泽县| 黄龙县| 宁乡县| 临清市| 灵宝市| 乌兰浩特市| 墨脱县| 含山县| 遂川县| 慈利县| 德化县| 横峰县| 龙南县| 牡丹江市| 会理县| 石景山区| 海林市| 长沙市| 敦化市| 乐昌市| 汨罗市| 灌南县| 正宁县| 江华| 呼伦贝尔市| 仁化县| 浦北县| 马山县| 历史| 崇礼县| 沁阳市| 宁城县| 腾冲县|