使用channel控制并发个数

最新推荐文章于 2024-02-19 20:37:01 发布

小叶曲

最新推荐文章于 2024-02-19 20:37:01 发布

阅读量199

点赞数

分类专栏： golang 文章标签： golang

本文链接：https://blog.youkuaiyun.com/qq_42490050/article/details/128555552

版权

golang 专栏收录该内容

3 篇文章

订阅专栏

该代码示例展示了一个Go程序，它使用并发处理字符串数组，计算每个字符串中每个字母出现的频率。通过引入`net/http/pprof`包，可能还涉及性能监控。程序首先定义了一个`LetterFreq`映射类型，然后在`CountLetters`函数中，利用通道和等待组来并发处理字符串并更新结果映射。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
    "fmt"
    _ "net/http/pprof"
    "sync"
    "time"
)

func main() {
    strArr := []string{"abc", "cde", "aabb", "love", "nba"}
    res := CountLetters(strArr, 2)
    fmt.Println(res)
}

type LetterFreq map[rune]int

func CountLetters(strArr []string, concurrency int) LetterFreq {
    res := make(LetterFreq)

    ch := make(chan struct{}, concurrency)
    defer close(ch)
    wg := sync.WaitGroup{}

    for i := 0; i < len(strArr); i++ {
        ch <- struct{}{}
        wg.Add(1)
        go func(index int) {
            defer wg.Done()
            defer func() { <-ch }()

            time.Sleep(1 * time.Second)
            fmt.Printf("do work %d %s\n", index, strArr[index])
        }(i)
    }
    wg.Wait()

    return res
}