add options to lnd client for max message length

lightning-power-users · Jan 12, 2019 · a8e15a0 · a8e15a0
1 parent 6d7e0f7
commit a8e15a0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/node_launcher/node_set/lnd_client/lnd_client.py b/node_launcher/node_set/lnd_client/lnd_client.py
@@ -31,6 +31,11 @@ def __init__(self, lnd: Lnd = None, lnddir: str = None,
         self._lnd_client = None
         self._wallet_unlocker = None
 
+        self.grpc_options = [
+            ('grpc.max_receive_message_length', 33554432),
+            ('grpc.max_send_message_length', 33554432),
+        ]
+
     def reset(self):
         self._lnd_client = None
         self._wallet_unlocker = None
@@ -46,8 +51,11 @@ def lnd_client(self):
             self.get_cert_credentials(),
             auth_credentials)
 
-        grpc_channel = grpc.secure_channel(f'{self.grpc_host}:{self.grpc_port}',
-                                           credentials)
+        grpc_channel = grpc.secure_channel(
+            f'{self.grpc_host}:{self.grpc_port}',
+            credentials,
+            options=self.grpc_options
+        )
         self._lnd_client = lnrpc.LightningStub(grpc_channel)
         return self._lnd_client
 
@@ -56,8 +64,10 @@ def wallet_unlocker(self):
         if self._wallet_unlocker is not None:
             return self._wallet_unlocker
 
-        grpc_channel = grpc.secure_channel(f'{self.grpc_host}:{self.grpc_port}',
-                                           credentials=self.get_cert_credentials())
+        grpc_channel = grpc.secure_channel(
+            f'{self.grpc_host}:{self.grpc_port}',
+            credentials=self.get_cert_credentials()
+        )
         self._wallet_unlocker = lnrpc.WalletUnlockerStub(grpc_channel)
         return self._wallet_unlocker
 

diff --git a/tools/channel_balancer.py b/tools/channel_balancer.py
@@ -317,8 +317,8 @@ def get_google_sheet_data(self):
 
 if __name__ == '__main__':
     channel_balancer = ChannelBalancer()
+    channel_balancer.reconnect()
     channel_balancer.get_channels()
-    # channel_balancer.reconnect()
     channel_balancer.get_google_sheet_data()
 
     # response = lnd_client.open_channel(