인공지능 분야의 Golang 코루틴 적용

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

// createGoroutines 创建协程
func createGoroutines(ctx context.Context, num int) []*int {
    wg := sync.WaitGroup{}
    wg.Add(num)
    results := make([]*int, num)

    for i := 0; i < num; i++ {
        go func(i int) {
            defer wg.Done()
            // 模拟耗时操作
            time.Sleep(time.Duration(i+1) * time.Second)
            results[i] = &i
        }(i)
    }
    go func() {
        wg.Wait()
        close(ctx.Done())
    }()
    return results
}

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 创建 10 个协程
    results := createGoroutines(ctx, 10)

    // 等待所有协程完成
    <-ctx.Done()

    // 输出协程返回的结果
    for _, result := range results {
        fmt.Printf("%d ", *result)
    }
    fmt.Println()
}