文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

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

2023-10-25 20: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来等待所有的查询操作完成,并在所有结果都被处理后关闭结果通道。然后遍历结果通道,输出查询结果。

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

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