fix: unregister store monitor on unsubscribe

DIG-Network · Oct 28, 2024 · 4b7efed · 4b7efed
1 parent 4b970a5
commit 4b7efed
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/controllers/storeController.ts b/src/controllers/storeController.ts
@@ -9,6 +9,7 @@ import {
   DigPeer,
   ServerCoin,
   NconfManager,
+  StoreMonitorRegistry
 } from "@dignetwork/dig-sdk";
 import { getStorageLocation } from "../utils/storage";
 import requestIp from 'request-ip';
@@ -66,10 +67,16 @@ export const unsubscribeToStore = async (
   try {
     const { storeId } = req.body;
 
+    const storeMonitorRegistryInstance = StoreMonitorRegistry.getInstance();
+    storeMonitorRegistryInstance.unregisterStore(storeId);
+
     const pathToStore = path.resolve(getStorageLocation(), "stores", storeId);
 
     // Remove the store directory
-    setTimeout(() => fs.rmdirSync(pathToStore, { recursive: true }), 0);
+    setTimeout(() => {
+      fs.rmdirSync(pathToStore, { recursive: true });
+      storeMonitorRegistryInstance.unregisterStore(storeId);
+    }, 0);
 
     res.status(200).json({ message: `Unsubscribed to store ${storeId}` });
   } catch (error) {