forked from TrueCloudLab/frostfs-node
145 lines
3.6 KiB
Go
145 lines
3.6 KiB
Go
package inspect
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
common "github.com/nspcc-dev/neofs-node/cmd/neofs-lens/internal"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobovnicza"
|
|
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/writecache"
|
|
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
|
|
"github.com/nspcc-dev/neofs-sdk-go/object"
|
|
addressSDK "github.com/nspcc-dev/neofs-sdk-go/object/address"
|
|
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
flagAddress = "address"
|
|
flagEnginePath = "path"
|
|
flagHeader = "header"
|
|
flagOutFile = "out"
|
|
flagWriteCache = "writecache"
|
|
)
|
|
|
|
var (
|
|
vAddress string
|
|
vHeader bool
|
|
vPath string
|
|
vOut string
|
|
vWriteCache bool
|
|
)
|
|
|
|
// Command contains `inspect` command definition.
|
|
var Command = &cobra.Command{
|
|
Use: "inspect",
|
|
Short: "Object inspection",
|
|
Long: `Inspect specific object in storage engine component.`,
|
|
Run: objectInspectCmd,
|
|
}
|
|
|
|
func init() {
|
|
Command.Flags().StringVar(&vAddress, flagAddress, "", "Object address")
|
|
_ = Command.MarkFlagRequired(flagAddress)
|
|
|
|
Command.Flags().StringVar(&vPath, flagEnginePath, "",
|
|
"Path to storage engine component",
|
|
)
|
|
_ = Command.MarkFlagFilename(flagEnginePath)
|
|
_ = Command.MarkFlagRequired(flagEnginePath)
|
|
|
|
Command.Flags().StringVar(&vOut, flagOutFile, "",
|
|
"File to save object payload")
|
|
|
|
Command.Flags().BoolVar(&vHeader, flagHeader, false, "Inspect only header")
|
|
Command.Flags().BoolVar(&vWriteCache, flagWriteCache, false,
|
|
"Process write-cache")
|
|
}
|
|
|
|
func objectInspectCmd(cmd *cobra.Command, _ []string) {
|
|
addr := addressSDK.NewAddress()
|
|
err := addr.Parse(vAddress)
|
|
common.ExitOnErr(cmd, common.Errf("invalid address argument: %w", err))
|
|
|
|
if vOut == "" && !vHeader {
|
|
common.ExitOnErr(cmd, fmt.Errorf("either --%s or --%s flag must be provided",
|
|
flagHeader, flagOutFile))
|
|
}
|
|
|
|
if vWriteCache {
|
|
db, err := writecache.OpenDB(vPath, true)
|
|
common.ExitOnErr(cmd, common.Errf("could not open write-cache db: %w", err))
|
|
|
|
defer db.Close()
|
|
|
|
data, err := writecache.Get(db, []byte(addr.String()))
|
|
common.ExitOnErr(cmd, common.Errf("could not fetch object: %w", err))
|
|
printObjectInfo(cmd, data)
|
|
return
|
|
}
|
|
|
|
blz := blobovnicza.New(
|
|
blobovnicza.WithPath(vPath),
|
|
blobovnicza.ReadOnly())
|
|
common.ExitOnErr(cmd, blz.Open())
|
|
|
|
defer blz.Close()
|
|
|
|
prm := new(blobovnicza.GetPrm)
|
|
prm.SetAddress(addr)
|
|
res, err := blz.Get(prm)
|
|
common.ExitOnErr(cmd, common.Errf("could not fetch object: %w", err))
|
|
|
|
printObjectInfo(cmd, res.Object())
|
|
}
|
|
|
|
func printContainerID(cmd *cobra.Command, recv func() (cid.ID, bool)) {
|
|
var val string
|
|
|
|
id, ok := recv()
|
|
if ok {
|
|
val = id.String()
|
|
} else {
|
|
val = "<empty>"
|
|
}
|
|
|
|
cmd.Println("CID:", val)
|
|
}
|
|
|
|
func printObjectID(cmd *cobra.Command, recv func() (oid.ID, bool)) {
|
|
var val string
|
|
|
|
id, ok := recv()
|
|
if ok {
|
|
val = id.String()
|
|
} else {
|
|
val = "<empty>"
|
|
}
|
|
|
|
cmd.Println("ID:", val)
|
|
}
|
|
|
|
func printObjectInfo(cmd *cobra.Command, data []byte) {
|
|
obj := object.New()
|
|
err := obj.Unmarshal(data)
|
|
common.ExitOnErr(cmd, common.Errf("can't unmarshal object: %w", err))
|
|
|
|
if vHeader {
|
|
cmd.Println("Version:", obj.Version())
|
|
cmd.Println("Type:", obj.Type())
|
|
printContainerID(cmd, obj.ContainerID)
|
|
printObjectID(cmd, obj.ID)
|
|
cmd.Println("Owner:", obj.OwnerID())
|
|
cmd.Println("CreatedAt:", obj.CreationEpoch())
|
|
cmd.Println("PayloadSize:", obj.PayloadSize())
|
|
cmd.Println("Attributes:")
|
|
for _, attr := range obj.Attributes() {
|
|
cmd.Printf(" %s: %s\n", attr.Key(), attr.Value())
|
|
}
|
|
}
|
|
|
|
if vOut != "" {
|
|
err := os.WriteFile(vOut, obj.Payload(), 0644)
|
|
common.ExitOnErr(cmd, common.Errf("couldn't write payload: %w", err))
|
|
}
|
|
}
|