extern/atuin
1
0
Fork 0
mirror of https://github.com/atuinsh/atuin.git synced 2024-12-28 18:09:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Don't waste requests

Browse Source
This commit is contained in:
Ellie Huxtable 2023-07-09 14:59:58 +01:00
parent 5ea3530306
commit 9626e35c06
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
atuin-client/src/record/sync.rs
View File

@ -139,7 +139,7 @@ async fn sync_download(
op: (Uuid, String, Uuid), op: (Uuid, String, Uuid),
) -> Result<i64> { ) -> Result<i64> {
// TODO(ellie): implement variable page sizing like on history sync // TODO(ellie): implement variable page sizing like on history sync
let download_page_size = 1; let download_page_size = 1000;
let mut total = 0; let mut total = 0;
@ -161,20 +161,33 @@ async fn sync_download(
println!("Downloading {:?}/{}/{:?} to local", op.0, op.1, op.2); println!("Downloading {:?}/{}/{:?} to local", op.0, op.1, op.2);
let mut records = client let mut records = client
.next_records(op.0, op.1.clone(), local_tail.map(|r| r.id), 1) .next_records(
op.0,
op.1.clone(),
local_tail.map(|r| r.id),
download_page_size,
)
.await?; .await?;
while records.len() > 0 { while records.len() > 0 {
total += std::cmp::min(download_page_size, records.len() as u64);
store.push_batch(records.iter()).await?; store.push_batch(records.iter()).await?;
records = client if records.last().unwrap().id == remote_tail {
.next_records(op.0, op.1.clone(), records.last().map(|r| r.id), 1) break;
.await?; }
total += download_page_size; records = client
.next_records(
op.0,
op.1.clone(),
records.last().map(|r| r.id),
download_page_size,
)
.await?;
} }
Ok(total) Ok(total as i64)
} }
pub async fn sync_remote( pub async fn sync_remote(