同时处理多种类型的任务

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《同时处理多种类型的任务》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

我有以下用例，我很难确定正确的并发模式。

我有一个服务，其中包含要执行的计算任务列表：

• 有些执行速度很快（几秒钟），有些可能需要几个小时
• 我绝不希望同一个任务同时执行两次
• 不同的任务可以并行执行
• 如果一个任务应该每小时刷新一次，但在某些时候，执行时间超过 1 小时，我想放弃新的执行，但将其记录在某处
• 我想要任务超时

我当然可以使用外部库。

我查看了singleflight，但这似乎是为了缓存，并且任务的新执行将被阻止，直到该任务完成计算。就我而言，我想记录它已经在计算的事实，但放弃新的执行（或者只是不执行）。

解决方案

维护正在进行的任务的地图。添加任务时，检查地图并记录任务是否正在进行；将任务添加到地图。完成后从地图中删除任务。

下面是一个示例，假设任务由 string 标识并由 func() 实现。

var (
    mu         sync.Mutex
    inProgress = map[string]bool{}
)

func startTask(id string, fn func()) {
    mu.Lock()
    ip := inProgress[id]
    if !ip {
        inProgress[id] = true
    }
    mu.Unlock()
    if ip {
        log.Printf("task %s in progress", id)
    } else {
        go func() {
            fn()
            mu.Lock()
            delete(inProgress, id)
            mu.Unlock()
        }()
    }
}

终于介绍完啦！小伙伴们，这篇关于《同时处理多种类型的任务》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！