Skip to content

perf(bigframes): Improve write api upload throughput#16641

Merged
TrevorBergeron merged 4 commits intomainfrom
tbergeron_write_api_redo
Apr 14, 2026
Merged

perf(bigframes): Improve write api upload throughput#16641
TrevorBergeron merged 4 commits intomainfrom
tbergeron_write_api_redo

Conversation

@TrevorBergeron
Copy link
Copy Markdown
Contributor

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

  • Make sure to open an issue as a bug/issue before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea
  • Ensure the tests and linter pass
  • Code coverage does not decrease (if any source code was changed)
  • Appropriate docs were updated (if necessary)

Fixes #<issue_number_goes_here> 🦕

@TrevorBergeron TrevorBergeron requested review from a team as code owners April 13, 2026 22:16
@TrevorBergeron TrevorBergeron requested review from sycai and removed request for a team April 13, 2026 22:16
Copy link
Copy Markdown
Contributor

@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

This pull request refactors the stream_worker function in loader.py to use AppendRowsStream and a futures-based approach for appending rows to BigQuery, which introduces concurrency control via a max_outstanding parameter. The review feedback identifies a regression where row-level errors are no longer checked, potentially leading to silent data loss, and suggests using a try...finally block to ensure the stream manager is properly closed.

@TrevorBergeron TrevorBergeron merged commit ef856b0 into main Apr 14, 2026
30 checks passed
@TrevorBergeron TrevorBergeron deleted the tbergeron_write_api_redo branch April 14, 2026 22:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants