roachprod: add parca-agent cli commands

Add start and stop commands for the parca-agent to the roachprod CLI.

Epic: None
Release note: None

Author: herkolategan

pkg/cmd/roachprod/cli/BUILD.bazel

@@ -18,6 +18,7 @@ go_library(
         "//pkg/roachprod",
         "//pkg/roachprod/agents/fluentbit",
         "//pkg/roachprod/agents/opentelemetry",
+        "//pkg/roachprod/agents/parca",
         "//pkg/roachprod/cloud",
         "//pkg/roachprod/config",
         "//pkg/roachprod/errors",

pkg/cmd/roachprod/cli/commands.go

@@ -2147,6 +2147,32 @@ func (cr *commandRegistry) buildOpentelemetryStopCmd() *cobra.Command {
 	}
 }
 
+func (cr *commandRegistry) buildParcaAgentStartCmd() *cobra.Command {
+	parcaAgentStartCmd := &cobra.Command{
+		Use:   "parca-agent-start <cluster>",
+		Short: "Install and start the Parca Agent",
+		Long:  "Install and start the Parca Agent",
+		Args:  cobra.ExactArgs(1),
+		Run: Wrap(func(cmd *cobra.Command, args []string) error {
+			return roachprod.StartParcaAgent(context.Background(), config.Logger, args[0], parcaAgentConfig)
+		}),
+	}
+	initParcaAgentStartCmdFlags(parcaAgentStartCmd)
+	return parcaAgentStartCmd
+}
+
+func (cr *commandRegistry) buildParcaAgentStopCmd() *cobra.Command {
+	return &cobra.Command{
+		Use:   "parca-agent-stop <cluster>",
+		Short: "Stop the Parca Agent",
+		Long:  "Stop the Parca Agent",
+		Args:  cobra.ExactArgs(1),
+		Run: Wrap(func(cmd *cobra.Command, args []string) error {
+			return roachprod.StopParcaAgent(context.Background(), config.Logger, args[0])
+		}),
+	}
+}
+
 func (cr *commandRegistry) buildFetchLogsCmd() *cobra.Command {
 	fetchLogsCmd := &cobra.Command{
 		Use:     "fetchlogs <cluster> <destination (optional)> [flags]",

pkg/cmd/roachprod/cli/flags.go

@@ -13,6 +13,7 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/roachprod"
 	"github.com/cockroachdb/cockroach/pkg/roachprod/agents/fluentbit"
 	"github.com/cockroachdb/cockroach/pkg/roachprod/agents/opentelemetry"
+	"github.com/cockroachdb/cockroach/pkg/roachprod/agents/parca"
 	"github.com/cockroachdb/cockroach/pkg/roachprod/config"
 	"github.com/cockroachdb/cockroach/pkg/roachprod/install"
 	"github.com/cockroachdb/cockroach/pkg/roachprod/ssh"
@@ -101,9 +102,9 @@ var (
 
 	sshKeyUser string
 
-	fluentBitConfig fluentbit.Config
-
+	fluentBitConfig     fluentbit.Config
 	opentelemetryConfig opentelemetry.Config
+	parcaAgentConfig    parca.Config
 
 	fetchLogsTimeout time.Duration
 )
@@ -393,6 +394,11 @@ func initOpentelemetryStartCmdFlags(opentelemetryStartCmd *cobra.Command) {
 		"Datadog tags as a comma-separated list in the format KEY1:VAL1,KEY2:VAL2")
 }
 
+func initParcaAgentStartCmdFlags(parcaAgentStartCmd *cobra.Command) {
+	parcaAgentStartCmd.Flags().StringVar(&parcaAgentConfig.Token, "parca-agent-token", "",
+		"Parca Agent Token")
+}
+
 func initGCCmdFlags(gcCmd *cobra.Command) {
 	gcCmd.Flags().BoolVarP(&dryrun,
 		"dry-run", "n", dryrun, "dry run (don't perform any actions)")

pkg/cmd/roachprod/cli/resgistry.go

@@ -68,6 +68,8 @@ func (cr *commandRegistry) register() {
 		cr.buildFluentBitStopCmd(),
 		cr.buildOpentelemetryStartCmd(),
 		cr.buildOpentelemetryStopCmd(),
+		cr.buildParcaAgentStartCmd(),
+		cr.buildParcaAgentStopCmd(),
 		cr.buildFetchLogsCmd(),
 		cr.buildGetLatestPProfCmd(),
 		cr.buildFetchCertsDir(),