File tree Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Original file line number Diff line number Diff line change @@ -377,26 +377,27 @@ def write(self):
377
377
)
378
378
379
379
def move_iterator (self , it , callback ):
380
- try :
381
- next_f = next (it )
382
- next_f .add_done_callback (lambda x : callback (it , x ))
383
- except StopIteration :
384
- return
380
+ next_f = next (it )
381
+ next_f .add_done_callback (lambda x : callback (it , x ))
385
382
386
383
def on_read_table_chunk (self , it , f ):
387
384
try :
388
385
f .result ()
389
386
self .stats .save_event (EventKind .READ_TABLE_CHUNK )
390
387
except ydb .Error as e :
391
- self .stats .save_event (EventKind .READ_TABLE_CHUNK , e )
388
+ self .stats .save_event (EventKind .READ_TABLE_CHUNK , e .status )
389
+ except StopIteration :
390
+ return
392
391
self .move_iterator (it , self .on_read_table_chunk )
393
392
394
393
def on_scan_query_chunk (self , it , f ):
395
394
try :
396
395
f .result ()
397
396
self .stats .save_event (EventKind .SCAN_QUERY_CHUNK )
398
397
except ydb .Error as e :
399
- self .stats .save_event (EventKind .SCAN_QUERY_CHUNK , e )
398
+ self .stats .save_event (EventKind .SCAN_QUERY_CHUNK , e .status )
399
+ except StopIteration :
400
+ return
400
401
401
402
self .move_iterator (it , self .on_scan_query_chunk )
402
403
You can’t perform that action at this time.
0 commit comments