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