RUM-16394 Pt.2: Implement QuotaChecker for profiling quota HTTP check by ambushwork · Pull Request #3496 · DataDog/dd-sdk-android

ambushwork · 2026-06-02T13:10:18Z

What does this PR do?

Implement QuotaChecker for profiling quota HTTP check

Motivation

RUM-16394

Review checklist (to be filled by reviewers)

Feature or bugfix MUST have appropriate tests (unit, integration, e2e)
Make sure you discussed the feature or bugfix with the maintaining team in an Issue
Make sure each commit and the PR mention the Issue number (cf the CONTRIBUTING doc)

Copilot

Pull request overview

Implements a profiling quota HTTP check component (QuotaChecker) and updates the quota result model to use a typed QuotaReason enum, with comprehensive unit tests to validate request building, response parsing, error handling, and deduplication behavior.

Changes:

Add QuotaChecker to perform async quota checks against the quota endpoint and emit QuotaResult.
Introduce QuotaReason enum and migrate QuotaResult.reason from String to QuotaReason (including a new QUOTA_EXCEEDED predefined result).
Add/extend unit tests covering parsing, HTTP errors, request construction, deduplication/reset, last-result behavior, and custom-endpoint bypass.

Reviewed changes

Copilot reviewed 5 out of 5 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
features/dd-sdk-android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt	New async HTTP quota checker, parsing, and telemetry logging.
features/dd-sdk-android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaReason.kt	New enum encapsulating quota reasons and raw backend values.
features/dd-sdk-android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaResult.kt	Switch `reason` to `QuotaReason` and add `QUOTA_EXCEEDED` constant.
features/dd-sdk-android-profiling/src/test/kotlin/com/datadog/android/profiling/internal/quota/QuotaCheckerTest.kt	New test suite validating `QuotaChecker` behavior end-to-end with mocked OkHttp.
features/dd-sdk-android-profiling/src/test/kotlin/com/datadog/android/profiling/internal/quota/QuotaResultTest.kt	Update tests for `QuotaReason`-based `QuotaResult`.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

codecov-commenter · 2026-06-02T15:27:37Z

Codecov Report

❌ Patch coverage is 87.91209% with 11 lines in your changes missing coverage. Please review.
✅ Project coverage is 72.43%. Comparing base (1ebad9f) to head (55fd32d).

Files with missing lines	Patch %	Lines
...g/android/profiling/internal/quota/QuotaChecker.kt	86.08%	6 Missing and 5 partials ⚠️

Additional details and impacted files

@@                       Coverage Diff                        @@
##           feature/continuous-profiling    #3496      +/-   ##
================================================================
+ Coverage                         72.41%   72.43%   +0.02%     
================================================================
  Files                               980      982       +2     
  Lines                             36268    36356      +88     
  Branches                           6037     6049      +12     
================================================================
+ Hits                              26261    26333      +72     
- Misses                             8361     8363       +2     
- Partials                           1646     1660      +14

Files with missing lines	Coverage Δ
...og/android/profiling/internal/quota/QuotaReason.kt	`100.00% <100.00%> (ø)`
...og/android/profiling/internal/quota/QuotaResult.kt	`100.00% <100.00%> (ø)`
...g/android/profiling/internal/quota/QuotaChecker.kt	`86.08% <86.08%> (ø)`

... and 32 files with indirect coverage changes

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

ambushwork · 2026-06-03T09:55:12Z

@codex review

chatgpt-codex-connector

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 972ded7447

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

Open a pull request for review
Mark a draft as ready
Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

0xnm · 2026-06-05T15:36:50Z

