diff --git a/docs/examples/python/highvolume_example.py b/docs/examples/python/highvolume_example.py index 00d593596510fa62495e5b8cba466cd868cf95fc..8c964395a428fca71bb9b26fa0cbf91f803a8377 100644 --- a/docs/examples/python/highvolume_example.py +++ b/docs/examples/python/highvolume_example.py @@ -182,14 +182,16 @@ class SQLWriter: def run_read_task(task_id: int, task_queues: List[Queue]): table_count_per_task = TABLE_COUNT // READ_TASK_COUNT data_source = MockDataSource(f"tb{task_id}", table_count_per_task) - for table_id, line in data_source: - i = table_id % len(task_queues) - task_queues[i].put(line, block=True) + try: + for table_id, line in data_source: + i = table_id % len(task_queues) + task_queues[i].put(line, block=True) + except KeyboardInterrupt: + pass # ANCHOR_END: read - # ANCHOR: write def run_write_task(task_id: int, queue: Queue): log = logging.getLogger(f"WriteTask-{task_id}") @@ -204,6 +206,8 @@ def run_write_task(task_id: int, queue: Queue): writer.flush() else: time.sleep(0.01) + except KeyboardInterrupt: + pass except BaseException as e: msg = f"line={line}, buffer_count={writer.buffered_count}" log.debug(msg)