From d66c8013500379c84a01047097eb18a1765478d0 Mon Sep 17 00:00:00 2001
From: Victor Shyba <victor1984@riseup.net>
Date: Tue, 17 Dec 2019 23:00:28 -0300
Subject: [PATCH] wake up mempool on broadcast

---
 torba/torba/server/mempool.py | 15 ++++++++++-----
 torba/torba/server/session.py |  1 +
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/torba/torba/server/mempool.py b/torba/torba/server/mempool.py
index fc2c96c7e..62449e40a 100644
--- a/torba/torba/server/mempool.py
+++ b/torba/torba/server/mempool.py
@@ -11,7 +11,6 @@ import asyncio
 import itertools
 import time
 from abc import ABC, abstractmethod
-from asyncio import Lock, sleep
 from collections import defaultdict
 
 import attr
@@ -104,7 +103,8 @@ class MemPool:
         self.refresh_secs = refresh_secs
         self.log_status_secs = log_status_secs
         # Prevents mempool refreshes during fee histogram calculation
-        self.lock = Lock()
+        self.lock = asyncio.Lock()
+        self.wakeup = asyncio.Event()
 
     async def _logging(self, synchronized_event):
         """Print regular logs of mempool stats."""
@@ -117,7 +117,7 @@ class MemPool:
         while True:
             self.logger.info(f'{len(self.txs):,d} txs '
                              f'touching {len(self.hashXs):,d} addresses')
-            await sleep(self.log_status_secs)
+            await asyncio.sleep(self.log_status_secs)
             await synchronized_event.wait()
 
     async def _refresh_histogram(self, synchronized_event):
@@ -126,7 +126,7 @@ class MemPool:
             async with self.lock:
                 # Threaded as can be expensive
                 await asyncio.get_event_loop().run_in_executor(None, self._update_histogram, 100_000)
-            await sleep(self.coin.MEMPOOL_HISTOGRAM_REFRESH_SECS)
+            await asyncio.sleep(self.coin.MEMPOOL_HISTOGRAM_REFRESH_SECS)
 
     def _update_histogram(self, bin_size):
         # Build a histogram by fee rate
@@ -212,7 +212,12 @@ class MemPool:
             synchronized_event.set()
             synchronized_event.clear()
             await self.api.on_mempool(touched, height)
-            await sleep(self.refresh_secs)
+            try:
+                await asyncio.wait_for(self.wakeup.wait(), timeout=self.refresh_secs)
+            except asyncio.TimeoutError:
+                pass
+            finally:
+                self.wakeup.clear()
 
     async def _process_mempool(self, all_hashes):
         # Re-sync with the new set of hashes
diff --git a/torba/torba/server/session.py b/torba/torba/server/session.py
index d9252d350..fe78133bc 100644
--- a/torba/torba/server/session.py
+++ b/torba/torba/server/session.py
@@ -567,6 +567,7 @@ class SessionManager:
 
     async def broadcast_transaction(self, raw_tx):
         hex_hash = await self.daemon.broadcast_transaction(raw_tx)
+        self.mempool.wakeup.set()
         self.txs_sent += 1
         return hex_hash