fix: use db session to fix not working batchinsert

raystack · Sep 16, 2023 · 61e62d7 · 61e62d7
1 parent c3d93eb
commit 61e62d7
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.CreateInBatches(models, 1000).Error; err != nil {
+ if err := r.db.
+ Session(&gorm.Session{CreateBatchSize: 1000}).
+ Create(models).Error; err != nil {
  return err
  }