fix: refactored logic for blobs.toc writing

Closes #102
GreenmaskIO · May 8, 2024 · e731ccc · e731ccc
1 parent 3d4c63b
commit e731ccc
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/internal/db/postgres/dumpers/large_object.go b/internal/db/postgres/dumpers/large_object.go
@@ -73,6 +73,16 @@ func (lod *BlobsDumper) Execute(ctx context.Context, tx pgx.Tx, st storages.Stor
 			Msg("dumping large object completed")
 	}
 
+	// Writing blobs.toc
+	if err := lod.generateBlobsToc(ctx, st); err != nil {
+		return nil, fmt.Errorf("cannot write large object blobs.toc: %w", err)
+	}
+
+	return lod.Blobs, nil
+}
+
+func (lod *BlobsDumper) generateBlobsToc(ctx context.Context, st storages.Storager) error {
+	log.Debug().Msg("writing blobs.toc")
 	// Writing blobs.toc
 	blobsTocBuf := bytes.NewBuffer(nil)
 
@@ -82,10 +92,9 @@ func (lod *BlobsDumper) Execute(ctx context.Context, tx pgx.Tx, st storages.Stor
 
 	err := st.PutObject(ctx, "blobs.toc", blobsTocBuf)
 	if err != nil {
-		return nil, fmt.Errorf("cannot write large object blobs.toc: %w", err)
+		return fmt.Errorf("cannot write large object blobs.toc: %w", err)
 	}
-
-	return lod.Blobs, nil
+	return nil
 }
 
 func largeObjectWriter(ctx context.Context, st storages.Storager, lo *entries.LargeObject, r countwriter.CountReadCloser) func() error {