Skip to content

Commit a37402a

Browse files
authored
Merge pull request #64 from codefuse-ai/xxh_dev
[Feat]Accelerate python extractor's db merge process
2 parents 86df6ab + 8b01eee commit a37402a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

Diff for: language/python/extractor/src/core/storage.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -438,7 +438,11 @@ def insert(self, type_object, object):
438438
object_dict.update({type_object: [object]})
439439

440440
def commit(self):
441+
batch_size = 50
441442
for key, value in object_dict.items():
442-
session.bulk_insert_mappings(key, value)
443-
session.commit()
443+
# batch insert
444+
for i in range(0, len(value), batch_size):
445+
batch = value[i:i + batch_size]
446+
session.bulk_insert_mappings(key, batch)
447+
session.commit()
444448
object_dict.clear()

0 commit comments

Comments
 (0)