Go语言并发数据库调用：Goroutine与Channel的策略与实践

package main

import (
    "fmt"
    "time"
)

// simulateDBQuery 模拟一个数据库查询操作
func simulateDBQuery(queryID int) {
    fmt.Printf("Goroutine %d: 开始查询数据库...\n", queryID)
    time.Sleep(time.Duration(queryID) * 100 * time.Millisecond) // 模拟查询延迟
    fmt.Printf("Goroutine %d: 查询完成。\n", queryID)
}

func main() {
    fmt.Println("主Goroutine: 启动多个并发数据库查询...")
    for i := 1; i <= 3; i++ {
        go simulateDBQuery(i) // 为每个查询启动一个Goroutine
    }

    // 为了让主Goroutine等待其他Goroutine完成，这里简单地等待一段时间
    // 实际应用中会使用sync.WaitGroup或channel进行同步
    time.Sleep(500 * time.Millisecond)
    fmt.Println("主Goroutine: 所有查询（可能）已启动。")
}

package main

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

// GraphData 结构体用于承载每个图表的数据
type GraphData struct {
    ID   int
    Data string
    Err  error // 用于传递可能的错误
}

// fetchGraphData 模拟从数据库获取单个图表数据
func fetchGraphData(graphID int, resultChan chan<- GraphData, wg *sync.WaitGroup) {
    defer wg.Done() // 确保Goroutine完成时递减WaitGroup计数器

    fmt.Printf("Fetching data for graph %d...\n", graphID)
    // 模拟数据库查询延迟
    time.Sleep(time.Duration(graphID) * 100 * time.Millisecond)

    // 模拟查询结果和可能的错误
    var data string
    var err error
    if graphID%2 == 0 { // 模拟偶数ID查询成功
        data = fmt.Sprintf("Graph %d Data: [Value_A, Value_B]", graphID)
    } else { // 模拟奇数ID查询失败
        err = fmt.Errorf("failed to fetch data for graph %d", graphID)
    }

    resultChan <- GraphData{ID: graphID, Data: data, Err: err}
    fmt.Printf("Finished fetching data for graph %d.\n", graphID)
}

func main() {
    graphIDs := []int{1, 2, 3, 4, 5} // 假设有5个图表需要数据
    numGraphs := len(graphIDs)

    // 创建一个带缓冲的Channel，用于发送结果
    resultChan := make(chan GraphData, numGraphs)
    var wg sync.WaitGroup // 使用WaitGroup来等待所有Goroutine完成

    fmt.Println("Starting concurrent database calls...")

    for _, id := range graphIDs {
        wg.Add(1) // 为每个Goroutine增加WaitGroup计数器
        go fetchGraphData(id, resultChan, &wg)
    }

    // 启动一个Goroutine来等待所有数据获取Goroutine完成，然后关闭Channel
    go func() {
        wg.Wait()         // 等待所有fetchGraphData Goroutine完成
        close(resultChan) // 关闭Channel，表示没有更多数据会发送
        fmt.Println("All data fetching goroutines finished, channel closed.")
    }()

    // 从Channel收集结果
    fmt.Println("\nCollecting results:")
    var allGraphData []GraphData
    for result := range resultChan { // 循环直到Channel被关闭
        allGraphData = append(allGraphData, result)
        if result.Err != nil {
            fmt.Printf("Error for Graph %d: %v\n", result.ID, result.Err)
        } else {
            fmt.Printf("Collected: %s\n", result.Data)
        }
    }

    fmt.Println("\nAll results collected. Application finished.")
    // 此时 allGraphData 包含了所有图表的数据和错误信息
    // 可以在这里进一步处理或聚合数据
}