Function waitForOrgRateLimit

pkg/cli/update_org.go:258–319

(ctx context.Context, resource string, verbose bool)

}

func waitForOrgRateLimit(ctx context.Context, resource string, verbose bool) error {
	output, err := workflow.RunGHContext(ctx, "Checking rate limit...", "api", "rate_limit")
	if err != nil {
		return err
	}

	var response orgRateLimitResponse
	if err := json.Unmarshal(output, &response); err != nil {
		return err
	}

	limit := response.Resources.Core
	buffer := orgUpdateCoreBuffer
	if resource == "search" {
		limit = response.Resources.Search
		buffer = orgUpdateSearchBuffer
	}
	if limit.Limit == 0 {
		return nil
	}

	orgUpdateLog.Printf("GitHub %s rate limit: %d/%d remaining (reset at %s)", resource, limit.Remaining, limit.Limit, time.Unix(limit.Reset, 0).Format(time.RFC3339))

	// Critical level: once consumption reaches limit-1000 API units, stop processing
	// additional work rather than risking quota exhaustion or a long reset wait. The
	// search resource has a tiny quota, so the critical threshold only applies to core.
	if resource != "search" {
		criticalRemaining := limit.Limit - orgUpdateCriticalConsumed
		if criticalRemaining > buffer && limit.Remaining <= criticalRemaining {
			orgUpdateLog.Printf("GitHub %s rate limit critical: %d/%d remaining (<= %d)", resource, limit.Remaining, limit.Limit, criticalRemaining)
			fmt.Fprintln(os.Stderr, console.FormatWarningMessage(
				fmt.Sprintf("GitHub %s API budget critical: %d/%d remaining (consumed %d API units)", resource, limit.Remaining, limit.Limit, limit.Limit-limit.Remaining),
			))
			return errOrgRateLimitCritical
		}
	}

	if limit.Remaining <= buffer {
		resetAt := time.Unix(limit.Reset, 0)
		waitFor := time.Until(resetAt) + rateLimitResetBuffer
		if waitFor > 0 {
			fmt.Fprintln(os.Stderr, console.FormatWarningMessage(
				fmt.Sprintf("Applying a %s delay to avoid reaching the GitHub %s rate limit (%d/%d remaining)", waitFor.Round(time.Second), resource, limit.Remaining, limit.Limit),
			))
			timer := time.NewTimer(waitFor)
			defer timer.Stop()
			select {
			case <-ctx.Done():
				return ctx.Err()
			case <-timer.C:
			}
			return nil
		}
	}

	if verbose {
		fmt.Fprintln(os.Stderr, console.FormatVerboseMessage(
			fmt.Sprintf("GitHub %s rate limit OK: %d/%d remaining", resource, limit.Remaining, limit.Limit),