From 36080d67712a33a3585b1e3bfed07f4f929dd9e6 Mon Sep 17 00:00:00 2001
From: Rusty Russell <rusty@rustcorp.com.au>
Date: Wed, 6 Apr 2022 13:49:29 +0930
Subject: [PATCH] commando: discard collated requests upon peer disconnect.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
---
 commando/commando.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/commando/commando.py b/commando/commando.py
index 079af25..bdf006e 100755
--- a/commando/commando.py
+++ b/commando/commando.py
@@ -237,6 +237,12 @@ def on_custommsg(peer_id, payload, plugin, request, **kwargs):
     request.set_result({'result': 'continue'})
 
 
+@plugin.subscribe('disconnect')
+def on_disconnect(id, plugin, request, **kwargs):
+    if id in plugin.in_reqs:
+        del plugin.in_reqs[id]
+
+
 @plugin.async_method("commando")
 def commando(plugin, request, peer_id, method, params=None, rune=None):
     """Send a command to node_id, and wait for a response"""