From 2a02de02474f5cfcef3c347570f3821ff9c4a850 Mon Sep 17 00:00:00 2001
From: Denis Mazur <denismazur8@gmail.com>
Date: Thu, 29 Jul 2021 18:19:30 +0300
Subject: [PATCH] Add environment variable for DHT max workers

---
 hivemind/dht/__init__.py |  2 +-
 hivemind/dht/node.py     | 11 +++++++++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/hivemind/dht/__init__.py b/hivemind/dht/__init__.py
index fe045f077..ab764e532 100644
--- a/hivemind/dht/__init__.py
+++ b/hivemind/dht/__init__.py
@@ -106,7 +106,7 @@ def run(self) -> None:
             async def _run():
                 self._node = await DHTNode.create(
                     initial_peers=self.initial_peers,
-                    num_workers=self.max_workers or 1,
+                    num_workers=self.max_workers,
                     record_validator=self._record_validator,
                     **self.kwargs,
                 )
diff --git a/hivemind/dht/node.py b/hivemind/dht/node.py
index 1354fc82e..eb10021e1 100644
--- a/hivemind/dht/node.py
+++ b/hivemind/dht/node.py
@@ -2,6 +2,7 @@
 
 import asyncio
 import dataclasses
+import os
 import random
 from collections import defaultdict, Counter
 from dataclasses import dataclass, field
@@ -154,7 +155,7 @@ async def create(
         :param backoff_rate: blacklist time will be multiplied by :backoff_rate: for each successive non-response
         :param validate: if True, use initial peers to validate that this node is accessible and synchronized
         :param strict: if True, any error encountered in validation will interrupt the creation of DHTNode
-        :param client_mode: if False (default), this node will accept incoming requests as a full DHT "citzen"
+        :param client_mode: if False (default), this node will accept incoming requests as a full DHT "citizen"
           if True, this node will refuse any incoming requests, effectively being only a client
         :param record_validator: instance of RecordValidatorBase used for signing and validating stored records
         :param authorizer: instance of AuthorizerBase used for signing and validating requests and response
@@ -164,7 +165,13 @@ async def create(
         """
         self = cls(_initialized_with_create=True)
         self.node_id = node_id if node_id is not None else DHTID.generate()
-        self.num_replicas, self.num_workers, self.chunk_size = num_replicas, num_workers, chunk_size
+
+        if num_workers is None:
+            self.num_workers = int(os.environ.get("HIVEMIND_DHT_NUM_WORKERS", 8))
+        else:
+            self.num_workers = num_workers
+
+        self.num_replicas, self.chunk_size = num_replicas, chunk_size
         self.is_alive = True  # if set to False, cancels all background jobs such as routing table refresh
 
         self.reuse_get_requests = reuse_get_requests