Go x/sync/errgroup.TryGo 怎样在并发已满时快速拒绝任务

当一批任务不能无限制地创建 goroutine，又不希望提交方因为等待空闲名额而卡住时，可以用 errgroup.Group.SetLimit 配合 TryGo。结论很明确：有空闲并发额度时，TryGo 启动函数并返回 true；活动 goroutine 已达到上限时，它不会启动函数，而是立即返回 false。

官方地址：https://pkg.go.dev/golang.org/x/sync/errgroup

因此，false 不是任务函数返回了错误，而是任务根本没有进入 errgroup。调用方必须为这个分支决定丢弃、排队或稍后重试。

先把并发上限设在 Group 上

SetLimit(n) 限制的是当前活动 goroutine 数量。正数表示最多允许 n 个任务同时活动，负数表示不设上限，零则会阻止新任务加入。上限应该在任务开始前设置，活动任务运行期间不要修改它。

package main

import (
	"fmt"
	"golang.org/x/sync/errgroup"
)

func main() {
	var group errgroup.Group
	// 只允许两个任务同时占用 Group 的活动额度。
	group.SetLimit(2)

	started := 0
	for id := 0; id 

这个示例的关键不是打印内容，而是提交契约：TryGo 的布尔值必须被消费。任务很短时，后续提交可能赶在前一个任务结束后获得名额；任务较慢时，达到上限的提交才会稳定落入 false 分支。

TryGo 与 Go 的差别在提交动作

group.Go(f) 在达到上限时会阻塞提交方，直到它能加入新的活动任务；group.TryGo(f) 则只尝试一次，名额不可用就返回 false。两者启动后的错误处理仍由 Wait 负责，但背压策略完全不同。

如果每个输入都必须处理，false 分支不能静默丢掉。可以把任务放进有界 channel，由单独的提交循环再次调用 TryGo；如果业务允许降级，则记录任务标识并直接返回“稍后处理”。有限重试要有次数或截止时间，否则它只是把阻塞从 Go 挪到了业务代码。

Wait 只汇总已启动任务的错误

被 TryGo 拒绝的函数从未执行，所以它不会出现在 Wait 的错误结果里。业务日志至少应区分“提交被拒绝”和“任务执行返回 error”两种状态，避免监控只看 Wait 而漏掉入口处的丢弃量。

// submit 负责接收任务，run 负责执行已被接受的任务。
func submit(group *errgroup.Group, job func() error, onReject func()) bool {
	// TryGo 只报告是否成功启动，不会替调用方保存未启动的 job。
	if !group.TryGo(job) {
		onReject() // 这里可以写入有界队列或拒绝计数。
		return false
	}
	return true
}

在统一收尾处调用 Wait，只处理已启动任务的首个非空错误；拒绝分支需要自己的记录、重试或补偿通道。

需要取消后续工作时使用 WithContext

如果任务之间共享一个请求生命周期，可以用 errgroup.WithContext 得到派生上下文。某个已启动任务返回非 nil 错误后，派生上下文会被取消，其他任务应在 I/O 或循环边界检查 ctx.Done()。这不会自动重试被 TryGo 拒绝的任务，也不会替你定义队列。

func runBatch(ctx context.Context, jobs []func(context.Context) error) error {
	group, derived := errgroup.WithContext(ctx)
	group.SetLimit(4)
	for _, job := range jobs {
		current := job
		if !group.TryGo(func() error {
			// 把取消信号传给任务，避免失败后继续做无效工作。
			return current(derived)
		}) {
			// 未启动的任务必须由调用方保存或明确放弃。
			continue
		}
	}
	return group.Wait()
}

使用前的三项速查

• 需要提交方等待名额时用 Go；需要立即知道“现在能不能接”时用 TryGo。
• 看到 false 先记录拒绝，再决定排队、重试或降级，不要把它交给 Wait 猜测。
• 固定好 SetLimit 后再启动任务；用 Wait 收口已接受任务，用 Context 传播取消。

常见问题

TryGo 返回 false 会返回一个 error 吗？

不会。它返回的是 bool，表示函数是否已经被启动；被拒绝的函数没有执行机会，业务方需要自己处理。

SetLimit(0) 适合用来暂停任务吗？

它会让新的任务无法加入，但不应把运行期修改上限当成通用暂停机制。更稳妥的做法是由提交层控制输入，并在任务边界使用 Context。