mirror of
https://github.com/atuinsh/atuin.git
synced 2024-12-28 18:09:25 +01:00
Use a page size for uploads, make it variable later
This commit is contained in:
parent
9626e35c06
commit
8f0eda013e
@ -88,6 +88,7 @@ async fn sync_upload(
|
|||||||
client: &Client<'_>,
|
client: &Client<'_>,
|
||||||
op: (Uuid, String, Uuid), // just easier to reason about this way imo
|
op: (Uuid, String, Uuid), // just easier to reason about this way imo
|
||||||
) -> Result<i64> {
|
) -> Result<i64> {
|
||||||
|
let upload_page_size = 100;
|
||||||
let mut total = 0;
|
let mut total = 0;
|
||||||
|
|
||||||
// so. we have an upload operation, with the tail representing the state
|
// so. we have an upload operation, with the tail representing the state
|
||||||
@ -118,17 +119,28 @@ async fn sync_upload(
|
|||||||
// remote tail = current local tail
|
// remote tail = current local tail
|
||||||
|
|
||||||
let mut record = Some(store.get(start).await.unwrap());
|
let mut record = Some(store.get(start).await.unwrap());
|
||||||
|
record = store.next(&record.unwrap()).await?;
|
||||||
|
|
||||||
|
let mut buf = Vec::with_capacity(upload_page_size);
|
||||||
|
|
||||||
// We are currently uploading them one at a time. Yes, this sucks. We are
|
|
||||||
// also processing all records in serial. That also sucks.
|
|
||||||
// Once sync works, we can then make it super fast.
|
|
||||||
while let Some(r) = record {
|
while let Some(r) = record {
|
||||||
client.post_records(&[r.clone()]).await?;
|
if buf.len() < upload_page_size {
|
||||||
|
buf.push(r.clone());
|
||||||
|
} else {
|
||||||
|
client.post_records(&buf).await?;
|
||||||
|
|
||||||
|
// can we reset what we have? len = 0 but keep capacity
|
||||||
|
buf = Vec::with_capacity(upload_page_size);
|
||||||
|
}
|
||||||
record = store.next(&r).await?;
|
record = store.next(&r).await?;
|
||||||
|
|
||||||
total += 1;
|
total += 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if buf.len() > 0 {
|
||||||
|
client.post_records(&buf).await?;
|
||||||
|
}
|
||||||
|
|
||||||
Ok(total)
|
Ok(total)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user