Skip to content

Commit cb03466

Browse files
clean up streaming source on exception
1 parent 7a62f64 commit cb03466

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

clickhouse_connect/driver/aiohttp_client.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -568,7 +568,11 @@ def parse_streaming():
568568
return result
569569

570570
# Run parser in executor (pulls from queue, decompresses & parses)
571-
query_result = await loop.run_in_executor(None, parse_streaming)
571+
try:
572+
query_result = await loop.run_in_executor(None, parse_streaming)
573+
except Exception:
574+
await streaming_source.aclose()
575+
raise
572576
query_result.summary = self._summary(response)
573577

574578
# Attach streaming_source to query_result.source to ensure it gets closed

0 commit comments

Comments
 (0)