forked from TrueCloudLab/frostfs-node
Airat Arifullin
9b13a18aac
* Update version within go.mod; * Fix deprecated frostfs-api-go/v2 package and use frostfs-sdk-go/api instead. Signed-off-by: Airat Arifullin <a.arifullin@yadro.com>
93 lines
3 KiB
Go
93 lines
3 KiB
Go
package control
|
|
|
|
import (
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/commonflags"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key"
|
|
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
|
|
ircontrol "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control/ir"
|
|
ircontrolsrv "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control/ir/server"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs"
|
|
rawclient "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/rpc/client"
|
|
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
ownerFlag = "owner"
|
|
)
|
|
|
|
var removeContainerCmd = &cobra.Command{
|
|
Use: "remove-container",
|
|
Short: "Schedules a container removal",
|
|
Long: `Schedules a container removal via a notary request.
|
|
Container data will be deleted asynchronously by policer.
|
|
To check removal status "frostfs-cli container list" command can be used.`,
|
|
Run: removeContainer,
|
|
}
|
|
|
|
func initControlIRRemoveContainerCmd() {
|
|
initControlIRFlags(removeContainerCmd)
|
|
|
|
flags := removeContainerCmd.Flags()
|
|
flags.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
|
|
flags.String(ownerFlag, "", "Container owner's wallet address.")
|
|
removeContainerCmd.MarkFlagsMutuallyExclusive(commonflags.CIDFlag, ownerFlag)
|
|
removeContainerCmd.MarkFlagsOneRequired(commonflags.CIDFlag, ownerFlag)
|
|
}
|
|
|
|
func removeContainer(cmd *cobra.Command, _ []string) {
|
|
req := prepareRemoveContainerRequest(cmd)
|
|
|
|
pk := key.Get(cmd)
|
|
c := getClient(cmd, pk)
|
|
|
|
commonCmd.ExitOnErr(cmd, "could not sign request: %w", ircontrolsrv.SignMessage(pk, req))
|
|
|
|
var resp *ircontrol.RemoveContainerResponse
|
|
err := c.ExecRaw(func(client *rawclient.Client) error {
|
|
var err error
|
|
resp, err = ircontrol.RemoveContainer(client, req)
|
|
return err
|
|
})
|
|
commonCmd.ExitOnErr(cmd, "failed to execute request: %w", err)
|
|
|
|
verifyResponse(cmd, resp.GetSignature(), resp.GetBody())
|
|
|
|
if len(req.GetBody().GetContainerId()) > 0 {
|
|
cmd.Println("Container scheduled to removal")
|
|
} else {
|
|
cmd.Println("User containers sheduled to removal")
|
|
}
|
|
printVUB(cmd, resp.GetBody().GetVub())
|
|
}
|
|
|
|
func prepareRemoveContainerRequest(cmd *cobra.Command) *ircontrol.RemoveContainerRequest {
|
|
req := &ircontrol.RemoveContainerRequest{
|
|
Body: &ircontrol.RemoveContainerRequest_Body{},
|
|
}
|
|
|
|
cidStr, err := cmd.Flags().GetString(commonflags.CIDFlag)
|
|
commonCmd.ExitOnErr(cmd, "failed to get cid: ", err)
|
|
|
|
ownerStr, err := cmd.Flags().GetString(ownerFlag)
|
|
commonCmd.ExitOnErr(cmd, "failed to get owner: ", err)
|
|
|
|
if len(ownerStr) > 0 {
|
|
var owner user.ID
|
|
commonCmd.ExitOnErr(cmd, "invalid owner ID: %w", owner.DecodeString(ownerStr))
|
|
var ownerID refs.OwnerID
|
|
owner.WriteToV2(&ownerID)
|
|
req.Body.Owner = ownerID.StableMarshal(nil)
|
|
}
|
|
|
|
if len(cidStr) > 0 {
|
|
var containerID cid.ID
|
|
commonCmd.ExitOnErr(cmd, "invalid container ID: %w", containerID.DecodeString(cidStr))
|
|
req.Body.ContainerId = containerID[:]
|
|
}
|
|
|
|
req.Body.Vub = parseVUB(cmd)
|
|
|
|
return req
|
|
}
|