Return table upload error instead of semaphore error

feat: return upload error instead of context canceled commit_hash:0936bcfcce88efe8858262f9d184c7f9ae0e8dc8
doublecloud · Sep 19, 2024 · b72216a · b72216a
1 parent 319a680
commit b72216a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/worker/tasks/load_snapshot.go b/pkg/worker/tasks/load_snapshot.go
@@ -872,6 +872,9 @@ func (l *SnapshotLoader) DoUploadTables(ctx context.Context, source abstract.Sto
 	for {
 		if err := parallelismSemaphore.Acquire(ctx, 1); err != nil {
 			logger.Log.Error("Failed to acquire semaphore to load next table", log.Any("worker_index", l.workerIndex), log.Error(ctx.Err()))
+			if tableUploadErr != nil {
+				return errors.CategorizedErrorf(categories.Internal, "Upload error: %w", err)
+			}
 			return errors.CategorizedErrorf(categories.Internal, "failed to acquire semaphore: %w", err)
 		}
 		waitToComplete.Add(1)