Method rateLimitLoop

workers/cache/cache.go:564–622

()

}

func (w *Worker) rateLimitLoop() {
	defer w.Stop()

	ticker := time.NewTicker(30 * time.Second)
	defer ticker.Stop()

	for {
		select {
		case <-w.quit:
			return
		case <-w.ctx.Done():
			slog.DebugContext(w.ctx, "context done")
			return
		case <-ticker.C:
			// update credentials rate limits
			for _, creds := range cache.GetAllGithubCredentials() {
				rateCli, err := github.NewRateLimitClient(w.ctx, creds)
				if err != nil {
					slog.With(slog.Any("error", err)).ErrorContext(w.ctx, "failed to create rate limit client")
					continue
				}
				rateLimit, err := rateCli.RateLimit(w.ctx)
				if err != nil {
					slog.With(slog.Any("error", err)).ErrorContext(w.ctx, "failed to get rate limit")
					continue
				}
				if rateLimit != nil {
					core := rateLimit.GetCore()
					limit := params.GithubRateLimit{
						Limit:     core.Limit,
						Used:      core.Used,
						Remaining: core.Remaining,
						Reset:     core.Reset.Unix(),
					}
					cache.SetCredentialsRateLimit(creds.ID, limit)

					// Record Prometheus metrics
					credID := fmt.Sprintf("%d", creds.ID)
					credName := creds.Name
					endpoint := creds.Endpoint.Name
					if endpoint == "" {
						endpoint = creds.BaseURL
					}

					labels := map[string]string{
						"credential_name": credName,
						"credential_id":   credID,
						"endpoint":        endpoint,
					}

					metrics.GithubRateLimitLimit.With(labels).Set(float64(core.Limit))
					metrics.GithubRateLimitRemaining.With(labels).Set(float64(core.Remaining))
					metrics.GithubRateLimitUsed.With(labels).Set(float64(core.Used))
					metrics.GithubRateLimitResetTimestamp.With(labels).Set(float64(core.Reset.Unix()))
				}
			}
		}
	}