Skip to content
This repository was archived by the owner on Mar 6, 2026. It is now read-only.

feat: support setting max_stream_count when fetching query result#2051

Merged
Linchin merged 7 commits intogoogleapis:mainfrom
kien-truong:max-stream-count-api
Nov 22, 2024
Merged

feat: support setting max_stream_count when fetching query result#2051
Linchin merged 7 commits intogoogleapis:mainfrom
kien-truong:max-stream-count-api

Conversation

@kien-truong
Copy link
Contributor

@kien-truong kien-truong commented Nov 3, 2024

Allow user to set max_stream_count when fetching result using BigQuery Storage API with incremental methods:

  • to_arrow_iterable
  • to_dataframe_iterable

Fixes #2030 🦕

@kien-truong kien-truong requested a review from a team November 3, 2024 03:29
@kien-truong kien-truong requested a review from a team as a code owner November 3, 2024 03:29
@product-auto-label product-auto-label bot added size: s Pull request size is small. api: bigquery Issues related to the googleapis/python-bigquery API. labels Nov 3, 2024
@alvarowolfx alvarowolfx requested review from chalmerlowe and removed request for alvarowolfx November 4, 2024 14:08
@Linchin Linchin added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Nov 11, 2024
@Linchin Linchin assigned Linchin and unassigned alvarowolfx Nov 11, 2024
Linchin
Linchin reviewed Nov 11, 2024
View reviewed changes
google/cloud/bigquery/table.py
created by the server. If ``max_queue_size`` is :data:`None`, the queue
size is infinite.

max_stream_count (Optional[int]):
Copy link
Contributor

@Linchin Linchin Nov 11, 2024

Choose a reason for hiding this comment

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

I think it would be more consistent if we use the same docstring as here. It also mentions the effect of preserve_order (in this case self._preserve_order), which I think we should make clear here.

Copy link
Contributor Author

@kien-truong kien-truong Nov 11, 2024

Choose a reason for hiding this comment

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

In this case, _preserve_order is automatically set by parsing the queries, and not a user-facing API. I'll update the docstring to mention that effect.

Copy link
Contributor Author

@kien-truong kien-truong Nov 16, 2024

Choose a reason for hiding this comment

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

Updated

Linchin
Linchin reviewed Nov 11, 2024
View reviewed changes
google/cloud/bigquery/table.py

.. versionadded:: 2.14.0

max_stream_count (Optional[int]):
Copy link
Contributor

@Linchin Linchin Nov 11, 2024

Choose a reason for hiding this comment

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

same as https://github.com/googleapis/python-bigquery/pull/2051/files#r1836568949

@Linchin
Copy link
Contributor

Linchin commented Nov 11, 2024

Thank you @kien-truong for adding further support for max_stream_count! I left some comments, and I wonder if you could add some unit tests too (we typically require 100% unit test coverage). You could find similar tests in PR #2039. If you need any help with it, feel free to let me know. I'd be more than glad to help!

@Linchin Linchin added the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 11, 2024
@yoshi-kokoro yoshi-kokoro removed kokoro:run Add this label to force Kokoro to re-run the tests. kokoro:force-run Add this label to force Kokoro to re-run the tests. labels Nov 11, 2024
@kien-truong
Copy link
Contributor Author

kien-truong commented Nov 11, 2024

Hi, the default code path with the default arguments is already covered by the current tests.
I'll see if I can add tests for when the user manually set the argument when I have time at weekend.

@Linchin
Copy link
Contributor

Linchin commented Nov 11, 2024

@kien-truong sounds good. The mypy test is also failing, could you fix it too? You can run it by running nox -s mypy in the repo. (You can install nox by pip install nox)

@product-auto-label product-auto-label bot added size: m Pull request size is medium. and removed size: s Pull request size is small. labels Nov 16, 2024
@kien-truong kien-truong force-pushed the max-stream-count-api branch 3 times, most recently from 31662ad to 0e52722 Compare November 16, 2024 03:51
@kien-truong
Copy link
Contributor Author

kien-truong commented Nov 16, 2024

I have added tests to cover user-provided max_stream_count flow and fixed the mypy test as well.

@kien-truong kien-truong mentioned this pull request Nov 16, 2024
Closed
@chalmerlowe chalmerlowe added the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 18, 2024
@yoshi-kokoro yoshi-kokoro removed the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 18, 2024
@Linchin Linchin added the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 18, 2024
@yoshi-kokoro yoshi-kokoro removed the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 18, 2024
Linchin
Linchin reviewed Nov 19, 2024
View reviewed changes
Linchin
Linchin reviewed Nov 19, 2024
View reviewed changes
@Linchin
Copy link
Contributor

Linchin commented Nov 19, 2024

Thanks @kien-truong, the PR mostly looks good. I just have some small question regarding ignored coverage for the tests.

@Linchin Linchin added the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 22, 2024
@yoshi-kokoro yoshi-kokoro removed the kokoro:run Add this label to force Kokoro to re-run the tests. label Nov 22, 2024
@Linchin Linchin self-requested a review November 22, 2024 21:34
Linchin
Linchin approved these changes Nov 22, 2024
View reviewed changes
Copy link
Contributor

@Linchin Linchin left a comment

Choose a reason for hiding this comment

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

LGTM. Thank you!

@Linchin Linchin merged commit d461297 into googleapis:main Nov 22, 2024
@release-please release-please bot mentioned this pull request Nov 18, 2024
Merged
@chalmerlowe chalmerlowe mentioned this pull request Nov 25, 2024
Closed
This was referenced Jan 21, 2025
Closed
Closed
Closed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Reviewers

@chalmerlowe chalmerlowe Awaiting requested review from chalmerlowe

1 more reviewer

@Linchin Linchin Linchin approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

@Linchin Linchin

Labels

api: bigquery Issues related to the googleapis/python-bigquery API. size: m Pull request size is medium.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Make max_stream_count configurable when using Bigquery Storage API

5 participants

@kien-truong @Linchin @alvarowolfx @chalmerlowe @yoshi-kokoro