Skip to content

Conversation

@bazel-io
Copy link
Member

After the first time a QueryWriteStatus call fails with UNIMPLEMENTED, stop calling the method to avoid an additional roundtrip on every retry.

Closes #28235.

PiperOrigin-RevId: 861621112
Change-Id: Iedf106f771fe3105c4c9e8e7b44d14907bd4e835

Commit fb62559

…rt it (bazelbuild#28235)

After the first time a `QueryWriteStatus` call fails with `UNIMPLEMENTED`, stop calling the method to avoid an additional roundtrip on every retry.

Closes bazelbuild#28235.

PiperOrigin-RevId: 861621112
Change-Id: Iedf106f771fe3105c4c9e8e7b44d14907bd4e835
@bazel-io bazel-io requested a review from a team as a code owner January 28, 2026 21:53
@bazel-io bazel-io added team-Remote-Exec Issues and PRs for the Execution (Remote) team awaiting-review PR is awaiting review from an assigned reviewer labels Jan 28, 2026
@bazel-io bazel-io requested a review from coeuvre January 28, 2026 21:53
@iancha1992 iancha1992 enabled auto-merge January 28, 2026 21:53
Copy link

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

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

Code Review

The changes effectively address the issue of repeatedly calling QueryWriteStatus when the server does not support it. The introduction of an AtomicBoolean flag and the necessary refactoring of AsyncUpload to a non-static inner class are appropriate and correctly implemented. The logic ensures that once an UNIMPLEMENTED status is received, subsequent calls to query() will bypass the QueryWriteStatus request, optimizing network roundtrips. This is a solid improvement to the ByteStreamUploader.

@iancha1992
Copy link
Member

@coeuvre Could you please take a look at the test error?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

awaiting-review PR is awaiting review from an assigned reviewer team-Remote-Exec Issues and PRs for the Execution (Remote) team

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants