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

chore: opt code under persistent_cache_task (#1098) 

chroe: opt code under persistent_cache_task

Signed-off-by: cormick <cormick1080@gmail.com>
This commit is contained in:
KennyMcCormick 2025-04-16 21:02:47 +08:00 committed by GitHub
parent ad335784fe
commit 938d17c0cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
dragonfly-client/src/resource/persistent_cache_task.rs
View File

@ -105,7 +105,7 @@ impl PersistentCacheTask {
id_generator,
storage,
scheduler_client,
piece: piece.clone(),
piece,
})
}
@ -129,8 +129,9 @@ impl PersistentCacheTask {
let ttl = Duration::try_from(request.ttl.ok_or(Error::UnexpectedResponse)?)
.or_err(ErrorType::ParseError)?;
// Get the content length of the file.
let content_length = std::fs::metadata(path.as_path())
// Get the content length of the file asynchronously.
let content_length = tokio::fs::metadata(path.as_path())
.await
.inspect_err(|err| {
error!("get file metadata error: {}", err);
})?
@ -150,8 +151,7 @@ impl PersistentCacheTask {
};
// Notify the scheduler that the persistent cache task is started.
match self
.scheduler_client
self.scheduler_client
.upload_persistent_cache_task_started(UploadPersistentCacheTaskStartedRequest {
host_id: host_id.to_string(),
task_id: task_id.to_string(),
@ -167,13 +167,7 @@ impl PersistentCacheTask {
ttl: request.ttl,
})
.await
{
Ok(_) => {}
Err(err) => {
error!("upload persistent cache task started: {}", err);
return Err(err);
}
}
.inspect_err(|err| error!("upload persistent cache task started: {}", err))?;
// Check if the storage has enough space to store the persistent cache task.
let has_enough_space = self.storage.has_enough_space(content_length)?;