Start9/start-os
2
0
Fork 0
mirror of https://github.com/Start9Labs/start-os.git synced 2026-03-31 04:23:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

include download errors in error handler

Browse Source
This commit is contained in:
Aiden McClelland
2022-01-20 17:43:03 -07:00
committed by Aiden McClelland
parent 0b1f544fd7
commit 3372da9f80
1 changed files with 50 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
appmgr/src/install/mod.rs
View File

@@ -114,6 +114,13 @@ pub async fn install(
.with_kind(crate::ErrorKind::Registry)?; .with_kind(crate::ErrorKind::Registry)?;
let man: Manifest = man_res.json().await.with_kind(crate::ErrorKind::Registry)?; let man: Manifest = man_res.json().await.with_kind(crate::ErrorKind::Registry)?;
if man.id.as_str() != id || !man.version.satisfies(&version) {
return Err(Error::new(
eyre!("Fetched package does not match requested id and version"),
ErrorKind::Registry,
));
}
let progress = InstallProgress::new(s9pk.content_length()); let progress = InstallProgress::new(s9pk.content_length());
let static_files = StaticFiles::remote(&man.id, &man.version); let static_files = StaticFiles::remote(&man.id, &man.version);
let mut db_handle = ctx.db.handle(); let mut db_handle = ctx.db.handle();
@@ -305,11 +312,11 @@ pub async fn sideload(
}; };
// gc the map // gc the map
let mut guard = ctx.rpc_stream_continuations.lock().await; let mut guard = ctx.rpc_stream_continuations.lock().await;
let gced = std::mem::take(&mut *guard) let garbage_collected = std::mem::take(&mut *guard)
.into_iter() .into_iter()
.filter(|(_, v)| v.created_at.elapsed() < Duration::from_secs(30)) .filter(|(_, v)| v.created_at.elapsed() < Duration::from_secs(30))
.collect::<BTreeMap<RequestGuid, RpcContinuation>>(); .collect::<BTreeMap<RequestGuid, RpcContinuation>>();
*guard = gced; *guard = garbage_collected;
drop(guard); drop(guard);
// insert the new continuation // insert the new continuation
ctx.rpc_stream_continuations ctx.rpc_stream_continuations
@@ -488,13 +495,15 @@ pub async fn download_install_s9pk(
let pkg_id = &temp_manifest.id; let pkg_id = &temp_manifest.id;
let version = &temp_manifest.version; let version = &temp_manifest.version;
if let Err(e) = async {
let pkg_archive_dir = ctx let pkg_archive_dir = ctx
.datadir .datadir
.join(PKG_ARCHIVE_DIR) .join(PKG_ARCHIVE_DIR)
.join(pkg_id) .join(pkg_id)
.join(version.as_str()); .join(version.as_str());
tokio::fs::create_dir_all(&pkg_archive_dir).await?; tokio::fs::create_dir_all(&pkg_archive_dir).await?;
let pkg_archive = pkg_archive_dir.join(AsRef::<Path>::as_ref(pkg_id).with_extension("s9pk")); let pkg_archive =
pkg_archive_dir.join(AsRef::<Path>::as_ref(pkg_id).with_extension("s9pk"));
let pkg_data_entry = crate::db::DatabaseModel::new() let pkg_data_entry = crate::db::DatabaseModel::new()
.package_data() .package_data()
@@ -528,30 +537,17 @@ pub async fn download_install_s9pk(
}) })
.await?; .await?;
install_s9pk_or_cleanup(&ctx, pkg_id, version, &mut s9pk_reader, progress).await?; install_s9pk(&ctx, pkg_id, version, &mut s9pk_reader, progress).await?;
Ok(()) Ok(())
} }
.await
#[instrument(skip(ctx, rdr))] {
pub async fn install_s9pk_or_cleanup<R: AsyncRead + AsyncSeek + Unpin>(
ctx: &RpcContext,
pkg_id: &PackageId,
version: &Version,
rdr: &mut S9pkReader<InstallProgressTracker<R>>,
progress: Arc<InstallProgress>,
) -> Result<(), Error> {
if let Err(e) = install_s9pk(ctx, pkg_id, version, rdr, progress).await {
let mut handle = ctx.db.handle(); let mut handle = ctx.db.handle();
let mut tx = handle.begin().await?; let mut tx = handle.begin().await?;
if let Err(e) = cleanup_failed(&ctx, &mut tx, pkg_id).await { if let Err(e) = cleanup_failed(&ctx, &mut tx, pkg_id).await {
tracing::error!( tracing::error!("Failed to clean up {}@{}: {}", pkg_id, version, e);
"Failed to clean up {}@{}: {}: Adding to broken packages",
pkg_id,
version,
e
);
tracing::debug!("{:?}", e); tracing::debug!("{:?}", e);
} else { } else {
tx.commit(None).await?; tx.commit(None).await?;