-
Notifications
You must be signed in to change notification settings - Fork 13.8k
[FLINK-19232][python] support iterating MapState and MapView #13739
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
Conversation
|
Thanks a lot for your contribution to the Apache Flink project. I'm the @flinkbot. I help the community Automated ChecksLast check on commit 9f75014 (Thu Oct 22 05:01:49 UTC 2020) Warnings:
Mention the bot in a comment to re-run the automated checks. Review Progress
Please see the Pull Request Review Guide for a full explanation of the review process. DetailsThe Bot is tracking the review progress through labels. Labels are applied according to the order of the review items. For consensus, approval by a Flink committer of PMC member is required Bot commandsThe @flinkbot bot supports the following commands:
|
| } else { | ||
| iterator = mapStateIteratorCache.get(iteratorToken); | ||
| if (iterator == null) { | ||
| throw new RuntimeException("The cached iterator not exist!"); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| throw new RuntimeException("The cached iterator not exist!"); | |
| throw new RuntimeException("The cached iterator does not exist!"); |
| public static final ConfigOption<Integer> MAP_STATE_ITERATE_CACHE_SIZE = ConfigOptions | ||
| .key("python.map-state.iterate-cache-size") | ||
| .defaultValue(1000) | ||
| .withDescription("The maximum number of entries read from Java side when iterating a " + |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What about improving the doc a bit to make it more clear what this config option is used for?
| if self._underlying_is_write_cache: | ||
| # Iterate the data in write cache firstly | ||
| try: | ||
| key, existed_and_value = next(self._underlying) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| key, existed_and_value = next(self._underlying) | |
| self._last_key, (exists, value) = next(self._underlying) |
| # cache. | ||
| if self._internal_map_state._cleared: | ||
| del self._internal_map_state._write_cache[self._last_key] | ||
| self._mod_count += 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could refactor a bit to remove the duplicate code
| self._write_cache.clear() | ||
| self._cleared = False | ||
|
|
||
| def _iterate_next_batch(self, iterate_type, iterator_token): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| def _iterate_next_batch(self, iterate_type, iterator_token): | |
| def next_batch(self, iterate_type, iterator_token): |
| </tr> | ||
| <tr> | ||
| <td><h5>python.map-state.read.cache.size</h5></td> | ||
| <td><h5>python.map-state.iterate-cache-size</h5></td> |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
python.map-state.iterate-req-batch-size
9f75014 to
844e71b
Compare
|
@dianfu Thanks for your comments! I have updated this PR. |
What is the purpose of the change
This pull request supports iterating MapState and MapView
Brief change log
Verifying this change
This change is already covered by existing tests.
Does this pull request potentially affect one of the following parts:
@Public(Evolving): (no)Documentation