hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-09 13:29:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Set a known buffer size if possible on BinaryWriter()s

Browse Source
This commit is contained in:
Lonami Exo
2017-08-29 20:25:49 +02:00
parent 7ad2ece168
commit b4811261e9
4 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/extensions/binary_writer.py
View File

@@ -1,4 +1,4 @@
from io import BufferedWriter, BytesIO
from io import BufferedWriter, BytesIO, DEFAULT_BUFFER_SIZE
from struct import pack
@@ -8,11 +8,16 @@ class BinaryWriter:
Also creates a "Memory Stream" if necessary
"""
def __init__(self, stream=None):
def __init__(self, stream=None, known_length=None):
if not stream:
stream = BytesIO()
self.writer = BufferedWriter(stream)
if known_length is None:
# On some systems, DEFAULT_BUFFER_SIZE defaults to 8192
# That's over 16 times as big as necessary for most messages
known_length = max(DEFAULT_BUFFER_SIZE, 1024)
self.writer = BufferedWriter(stream, buffer_size=known_length)
self.written_count = 0
# region Writing