introduce benchmark for declcfg.LoadFS

Signed-off-by: Joe Lanford <[email protected]>

Author: joelanford

alpha/action/migrate.go

@@ -1,13 +1,10 @@
 package action
 
 import (
-	"bytes"
 	"context"
 	"fmt"
-	"io"
 	"io/ioutil"
 	"os"
-	"path/filepath"
 
 	"github.com/operator-framework/operator-registry/alpha/declcfg"
 	"github.com/operator-framework/operator-registry/pkg/image"
@@ -16,14 +13,13 @@ import (
 type Migrate struct {
 	CatalogRef string
 	OutputDir  string
+	FS         declcfg.WriterFS
 
-	WriteFunc WriteFunc
+	WriteFunc declcfg.WriteFunc
 	FileExt   string
 	Registry  image.Registry
 }
 
-type WriteFunc func(config declcfg.DeclarativeConfig, w io.Writer) error
-
 func (m Migrate) Run(ctx context.Context) error {
 	entries, err := ioutil.ReadDir(m.OutputDir)
 	if err != nil && !os.IsNotExist(err) {
@@ -52,48 +48,10 @@ func (m Migrate) Run(ctx context.Context) error {
 		return fmt.Errorf("render catalog image: %w", err)
 	}
 
-	return writeToFS(*cfg, m.OutputDir, m.WriteFunc, m.FileExt)
-}
-
-func writeToFS(cfg declcfg.DeclarativeConfig, rootDir string, writeFunc WriteFunc, fileExt string) error {
-	channelsByPackage := map[string][]declcfg.Channel{}
-	for _, c := range cfg.Channels {
-		channelsByPackage[c.Package] = append(channelsByPackage[c.Package], c)
-	}
-	bundlesByPackage := map[string][]declcfg.Bundle{}
-	for _, b := range cfg.Bundles {
-		bundlesByPackage[b.Package] = append(bundlesByPackage[b.Package], b)
-	}
-
-	if err := os.MkdirAll(rootDir, 0777); err != nil {
-		return err
-	}
-
-	for _, p := range cfg.Packages {
-		fcfg := declcfg.DeclarativeConfig{
-			Packages: []declcfg.Package{p},
-			Channels: channelsByPackage[p.Name],
-			Bundles:  bundlesByPackage[p.Name],
-		}
-		pkgDir := filepath.Join(rootDir, p.Name)
-		if err := os.MkdirAll(pkgDir, 0777); err != nil {
-			return err
-		}
-		filename := filepath.Join(pkgDir, fmt.Sprintf("catalog%s", fileExt))
-		if err := writeFile(fcfg, filename, writeFunc); err != nil {
-			return err
-		}
+	fsys := m.FS
+	if fsys == nil {
+		fsys = declcfg.OS{}
 	}
-	return nil
-}
 
-func writeFile(cfg declcfg.DeclarativeConfig, filename string, writeFunc WriteFunc) error {
-	buf := &bytes.Buffer{}
-	if err := writeFunc(cfg, buf); err != nil {
-		return fmt.Errorf("write to buffer for %q: %v", filename, err)
-	}
-	if err := ioutil.WriteFile(filename, buf.Bytes(), 0666); err != nil {
-		return fmt.Errorf("write file %q: %v", filename, err)
-	}
-	return nil
+	return declcfg.WriteFS(*cfg, fsys, m.OutputDir, m.WriteFunc, m.FileExt)
 }