Merge pull request #19 from atugushev/increase-chunk-size

Increase chunk size for IO stream reader
list-family · Jul 23, 2020 · b4312f9 · b4312f9
2 parents 4b9716e + 859b92a
commit b4312f9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ansq/tcp/connection.py b/ansq/tcp/connection.py
@@ -6,6 +6,7 @@
 from time import time
 from typing import Optional, Callable, Any, Union, Generator
 
+from ansq.tcp import consts
 from ansq.tcp.exceptions import ProtocolError, get_exception, NSQUnauthorized
 from ansq.tcp.types import (
     TCPConnection as NSQConnectionBase, ConnectionStatus, NSQMessage,
@@ -218,7 +219,7 @@ async def _read_data_task(self):
         """Response reader task."""
         while not self._reader.at_eof():
             try:
-                data = await self._reader.read(52)
+                data = await self._reader.read(consts.MAX_CHUNK_SIZE)
             except asyncio.CancelledError:
                 # useful during update to TLS, task canceled but connection
                 # should not be closed

diff --git a/ansq/tcp/consts.py b/ansq/tcp/consts.py
@@ -7,3 +7,4 @@
 ATTEMPTS_SIZE = 2
 MSG_ID_SIZE = 16
 MSG_HEADER = TIMESTAMP_SIZE + ATTEMPTS_SIZE + MSG_ID_SIZE
+MAX_CHUNK_SIZE = 4096