forked from TrueCloudLab/frostfs-node
95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
|
package control
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
|
||
|
|
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/refs"
|
||
|
|
rawclient "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/client"
|
||
|
|
"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"
|
||
|
|
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() {
|
||
|
|
initControlFlags(removeContainerCmd)
|
||
|
|
|
||
|
|
flags := removeContainerCmd.Flags()
|
||
|
|
flags.String(commonflags.CIDFlag, "", commonflags.CIDFlagUsage)
|
||
|
|
flags.String(ownerFlag, "", "Container owner's wallet address.")
|
||
|
|
removeContainerCmd.MarkFlagsMutuallyExclusive(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")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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 && len(cidStr) == 0 {
|
||
|
|
commonCmd.ExitOnErr(cmd, "invalid usage: %w", errors.New("neither owner's wallet address nor container ID are specified"))
|
||
|
|
}
|
||
|
|
|
||
|
|
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[:]
|
||
|
|
}
|
||
|
|
return req
|
||
|
|
}
|