Fix generating on restart

Author: kamil-certat

intelmq/bots/outputs/misp/output_feed.py

@@ -111,6 +111,10 @@ def init(self):
                     self._tagging_objects[key].append(tag)
             self.logger.debug("Generated tags: %r.", self._tagging_objects)
 
+        if self.current_events and self.cache_length():
+            # Ensure we do generate feed on reload / restart
+            self._generate_feed()
+
     def _load_event(self, file_path: Path, key: str):
         if file_path.exists():
             self.current_events[key] = MISPEvent()

intelmq/lib/mixins/cache.py

@@ -15,15 +15,17 @@
 class CacheMixin:
     """Provides caching possibilities for bots
 
-        For key-value cache, use methods:
-            cache_exists
-            cache_get
-            cache_set
-
-        To store dict elements in a cache queue named after bot id, use methods:
-            cache_put
-            cache_pop
+    For key-value cache, use methods:
+        cache_exists
+        cache_get
+        cache_set
+
+    To store dict elements in a cache queue named after bot id, use methods:
+        cache_put
+        cache_pop
+        cache_length
     """
+
     __redis: redis.Redis = None
     redis_cache_host: str = "127.0.0.1"
     redis_cache_port: int = 6379
@@ -70,6 +72,9 @@ def cache_put(self, value: dict) -> int:
         size = self.__redis.lpush(self.bot_id, json.dumps(value))
         return size
 
+    def cache_length(self) -> int:
+        return self.__redis.llen(self.bot_id)
+
     def cache_pop(self) -> dict:
         data = self.__redis.rpop(self.bot_id)
         if data is None: