[#1175] CLI: Implement command to lock the objects

Implement `lock` command and add it to `object` section. The command
accepts container argument (string `cid.ID`) and list of locked objects
(string `oid.ID` list). From the provided input `LOCK` object is
constructed and stored using NeoFS API protocol.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2022-02-16 18:56:16 +03:00 committed by LeL
parent 8a2a096680
commit d92b1d1bf8
2 changed files with 67 additions and 0 deletions

View file

@ -0,0 +1,65 @@
package cmd
import (
"fmt"
internalclient "github.com/nspcc-dev/neofs-node/cmd/neofs-cli/internal/client"
objectcore "github.com/nspcc-dev/neofs-node/pkg/core/object"
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
"github.com/nspcc-dev/neofs-sdk-go/object"
oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
"github.com/spf13/cobra"
)
// object lock command.
var cmdObjectLock = &cobra.Command{
Use: "lock CONTAINER OBJECT...",
Short: "Lock object in container",
Long: "Lock object in container",
Args: cobra.MinimumNArgs(2),
Run: func(cmd *cobra.Command, args []string) {
var cnr cid.ID
err := cnr.Parse(args[0])
exitOnErr(cmd, errf("Incorrect container arg: %v", err))
argsList := args[1:]
lockList := make([]oid.ID, len(argsList))
for i := range argsList {
err = lockList[i].Parse(argsList[i])
exitOnErr(cmd, errf(fmt.Sprintf("Incorrect object arg #%d: %%v", i+1), err))
}
key, err := getKey()
exitOnErr(cmd, errf("can't fetch private key: %w", err))
idOwner, err := getOwnerID(key)
exitOnErr(cmd, err)
var lock object.Lock
lock.WriteMembers(lockList)
obj := object.NewRaw()
obj.SetContainerID(&cnr)
obj.SetOwnerID(idOwner)
obj.SetType(object.TypeLock)
obj.SetPayload(lock.Marshal())
var prm internalclient.PutObjectPrm
prepareSessionPrmWithOwner(cmd, objectcore.AddressOf(obj), key, idOwner, &prm)
prepareObjectPrm(cmd, &prm)
prm.SetHeader(obj.Object())
_, err = internalclient.PutObject(prm)
exitOnErr(cmd, errf("Store lock object in NeoFS: %w", err))
cmd.Println("Objects successfully locked.")
},
}
func initCommandObjectLock() {
initCommonFlags(cmdObjectLock)
}

View file

@ -246,6 +246,7 @@ func init() {
objectHeadCmd, objectHeadCmd,
objectHashCmd, objectHashCmd,
objectRangeCmd, objectRangeCmd,
cmdObjectLock,
} }
rootCmd.AddCommand(objectCmd) rootCmd.AddCommand(objectCmd)
@ -276,6 +277,7 @@ func init() {
initObjectHeadCmd() initObjectHeadCmd()
initObjectHashCmd() initObjectHashCmd()
initObjectRangeCmd() initObjectRangeCmd()
initCommandObjectLock()
} }
type clientKeySession interface { type clientKeySession interface {