[#922] cli: Add new command `control list-targets`

Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com>
pull/954/head
Anton Nikiforov 2024-01-26 17:15:28 +03:00 committed by Evgenii Stratonikov
parent f64c48b157
commit befbaf9d56
2 changed files with 84 additions and 0 deletions

View File

@ -0,0 +1,82 @@
package control
import (
"bytes"
"crypto/sha256"
"fmt"
"strconv"
"text/tabwriter"
"git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/rpc/client"
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/key"
commonCmd "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/internal/common"
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
cid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id"
"github.com/spf13/cobra"
)
const (
chainNameFlag = "chain-name"
chainNameFlagUsage = "Chain name(ingress|s3)"
)
var listTargetsCmd = &cobra.Command{
Use: "list-targets",
Short: "List local targets",
Long: "List local APE overrides of the node",
Run: listTargets,
}
func listTargets(cmd *cobra.Command, _ []string) {
pk := key.Get(cmd)
var cnr cid.ID
chainName, _ := cmd.Flags().GetString(chainNameFlag)
rawCID := make([]byte, sha256.Size)
cnr.Encode(rawCID)
req := &control.ListTargetsLocalOverridesRequest{
Body: &control.ListTargetsLocalOverridesRequest_Body{
ChainName: chainName,
},
}
signRequest(cmd, pk, req)
cli := getClient(cmd, pk)
var resp *control.ListTargetsLocalOverridesResponse
var err error
err = cli.ExecRaw(func(client *client.Client) error {
resp, err = control.ListTargetsLocalOverrides(client, req)
return err
})
commonCmd.ExitOnErr(cmd, "rpc error: %w", err)
verifyResponse(cmd, resp.GetSignature(), resp.GetBody())
targets := resp.GetBody().GetTargets()
if len(targets) == 0 {
cmd.Println("Local overrides are not defined for the container.")
return
}
buf := bytes.NewBuffer(nil)
tw := tabwriter.NewWriter(buf, 0, 2, 2, ' ', 0)
_, _ = tw.Write([]byte("#\tName\tType\n"))
for i, t := range targets {
_, _ = tw.Write([]byte(fmt.Sprintf("%s\t%s\t%s\n", strconv.Itoa(i), t.GetName(), t.GetType())))
}
_ = tw.Flush()
cmd.Print(buf.String())
}
func initControlListTargetsCmd() {
initControlFlags(listTargetsCmd)
ff := listTargetsCmd.Flags()
ff.String(chainNameFlag, "", chainNameFlagUsage)
_ = cobra.MarkFlagRequired(ff, chainNameFlag)
}

View File

@ -38,6 +38,7 @@ func init() {
removeRuleCmd,
listRulesCmd,
getRuleCmd,
listTargetsCmd,
)
initControlHealthCheckCmd()
@ -50,4 +51,5 @@ func init() {
initControlRemoveRuleCmd()
initControlListRulesCmd()
initControGetRuleCmd()
initControlListTargetsCmd()
}