initial commit

Author: cpj2195

pkg/cmd/roachprod/cli/commands.go

@@ -273,8 +273,10 @@ func (cr *commandRegistry) buildLoadBalancerCmd() *cobra.Command {
 	}
 	loadBalancerCmd.AddCommand(
 		buildCreateLoadBalancerCmd(),
+		buildDeleteLoadBalancerCmd(),
 		buildLoadBalancerPGUrl(),
 		buildLoadBalancerIP(),
+		buildLoadBalancerList(),
 	)
 	return loadBalancerCmd
 }
@@ -304,6 +306,26 @@ These resources will automatically be destroyed when the cluster is destroyed.
 	return createLoadBalancerCmd
 }
 
+func buildDeleteLoadBalancerCmd() *cobra.Command {
+	deleteLoadBalancerCmd := &cobra.Command{
+		Use:   "destroy <cluster>",
+		Short: "destroy a load balancer for a cluster",
+		Long: `Destroy a load balancer for a specific service (port), system by default, for the given cluster.
+
+The load balancer is deleted using the cloud provider's API.
+`,
+		Args: cobra.ExactArgs(1),
+		Run: Wrap(func(cmd *cobra.Command, args []string) error {
+			return roachprod.DeleteLoadBalancer(context.Background(), config.Logger,
+				args[0], isSecure, virtualClusterName, sqlInstance,
+			)
+		}),
+	}
+	initFlagInsecureForCmd(deleteLoadBalancerCmd)
+	initFlagsClusterNSQLForCmd(deleteLoadBalancerCmd)
+	return deleteLoadBalancerCmd
+}
+
 func buildLoadBalancerPGUrl() *cobra.Command {
 	loadBalancerPGUrl := &cobra.Command{
 		Use:   "pgurl <cluster>",
@@ -357,6 +379,30 @@ func buildLoadBalancerIP() *cobra.Command {
 	return loadBalancerIP
 }
 
+func buildLoadBalancerList() *cobra.Command {
+	loadBalancerList := &cobra.Command{
+		Use:   "list <cluster>",
+		Short: "list all load balancers for a cluster",
+		Long:  "List all load balancers and their addresses for the given cluster.",
+		Args:  cobra.ExactArgs(1),
+		Run: Wrap(func(cmd *cobra.Command, args []string) error {
+			addresses, err := roachprod.ListLoadBalancers(config.Logger, args[0])
+			if err != nil {
+				return err
+			}
+			if len(addresses) == 0 {
+				fmt.Println("No load balancers found for cluster.")
+				return nil
+			}
+			for _, addr := range addresses {
+				fmt.Printf("%s:%d\n", addr.IP, addr.Port)
+			}
+			return nil
+		}),
+	}
+	return loadBalancerList
+}
+
 func (cr *commandRegistry) buildListCmd() *cobra.Command {
 	listCmd := &cobra.Command{
 		Use:   "list [--details | --json] [ --mine | --pattern ]",

pkg/roachprod/roachprod.go

@@ -3108,6 +3108,66 @@ func LoadBalancerIP(
 	return addr.IP, nil
 }
 
+// ListLoadBalancers returns all load balancers for the given cluster.
+func ListLoadBalancers(
+	l *logger.Logger, clusterName string,
+) ([]vm.ServiceAddress, error) {
+	c, err := GetClusterFromCache(l, clusterName)
+	if err != nil {
+		return nil, err
+	}
+	return c.ListLoadBalancers(l)
+}
+
+// DeleteLoadBalancer deletes the load balancer for the SQL service on the given
+// cluster.
+func DeleteLoadBalancer(
+	ctx context.Context,
+	l *logger.Logger,
+	clusterName string,
+	secure install.ComplexSecureOption,
+	virtualClusterName string,
+	sqlInstance int,
+) error {
+	c, err := GetClusterFromCache(l, clusterName, secure)
+	if err != nil {
+		return err
+	}
+
+	// If virtualClusterName is not provided, use the system interface name.
+	if virtualClusterName == "" {
+		virtualClusterName = install.SystemInterfaceName
+	}
+
+	// Find the SQL ports for the service on all nodes.
+	services, err := c.ServiceDescriptors(
+		ctx, c.Nodes, virtualClusterName, install.ServiceTypeSQL, sqlInstance,
+	)
+	if err != nil {
+		return err
+	}
+
+	port := config.DefaultSQLPort
+	if len(services) == 0 {
+		l.Errorf("WARNING: %s SQL service not found on cluster %s, using default SQL port %d",
+			virtualClusterName, clusterName, port)
+	} else {
+		port = services[0].Port
+		// Confirm that the service has the same port on all nodes.
+		for _, service := range services[1:] {
+			if port != service.Port {
+				return errors.Errorf("service %s must share the same port on all nodes, different ports found %d and %d",
+					virtualClusterName, port, service.Port)
+			}
+		}
+	}
+
+	// Delete the load balancer for the service's port.
+	return vm.FanOut(c.VMs, func(provider vm.Provider, vms vm.List) error {
+		return provider.DeleteLoadBalancer(l, vms, port)
+	})
+}
+
 // Deploy deploys a new version of cockroach to the given cluster. It currently
 // does not support clusters running external SQL instances.
 // TODO(herko): Add support for virtual clusters (external SQL processes)