mirror of
https://github.com/simon987/hexlib.git
synced 2025-04-22 02:56:43 +00:00
Compare commits
No commits in common. "084acbe1849792074842e569a86d5f2e41816e34" and "cd5a1ac50c09c48c3b1feee20f64a4af71a89bb1" have entirely different histories.
084acbe184
...
cd5a1ac50c
@ -34,7 +34,7 @@ class StatelessStreamWorker:
|
||||
|
||||
|
||||
class StatelessStreamProcessor:
|
||||
def __init__(self, worker_factory, chunk_size=128, processes=1, timeout=60):
|
||||
def __init__(self, worker_factory, chunk_size=128, processes=1):
|
||||
self._chunk_size = 128
|
||||
self._queue = MPQueue(maxsize=chunk_size)
|
||||
self._queue_out = MPQueue(maxsize=processes * 2)
|
||||
@ -42,7 +42,6 @@ class StatelessStreamProcessor:
|
||||
self._processes = []
|
||||
self._factory = worker_factory
|
||||
self._workers = []
|
||||
self._timeout = timeout
|
||||
|
||||
if processes > 1:
|
||||
for _ in range(processes):
|
||||
@ -68,7 +67,7 @@ class StatelessStreamProcessor:
|
||||
ingest_thread = Thread(target=self._ingest, args=(iterable,))
|
||||
ingest_thread.start()
|
||||
|
||||
for results in queue_iter(self._queue_out, joinable=False, timeout=self._timeout):
|
||||
for results in queue_iter(self._queue_out, joinable=False, timeout=10):
|
||||
yield from results
|
||||
|
||||
ingest_thread.join()
|
||||
|
@ -85,7 +85,7 @@ def ndjson_iter(*files, compression=""):
|
||||
line_iter = BufferedReader(gzip.open(file))
|
||||
elif compression == COMPRESSION_ZSTD:
|
||||
fp = open(file, "rb")
|
||||
dctx = zstandard.ZstdDecompressor(max_window_size=2147483648)
|
||||
dctx = zstandard.ZstdDecompressor()
|
||||
reader = dctx.stream_reader(fp)
|
||||
line_iter = BufferedReader(reader)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user