Skip to content

Commit 9a1e9df

Browse files
committed
making in memory representation same as wire contract
1 parent 2a4538c commit 9a1e9df

4 files changed

Lines changed: 230 additions & 309 deletions

File tree

sdk/cosmos/azure-cosmos/azure/cosmos/_cosmos_client_connection.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3418,7 +3418,7 @@ def __GetBodiesFromQueryResult(result: dict[str, Any]) -> list[dict[str, Any]]:
34183418
consecutive_no_progress_pages = 0
34193419

34203420
while pagination_state.can_issue_request():
3421-
head_feedrange = pagination_state.head_feedrange
3421+
head_feedrange = pagination_state.head_range
34223422
if head_feedrange is None:
34233423
break
34243424

@@ -3457,7 +3457,7 @@ def __GetBodiesFromQueryResult(result: dict[str, Any]) -> list[dict[str, Any]]:
34573457
# once more after resume when children X1/X2 restart from the
34583458
# start of their slices.
34593459
while pagination_state.explode_on_multi_overlap(overlapping):
3460-
head_feedrange = pagination_state.head_feedrange
3460+
head_feedrange = pagination_state.head_range
34613461
if head_feedrange is None:
34623462
break
34633463
overlapping = self._routing_map_provider.get_overlapping_ranges(
@@ -3470,8 +3470,8 @@ def __GetBodiesFromQueryResult(result: dict[str, Any]) -> list[dict[str, Any]]:
34703470
backend_request_options = dict(options)
34713471
if pagination_state.remaining_page_item_count is not None:
34723472
backend_request_options["maxItemCount"] = pagination_state.remaining_page_item_count
3473-
if pagination_state.backend_continuation is not None:
3474-
backend_request_options["continuation"] = pagination_state.backend_continuation
3473+
if pagination_state.head_bc is not None:
3474+
backend_request_options["continuation"] = pagination_state.head_bc
34753475
else:
34763476
backend_request_options.pop("continuation", None)
34773477

@@ -3537,8 +3537,8 @@ def __GetBodiesFromQueryResult(result: dict[str, Any]) -> list[dict[str, Any]]:
35373537
elif backend_query_result:
35383538
results = backend_query_result
35393539

3540-
previous_feedrange = pagination_state.head_feedrange
3541-
previous_backend_continuation = pagination_state.backend_continuation
3540+
previous_feedrange = pagination_state.head_range
3541+
previous_backend_continuation = pagination_state.head_bc
35423542
page_items_returned = _count_page_items_from_partial_result(backend_query_result, query)
35433543
if response_headers_list is not None:
35443544
response_headers_list.append(backend_response_headers.copy())
@@ -3553,8 +3553,8 @@ def __GetBodiesFromQueryResult(result: dict[str, Any]) -> list[dict[str, Any]]:
35533553
page_items_returned,
35543554
previous_feedrange,
35553555
previous_backend_continuation,
3556-
pagination_state.head_feedrange,
3557-
pagination_state.backend_continuation,
3556+
pagination_state.head_range,
3557+
pagination_state.head_bc,
35583558
)
35593559
if consecutive_no_progress_pages >= _MAX_CONSECUTIVE_NO_PROGRESS_PAGES:
35603560
pagination_state.write_outbound_continuation(

0 commit comments

Comments
 (0)