fix bug on duplicate key

sendbird · Nov 17, 2024 · af8fa29 · af8fa29
1 parent 42b071f
commit af8fa29
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/sbosc/worker/worker.py b/src/sbosc/worker/worker.py
@@ -116,6 +116,10 @@ def bulk_import(self):
             start_pk = self.get_start_pk(chunk_info)
             if start_pk is None:
                 return
+            # If the start_pk is greater than the end_pk, set the last_pk_inserted to end_pk
+            # This can happen when chunk ended with a duplicate key error
+            elif start_pk > chunk_info.end_pk:
+                chunk_info.last_pk_inserted = chunk_info.end_pk
 
             end_pk = chunk_info.end_pk
             chunk_info.status = ChunkStatus.IN_PROGRESS