From 6d215292bdecfd6fb6d243206ca8cd0f21666915 Mon Sep 17 00:00:00 2001
From: Pakin <pakin.t@forth.co.th>
Date: Mon, 11 Aug 2025 14:40:26 +0700
Subject: [PATCH] add more unregisters

---
 src/storage.rs | 2 +-
 src/updater.rs | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/storage.rs b/src/storage.rs
index 90da6ed..c31aa95 100644
--- a/src/storage.rs
+++ b/src/storage.rs
@@ -73,7 +73,7 @@ impl Storage {
             .remove(key.as_bytes())
             .context("Failed to delete container from storage")?;
 
-        debug!("🗑️ Deleted container: {}", container_id);
+        info!("🗑️ Deleted container: {}", container_id);
         Ok(())
     }
 }
diff --git a/src/updater.rs b/src/updater.rs
index c466a87..c4212cf 100644
--- a/src/updater.rs
+++ b/src/updater.rs
@@ -129,7 +129,9 @@ impl UpdateManager {
         if let Ok(mut container) = self.storage.get_container(container_id).await {
             container.status = ContainerStatus::Lost;
             container.last_seen = Utc::now();
-            self.storage.save_container(&container).await?;
+            self.storage
+                .delete_container(&container.container_id)
+                .await?;
 
             info!("📤 Marked container {} as lost", container.name);