package tree 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" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/tree" "github.com/spf13/cobra" ) var healthcheckCmd = &cobra.Command{ Use: "healthcheck", Short: "Check tree service availability", Run: healthcheck, PersistentPreRun: func(cmd *cobra.Command, _ []string) { commonflags.Bind(cmd) }, } func initHealthcheckCmd() { commonflags.Init(healthcheckCmd) ff := healthcheckCmd.Flags() _ = cobra.MarkFlagRequired(ff, commonflags.RPC) } func healthcheck(cmd *cobra.Command, _ []string) { pk := key.GetOrGenerate(cmd) ctx := cmd.Context() cli, err := _client(ctx) commonCmd.ExitOnErr(cmd, "client: %w", err) req := &tree.HealthcheckRequest{ Body: &tree.HealthcheckRequest_Body{}, } commonCmd.ExitOnErr(cmd, "message signing: %w", tree.SignMessage(req, pk)) _, err = cli.Healthcheck(ctx, req) commonCmd.ExitOnErr(cmd, "rpc call: %w", err) cmd.Println("Success.") }