Merge pull request #26 from sendbird/bugfix/update-last-pk-inserted-o…

…n-duplicate-key Fix bug on duplicate key
sendbird · Nov 17, 2024 · 38d9d9a · 38d9d9a
2 parents 42b071f + af8fa29
commit 38d9d9a
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