Merge 1ec1562 into c3ecda7

components/cluster/command/root.go

@@ -210,6 +210,7 @@ func init() {
 		newTelemetryCmd(),
 		newReplayCmd(),
 		newTemplateCmd(),
+		newTLSCmd(),
 	)
 }
 

components/cluster/command/tls.go

@@ -0,0 +1,71 @@
+// Copyright 2021 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package command
+
+import (
+	"strings"
+
+	perrs "github.com/pingcap/errors"
+	"github.com/spf13/cobra"
+)
+
+func newTLSCmd() *cobra.Command {
+	var (
+		reloadCertificate bool // reload certificate when the cluster enable encrypted communication
+		cleanCertificate  bool // cleanup certificate when the cluster disable encrypted communication
+		enableTLS         bool
+	)
+
+	cmd := &cobra.Command{
+		Use:   "tls <cluster-name> <enable/disable>",
+		Short: "Enable/Disable TLS between TiDB components",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) != 2 {
+				return cmd.Help()
+			}
+
+			if err := validRoles(gOpt.Roles); err != nil {
+				return err
+			}
+			clusterName := args[0]
+			clusterReport.ID = scrubClusterName(clusterName)
+			teleCommand = append(teleCommand, scrubClusterName(clusterName))
+
+			switch strings.ToLower(args[1]) {
+			case "enable":
+				enableTLS = true
+			case "disable":
+				enableTLS = false
+			default:
+				return perrs.New("enable or disable must be specified at least one")
+			}
+
+			if enableTLS && cleanCertificate {
+				return perrs.New("clean-certificate only works when tls disable")
+			}
+
+			if !enableTLS && reloadCertificate {
+				return perrs.New("reload-certificate only works when tls enable")
+			}
+
+			return cm.TLS(clusterName, gOpt, enableTLS, cleanCertificate, reloadCertificate, skipConfirm)
+		},
+	}
+
+	cmd.Flags().BoolVar(&cleanCertificate, "clean-certificate", false, "Cleanup the certificate file if it already exists when tls disable")
+	cmd.Flags().BoolVar(&reloadCertificate, "reload-certificate", false, "Load the certificate file whether it exists or not when tls enable")
+	cmd.Flags().BoolVar(&gOpt.Force, "force", false, "Force enable/disable tls regardless of the current state")
+
+	return cmd
+}

components/dm/spec/logic.go

@@ -140,10 +140,21 @@ func (i *MasterInstance) InitConfig(
 		return err
 	}
 
+	// doesn't work
+	if _, err := i.setTLSConfig(ctx, false, nil, paths); err != nil {
+		return err
+	}
+
 	specConfig := spec.Config
 	return i.MergeServerConfig(ctx, e, i.topo.ServerConfigs.Master, specConfig, paths)
 }
 
+// setTLSConfig set TLS Config to support enable/disable TLS
+// MasterInstance no need to configure TLS
+func (i *MasterInstance) setTLSConfig(ctx context.Context, enableTLS bool, configs map[string]interface{}, paths meta.DirPaths) (map[string]interface{}, error) {
+	return nil, nil
+}
+
 // ScaleConfig deploy temporary config on scaling
 func (i *MasterInstance) ScaleConfig(
 	ctx context.Context,
@@ -271,10 +282,21 @@ func (i *WorkerInstance) InitConfig(
 		return err
 	}
 
+	// doesn't work
+	if _, err := i.setTLSConfig(ctx, false, nil, paths); err != nil {
+		return err
+	}
+
 	specConfig := spec.Config
 	return i.MergeServerConfig(ctx, e, i.topo.ServerConfigs.Worker, specConfig, paths)
 }
 
+// setTLSConfig set TLS Config to support enable/disable TLS
+// workrsInstance no need to configure TLS
+func (i *WorkerInstance) setTLSConfig(ctx context.Context, enableTLS bool, configs map[string]interface{}, paths meta.DirPaths) (map[string]interface{}, error) {
+	return nil, nil
+}
+
 // ScaleConfig deploy temporary config on scaling
 func (i *WorkerInstance) ScaleConfig(
 	ctx context.Context,
@@ -322,7 +344,7 @@ func (topo *Specification) ComponentsByStartOrder() (comps []Component) {
 	// "dm-master", "dm-worker"
 	comps = append(comps, &DMMasterComponent{topo})
 	comps = append(comps, &DMWorkerComponent{topo})
-	comps = append(comps, &spec.MonitorComponent{Topology: topo})
+	comps = append(comps, &spec.MonitorComponent{Topology: topo}) // prometheus
 	comps = append(comps, &spec.GrafanaComponent{Topology: topo})
 	comps = append(comps, &spec.AlertManagerComponent{Topology: topo})
 	return