tikv
/
client-rust
mirror of https://github.com/tikv/client-rust.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added a fix for the bug with the range 

Signed-off-by: limbooverlambda <schakra1@gmail.com>
This commit is contained in:
limbooverlambda 2024-08-12 12:19:26 -07:00
parent 6b40932b1c
commit a7b4b74e0a
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.toml
View File

@ -14,7 +14,6 @@ prometheus = ["prometheus/push", "prometheus/process"]
# Enable integration tests with a running TiKV and PD instance. # Enable integration tests with a running TiKV and PD instance.
# Use $PD_ADDRS, comma separated, to set the addresses the tests use. # Use $PD_ADDRS, comma separated, to set the addresses the tests use.
integration-tests = [] integration-tests = []
protobuf-codec = []
[lib] [lib]
name = "tikv_client" name = "tikv_client"

4
src/raw/client.rs
View File

@ -759,7 +759,7 @@ impl<PdC: PdClient> Client<PdC> {
}); });
} }
let backoff = DEFAULT_STORE_BACKOFF; let backoff = DEFAULT_STORE_BACKOFF;
let range = range.into().encode_keyspace(self.keyspace, KeyMode::Raw); let mut range = range.into().encode_keyspace(self.keyspace, KeyMode::Raw);
let mut result = Vec::new(); let mut result = Vec::new();
let mut current_limit = limit; let mut current_limit = limit;
let (start_key, end_key) = range.clone().into_keys(); let (start_key, end_key) = range.clone().into_keys();
@ -788,6 +788,7 @@ impl<PdC: PdClient> Client<PdC> {
break; break;
} else { } else {
current_key = next_key; current_key = next_key;
range = BoundRange::new(std::ops::Bound::Included(current_key.clone()), range.to);
} }
} }
@ -805,7 +806,6 @@ impl<PdC: PdClient> Client<PdC> {
mut scan_args: ScanInnerArgs, mut scan_args: ScanInnerArgs,
) -> Result<(Option<RawScanResponse>, Key)> { ) -> Result<(Option<RawScanResponse>, Key)> {
let start_key = scan_args.start_key; let start_key = scan_args.start_key;
loop { loop {
let region = self.rpc.clone().region_for_key(&start_key).await?; let region = self.rpc.clone().region_for_key(&start_key).await?;
let store = self.rpc.clone().store_for_id(region.id()).await?; let store = self.rpc.clone().store_for_id(region.id()).await?;