mirrors/lnbits-legend
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits-legend.git synced 2025-02-22 14:22:55 +01:00
Code Issues Projects Releases Packages Wiki Activity

make it so LNbitsWallet reconnects if the listener stream goes off.

Browse source
This commit is contained in:
fiatjaf 2021-04-18 10:29:22 -03:00
parent 42b3359d12
commit 3d489bf2ee
1 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
lnbits/wallets/lnbits.py
View file

@ -1,3 +1,4 @@
import trio # type: ignore
import json
import httpx
from os import getenv
@ -116,16 +117,25 @@ class LNbitsWallet(Wallet):
async def paid_invoices_stream(self) -> AsyncGenerator[str, None]:
url = f"{self.endpoint}/api/v1/payments/sse"
async with httpx.AsyncClient(timeout=None, headers=self.key) as client:
async with client.stream("GET", url) as r:
async for line in r.aiter_lines():
if line.startswith("data:"):
try:
data = json.loads(line[5:])
except json.decoder.JSONDecodeError:
continue
while True:
try:
async with httpx.AsyncClient(timeout=None, headers=self.key) as client:
async with client.stream("GET", url) as r:
async for line in r.aiter_lines():
if line.startswith("data:"):
if type(data) is not dict:
continue
try:
data = json.loads(line[5:])
except json.decoder.JSONDecodeError:
continue
yield data["payment_hash"] # payment_hash
if type(data) is not dict:
continue
yield data["payment_hash"] # payment_hash
except (OSError, httpx.ReadError, httpx.ConnectError):
pass
print("lost connection to lnbits /payments/sse, retrying in 5 seconds")
await trio.sleep(5)