Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(kafka): Remove rate limits for kafka ingestion #14460

Merged
merged 4 commits into from
Oct 14, 2024

Conversation

benclive
Copy link
Contributor

What this PR does / why we need it:

When ingesting from Kafka, especially replaying during startup, we often hit stream rate limits. This PR enables two implementations for the stream rate limiter:

  • The existing per-tenant solution we use for currently ingesters. This allows us to defend our system against spikey traffic
  • A NoLimit solution which just returns validation.Unlimited for all tenants. We use this for the kafka write path because our ingest rate is controlled via how fast we read from kafka, rather than how fast clients send us data.

Which issue(s) this PR fixes:
Fixes https://github.com/grafana/loki-private/issues/1145

Special notes for your reviewer:

Checklist

  • Reviewed the CONTRIBUTING.md guide (required)
  • Documentation added
  • Tests updated
  • Title matches the required conventional commits format, see here
    • Note that Promtail is considered to be feature complete, and future development for logs collection will be in Grafana Alloy. As such, feat PRs are unlikely to be accepted unless a case can be made for the feature actually being a bug fix to existing behavior.
  • Changes that require user attention or interaction to upgrade are documented in docs/sources/setup/upgrade/_index.md
  • If the change is deprecating or removing a configuration option, update the deprecated-config.yaml and deleted-config.yaml files respectively in the tools/deprecated-config-checker directory. Example PR

@benclive benclive requested a review from a team as a code owner October 10, 2024 15:39
@@ -404,18 +404,21 @@ func New(cfg Config, clientConfig client.Config, store Store, limits Limits, con
i.SetExtractorWrapper(i.cfg.SampleExtractorWrapper)
}

var limiterStrategy limiterRingStrategy
var streamCountLimiterStrategy limiterRingStrategy
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is a massive nit, but just two suggestions to make the names a little shorter:

Suggested change
var streamCountLimiterStrategy limiterRingStrategy
var streamCountLimiter limiterRingStrategy

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I like this, thanks!

var ownedStreamsStrategy ownershipStrategy
var streamRateLimitStrategy RateLimiterStrategy
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
var streamRateLimitStrategy RateLimiterStrategy
var streamRateLimiter RateLimiterStrategy

@benclive benclive merged commit 83a8893 into main Oct 14, 2024
60 checks passed
@benclive benclive deleted the remove-rate-limits-for-kafka-ingestion branch October 14, 2024 10:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants