Go并发数据库调用：Goroutine与Channel的合理运用策略

package main

import (
    "fmt"
    "time"
)

// 模拟数据库查询函数
func fetchData(query string) string {
    fmt.Printf("开始查询: %s\n", query)
    time.Sleep(time.Millisecond * 500) // 模拟数据库查询耗时
    fmt.Printf("完成查询: %s\n", query)
    return fmt.Sprintf("数据来自 %s", query)
}

func main() {
    // 顺序调用
    fmt.Println("--- 顺序调用 ---")
    result1 := fetchData("用户统计")
    result2 := fetchData("订单趋势")
    fmt.Printf("结果1: %s, 结果2: %s\n", result1, result2)

    // 并发调用 (仅启动Goroutine，未收集结果)
    fmt.Println("\n--- 并发调用 (未收集结果) ---")
    go fetchData("产品销量")
    go fetchData("地区分布")
    time.Sleep(time.Second * 1) // 实际应用中需要更严谨的等待机制来确保Goroutine完成
    fmt.Println("并发调用完成 (结果未捕获)")
}

package main

import (
    "fmt"
    "sync"
    "time"
)

// 模拟数据库查询函数，返回查询结果和可能的错误
func queryDB(id int, query string) (string, error) {
    fmt.Printf("Goroutine %d: 开始查询 '%s'\n", id, query)
    time.Sleep(time.Millisecond * time.Duration(200+id*50)) // 模拟不同查询耗时
    // if id%3 == 0 {
    //  return "", fmt.Errorf("Goroutine %d: 查询 '%s' 失败", id, query) // 模拟查询失败
    // }
    fmt.Printf("Goroutine %d: 完成查询 '%s'\n", id, query)
    return fmt.Sprintf("数据[%d]: %s", id, query), nil
}

// 结构体用于承载查询结果
type QueryResult struct {
    ID     int
    Data   string
    Error  error
}

func main() {
    fmt.Println("--- 并发数据库查询示例 ---")

    queries := []string{
        "用户活跃度",
        "商品库存",
        "订单量",
        "销售额",
        "访问来源",
    }

    // 创建一个带缓冲的Channel，用于收集所有Goroutine的查询结果
    results := make(chan QueryResult, len(queries))
    var wg sync.WaitGroup

    for i, q := range queries {
        wg.Add(1) // 每次启动一个Goroutine，计数器加1
        go func(goroutineID int, queryStr string) {
            defer wg.Done() // Goroutine完成时，计数器减1

            data, err := queryDB(goroutineID, queryStr)
            results <- QueryResult{
                ID:    goroutineID,
                Data:  data,
                Error: err,
            }
        }(i+1, q) // 传递局部变量，避免闭包陷阱
    }

    // 启动一个Goroutine等待所有查询完成，然后关闭结果Channel
    go func() {
        wg.Wait()      // 等待所有Goroutine执行完毕
        close(results) // 关闭Channel，表示没有更多数据会写入
    }()

    // 从结果Channel中收集数据
    fmt.Println("\n--- 收集查询结果 ---")
    for res := range results { // 循环直到Channel被关闭且所有数据都被读取
        if res.Error != nil {
            fmt.Printf("查询失败 (ID: %d): %v\n", res.ID, res.Error)
        } else {
            fmt.Printf("查询成功 (ID: %d): %s\n", res.ID, res.Data)
        }
    }

    fmt.Println("\n所有查询结果已收集并处理。")
}