go 线程限制数量 --chatGPT

问:runTask(names, limit), 遍历启动以names的子名称的工作线程 name测试打印,上限数量是limit, 要求打印所有names

gpt:

你可以使用 Go 协程来实现 `runTask` 函数,该函数会遍历启动以 `names` 子名称的工作线程,并在达到上限数量 `limit` 后等待所有工作线程完成。以下是一个示例代码:

```go
package main

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

func worker(name string) {
    for i := 0; i < 3; i++ { // 模拟工作线程执行3次
        fmt.Printf("[%s] 执行任务 %d\n", name, i)
        time.Sleep(time.Second)
    }
}

func runTask(names []string, limit int) {
    var wg sync.WaitGroup
    semaphore := make(chan struct{}, limit)

    for _, name := range names {
        semaphore <- struct{}{} // 占用一个信号量

        wg.Add(1)
        go func(name string) {
            defer func() {
                <-semaphore // 释放信号量
                wg.Done()
            }()
            worker(name)
        }(name)
    }

    wg.Wait()
}

func main() {
    names := []string{"Worker1", "Worker2", "Worker3", "Worker4", "Worker5"}
    limit := 2 // 上限数量
    runTask(names, limit)
}
```

在这个示例中,我们定义了 `worker` 函数来模拟工作线程的工作。`runTask` 函数接收一个包含工作线程名称的切片 `names` 和一个上限数量 `limit`。它使用信号量来控制并发,并等待所有工作线程完成。

在 `main` 函数中,我们定义了一个包含工作线程名称的切片 `names`,并指定了上限数量 `limit`。然后,我们调用 `runTask` 函数来启动工作线程,并在达到上限数量后等待它们完成。你可以根据需要修改 `names` 和 `limit` 的值。

你可能感兴趣的:(golang,开发语言,后端)