feat(resource): support batch in BulkUpsert (#71)

* feat(resource): support batch in BulkUpsert * fix: use db session to fix not working batchinsert
goto · Sep 16, 2023 · 2a882f3 · 2a882f3
1 parent 3bd8e74
commit 2a882f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/store/postgres/resource_repository.go b/internal/store/postgres/resource_repository.go
@@ -115,7 +115,9 @@ func (r *ResourceRepository) BulkUpsert(ctx context.Context, resources []*domain
 	if len(models) > 0 {
 		return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
 			// upsert clause is moved to model.Resource.BeforeCreate() (gorm's hook) to apply the same for associations (model.Resource.Children)
-			if err := r.db.Create(models).Error; err != nil {
+			if err := r.db.
+				Session(&gorm.Session{CreateBatchSize: 1000}).
+				Create(models).Error; err != nil {
 				return err
 			}