File tree 1 file changed +3
-2
lines changed
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -858,12 +858,12 @@ def __exit__(
858
858
self ._current_writer .__exit__ (exc_type , exc_value , traceback )
859
859
860
860
def _get_current_writer (self ) -> ManifestWriter :
861
+ if self ._should_roll_to_new_file ():
862
+ self ._close_current_writer ()
861
863
if not self ._current_writer :
862
864
self ._current_writer = next (self ._supplier )
863
865
self ._current_writer .__enter__ ()
864
866
return self ._current_writer
865
- if self ._should_roll_to_new_file ():
866
- self ._close_current_writer ()
867
867
return self ._current_writer
868
868
869
869
def _should_roll_to_new_file (self ) -> bool :
@@ -890,6 +890,7 @@ def add_entry(self, entry: ManifestEntry) -> RollingManifestWriter:
890
890
if self ._closed :
891
891
raise RuntimeError ("Cannot add entry to closed manifest writer" )
892
892
self ._get_current_writer ().add_entry (entry )
893
+ self ._current_file_rows += entry .data_file .record_count
893
894
return self
894
895
895
896
You can’t perform that action at this time.
0 commit comments