[#421] Try using badger for the write-cache

Signed-off-by: Alejandro Lopez <a.lopez@yadro.com>

cmd/frostfs-lens/internal/writecache/inspect.go

@@ -1,8 +1,13 @@
 package writecache
 
 import (
+	"fmt"
+	"os"
+
 	common "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal"
-	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache/writecachebadger"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache/writecachebbolt"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
 	objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
 	"github.com/spf13/cobra"
 )
@@ -18,14 +23,34 @@ func init() {
 	common.AddAddressFlag(inspectCMD, &vAddress)
 	common.AddComponentPathFlag(inspectCMD, &vPath)
 	common.AddOutputFileFlag(inspectCMD, &vOut)
+	common.AddDBTypeFlag(inspectCMD, &vDBType)
 }
 
 func inspectFunc(cmd *cobra.Command, _ []string) {
-	db := openWC(cmd)
-	defer db.Close()
+	var data []byte
 
-	data, err := writecache.Get(db, []byte(vAddress))
-	common.ExitOnErr(cmd, common.Errf("could not fetch object: %w", err))
+	switch vDBType {
+	case "bbolt":
+		db, err := writecachebbolt.OpenDB(vPath, true, os.OpenFile)
+		common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
+		defer db.Close()
+
+		data, err = writecachebbolt.Get(db, []byte(vAddress))
+		common.ExitOnErr(cmd, common.Errf("could not fetch object: %w", err))
+
+	case "badger":
+		log, err := logger.NewLogger(&logger.Prm{})
+		common.ExitOnErr(cmd, common.Errf("could not create logger: %w", err))
+
+		db, err := writecachebadger.OpenDB(vPath, true, log)
+		common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
+
+		data, err = writecachebadger.Get(db, []byte(vAddress))
+		common.ExitOnErr(cmd, common.Errf("could not fetch object: %w", err))
+
+	default:
+		common.ExitOnErr(cmd, fmt.Errorf("invalid dbtype: %q (possible values: bbolt, badger)", vDBType))
+	}
 
 	var o objectSDK.Object
 	common.ExitOnErr(cmd, common.Errf("could not unmarshal object: %w", o.Unmarshal(data)))

cmd/frostfs-lens/internal/writecache/list.go

@@ -3,9 +3,12 @@ package writecache
 import (
 	"fmt"
 	"io"
+	"os"
 
 	common "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal"
-	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache/writecachebadger"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache/writecachebbolt"
+	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
 	oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
 	"github.com/spf13/cobra"
 )
@@ -30,9 +33,26 @@ func listFunc(cmd *cobra.Command, _ []string) {
 		return err
 	}
 
-	db := openWC(cmd)
-	defer db.Close()
+	switch vDBType {
+	case "bbolt":
+		db, err := writecachebbolt.OpenDB(vPath, true, os.OpenFile)
+		common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
+		defer db.Close()
 
-	err := writecache.IterateDB(db, wAddr)
-	common.ExitOnErr(cmd, common.Errf("write-cache iterator failure: %w", err))
+		err = writecachebbolt.IterateDB(db, wAddr)
+		common.ExitOnErr(cmd, common.Errf("write-cache iterator failure: %w", err))
+
+	case "badger":
+		log, err := logger.NewLogger(&logger.Prm{})
+		common.ExitOnErr(cmd, common.Errf("could not create logger: %w", err))
+
+		db, err := writecachebadger.OpenDB(vPath, true, log)
+		common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
+
+		err = writecachebadger.IterateDB(db, wAddr)
+		common.ExitOnErr(cmd, common.Errf("write-cache iterator failure: %w", err))
+
+	default:
+		common.ExitOnErr(cmd, fmt.Errorf("invalid dbtype: %q (possible values: bbolt, badger)", vDBType))
+	}
 }

cmd/frostfs-lens/internal/writecache/root.go

@@ -1,18 +1,14 @@
 package writecache
 
 import (
-	"os"
-
-	common "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal"
-	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache"
 	"github.com/spf13/cobra"
-	"go.etcd.io/bbolt"
 )
 
 var (
 	vAddress string
 	vPath    string
 	vOut     string
+	vDBType  string
 )
 
 // Root contains `write-cache` command definition.
@@ -24,10 +20,3 @@ var Root = &cobra.Command{
 func init() {
 	Root.AddCommand(listCMD, inspectCMD)
 }
-
-func openWC(cmd *cobra.Command) *bbolt.DB {
-	db, err := writecache.OpenDB(vPath, true, os.OpenFile)
-	common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
-
-	return db
-}