def build_updated_request(self): """ Updates the given message request as per last scanned key """ r_kwargs = { "table_name": self.message.table_name, "filter": self.message.filter, } if self.message.rows_limit != 0: r_kwargs["rows_limit"] = max( 1, self.message.rows_limit - self.rows_read_so_far ) # if neither RowSet.row_keys nor RowSet.row_ranges currently exist, # add row_range that starts with last_scanned_key as start_key_open # to request only rows that have not been returned yet if not self.message.HasField("rows"): row_range = data_v2_pb2.RowRange(start_key_open=self.last_scanned_key) r_kwargs["rows"] = data_v2_pb2.RowSet(row_ranges=[row_range]) else: row_keys = self._filter_rows_keys() row_ranges = self._filter_row_ranges() r_kwargs["rows"] = data_v2_pb2.RowSet( row_keys=row_keys, row_ranges=row_ranges ) return data_messages_v2_pb2.ReadRowsRequest(**r_kwargs)
def build_updated_request(self): """ Updates the given message request as per last scanned key """ r_kwargs = { 'table_name': self.message.table_name, 'filter': self.message.filter } if self.message.rows_limit != 0: r_kwargs['rows_limit'] = max( 1, self.message.rows_limit - self.rows_read_so_far) row_keys = self._filter_rows_keys() row_ranges = self._filter_row_ranges() r_kwargs['rows'] = data_v2_pb2.RowSet(row_keys=row_keys, row_ranges=row_ranges) return data_messages_v2_pb2.ReadRowsRequest(**r_kwargs)