historian: don't crash on rabbitmq stream loss

Author: endothermicdev

archived/historian/historian.py

@@ -297,22 +297,30 @@ def publish(self, raw: bytes) -> None:
                          gossipd.ChannelAnnouncement,
                          gossipd.NodeAnnouncement]:
             if isinstance(msg, msg_type):
-                self.channel.basic_publish(exchange='router.gossip',
-                                           # unused by fanout exchange
-                                           routing_key='',
-                                           body=serialize(raw, self.node_id,
-                                                          self.network))
+                try:
+                    self.channel.basic_publish(exchange='router.gossip',
+                                               # unused by fanout exchange
+                                               routing_key='',
+                                               body=serialize(raw, self.node_id,
+                                                              self.network))
+
+                except pika.exceptions.StreamLostError:
+                    plugin.log("lost connection to rabbitmq, reconnecting")
+                    self.rabbitmq_connect()
                 return
 
 
 
 @plugin.init()
 def init(plugin, configuration, options):
     print(options)
-    with create_engine(options['historian-dsn'], echo=False) as engine:
+    try:
+        engine = create_engine(options['historian-dsn'], echo=False)
         Base.metadata.create_all(engine)
         plugin.engine = engine
         Flusher(engine).start()
+    finally:
+        engine.dispose()
 
 
 @plugin.method('historian-stats')