Skip to content

Commit

Permalink
speedup with bytearray
Browse files Browse the repository at this point in the history
  • Loading branch information
ppigazzini committed Apr 25, 2024
1 parent a686aab commit ebd12da
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions server/fishtest/rundb.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def get_port():
class GeneratorToFileLikeWrapper:
def __init__(self, generator):
self.generator = generator
self.buffer = b""
self.buffer = bytearray()

def read(self, size=-1):
while size < 0 or len(self.buffer) < size:
Expand All @@ -91,7 +91,7 @@ def read(self, size=-1):
except StopIteration:
break
result, self.buffer = self.buffer[:size], self.buffer[size:]
return result
return bytes(result)

def close(self):
pass # No cleanup needed, but method is required
Expand Down

0 comments on commit ebd12da

Please sign in to comment.