+
+    @Suppress("UnsafeThirdPartyFunctionCall") // Wrapped in the try-catch
+    private fun buildRequest(sessionId: String, datadogContext: DatadogContext): Request {
+        val intakeHost = datadogContext.site.intakeEndpoint.removePrefix("https://")


why do we remove prefix?

because the quota endpoint is https://quota.$intakeHostName and intakeEndpoint is https://$intakeHostName, so we need to remove the prefix to extract the intakeHostName here.

0xnm · 2026-06-05T15:39:28Z

+                .getJSONObject(JSON_KEY_DATA)
+                .getJSONObject(JSON_KEY_ATTRIBUTES)
+            val admitted = attrs.optBoolean(JSON_KEY_ADMITTED, true)
+            val rawReason = attrs.optString(JSON_KEY_REASON).ifEmpty { null }


Suggested change

val rawReason = attrs.optString(JSON_KEY_REASON).ifEmpty { null }

val rawReason = attrs.optString(JSON_KEY_REASON)

default fallback is empty string and rawReason is used only in the parseReason, both null and empty string yield to the same branch there.

0xnm · 2026-06-05T15:43:34Z

+        internal const val LOG_UNEXPECTED_ERROR = "Quota check unexpected error: %s"
+        internal const val LOG_QUOTA_DENIED = "Profiling quota denied: reason=%s"
+
+        internal const val QUOTA_URL_TEMPLATE = "https://quota.%s/api/v2/profiling/quota?session_id=%s"


to confirm that this is deployed in all DCs

0xnm · 2026-06-05T15:44:25Z

+        private set
+
+    fun checkAsync(sessionId: String, datadogContext: DatadogContext) {
+        if (hasCustomEndpoint) {


what is the use-case for that, who can set it?

This is mentioned by @simaoseica-dd before, when user uses custom endpoint, we shouldn't check quota API

0xnm · 2026-06-05T15:46:20Z

+    var lastResult: QuotaResult? = null
+        private set
+
+    fun checkAsync(sessionId: String, datadogContext: DatadogContext) {


what will be the logic behind calling it? for example, we made a call, but got IO exception because there is no network. Obviously, we need to repeat the call, but a) when; b) with back-off?

With current logic, we make a call only when the session is renewed, and we don't retry if there is a network error, because the network error is a fail open according to the browser profiling logic, which can be changed later ofc.

I have a concern that browser logic regarding network errors can be be applied to mobile. The fair share of Browser SDK is on the desktop, there is always a connectivity there. Of course mobile browser can be used as well, but once again it should be a connectivity when page is loaded (and probably RUM session is started at this point).

Also Browser profiling requires a document policy sent by the server https://developer.mozilla.org/en-US/docs/Web/API/JS_Self-Profiling_API#security_requirements, this also implies having a connectivity.

0xnm · 2026-06-05T15:49:35Z

+        }
+        val previousId = lastSessionId.getAndSet(sessionId)
+        if (previousId == sessionId) return // same session: in-flight check (if any) is still valid
+        pendingFuture.getAndSet(null)?.cancel(true)


nb: if it is BackPressureExecutorService or LoggingScheduledThreadPoolExecutor, .cancel will trigger error log entry

The plan is to use SingleThreadScheduledExecutor to avoid this.

This comment has been minimized.

Sign in to view

ambushwork requested a review from Copilot June 2, 2026 13:21

Copilot started reviewing on behalf of ambushwork June 2, 2026 13:22 View session

Copilot AI reviewed Jun 2, 2026

View reviewed changes

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt

ambushwork force-pushed the yl/profiling/add-quota-checker branch 3 times, most recently from 146ea0f to 972ded7 Compare June 2, 2026 14:49

chatgpt-codex-connector Bot reviewed Jun 3, 2026

View reviewed changes

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt Outdated

ambushwork force-pushed the yl/profiling/add-quota-checker branch from 972ded7 to e5d4bc1 Compare June 3, 2026 12:18

ambushwork marked this pull request as ready for review June 3, 2026 12:26

ambushwork requested review from a team as code owners June 3, 2026 12:26

abrooksv reviewed Jun 3, 2026

View reviewed changes

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt

ambushwork force-pushed the yl/profiling/add-quota-checker branch from e5d4bc1 to fdf7907 Compare June 4, 2026 13:31

abrooksv previously approved these changes Jun 4, 2026

View reviewed changes

satween reviewed Jun 5, 2026

View reviewed changes

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt

Comment thread ...android-profiling/src/main/java/com/datadog/android/profiling/internal/quota/QuotaChecker.kt Outdated

ambushwork dismissed abrooksv’s stale review via 4bd33a8 June 5, 2026 13:33

ambushwork force-pushed the yl/profiling/add-quota-checker branch from fdf7907 to 4bd33a8 Compare June 5, 2026 13:33

ambushwork requested a review from satween June 5, 2026 13:33

satween previously approved these changes Jun 5, 2026

View reviewed changes

ambushwork dismissed satween’s stale review via 0152785 June 5, 2026 14:42

ambushwork force-pushed the yl/profiling/add-quota-checker branch from 4bd33a8 to 0152785 Compare June 5, 2026 14:42

ambushwork requested review from abrooksv and satween June 5, 2026 15:49

0xnm reviewed Jun 5, 2026

View reviewed changes

RUM-16394 Pt.2: Implement QuotaChecker for profiling quota HTTP check

55fd32d

ambushwork force-pushed the yl/profiling/add-quota-checker branch from 0152785 to 55fd32d Compare June 5, 2026 16:53

	val rawReason = attrs.optString(JSON_KEY_REASON).ifEmpty { null }
	val rawReason = attrs.optString(JSON_KEY_REASON)

Conversation

ambushwork commented Jun 2, 2026

What does this PR do?

Motivation

Review checklist (to be filled by reviewers)

Uh oh!

This comment has been minimized.

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

codecov-commenter commented Jun 2, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

ambushwork commented Jun 3, 2026

Uh oh!

chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

💡 Codex Review

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

0xnm Jun 5, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

6 participants

codecov-commenter commented Jun 2, 2026 •

edited

Loading

0xnm Jun 5, 2026 •

edited

Loading