Remove unwrap

solana-labs · Apr 25, 2023 · 448c662 · 448c662
1 parent ec89de8
commit 448c662
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/runtime/src/snapshot_package.rs b/runtime/src/snapshot_package.rs
@@ -8,7 +8,9 @@ use {
         rent_collector::RentCollector,
         snapshot_archive_info::{SnapshotArchiveInfo, SnapshotArchiveInfoGetter},
         snapshot_hash::SnapshotHash,
-        snapshot_utils::{self, ArchiveFormat, BankSnapshotInfo, Result, SnapshotVersion},
+        snapshot_utils::{
+            self, ArchiveFormat, BankSnapshotInfo, Result, SnapshotError, SnapshotVersion,
+        },
     },
     log::*,
     solana_sdk::{clock::Slot, feature_set, sysvar::epoch_schedule::EpochSchedule},
@@ -73,10 +75,10 @@ impl AccountsPackage {
             }
         }
 
-        let snapshot_links = bank_snapshot_info
-            .snapshot_dir
+        let bank_snapshot_dir = &bank_snapshot_info.snapshot_dir;
+        let snapshot_links = bank_snapshot_dir
             .parent()
-            .unwrap()
+            .ok_or_else(|| SnapshotError::InvalidSnapshotDirPath(bank_snapshot_dir.to_path_buf()))?
             .to_path_buf();
 
         let snapshot_info = SupplementalSnapshotInfo {