github.com/pingcap/tiup@v1.15.1/components/dm/command/reload.go (about)

     1  // Copyright 2020 PingCAP, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package command
    15  
    16  import (
    17  	perrs "github.com/pingcap/errors"
    18  	"github.com/pingcap/tiup/components/dm/spec"
    19  	"github.com/spf13/cobra"
    20  )
    21  
    22  func newReloadCmd() *cobra.Command {
    23  	var skipRestart bool
    24  	cmd := &cobra.Command{
    25  		Use:   "reload <cluster-name>",
    26  		Short: "Reload a DM cluster's config and restart if needed",
    27  		RunE: func(cmd *cobra.Command, args []string) error {
    28  			if len(args) != 1 {
    29  				return cmd.Help()
    30  			}
    31  
    32  			if err := validRoles(gOpt.Roles); err != nil {
    33  				return err
    34  			}
    35  
    36  			clusterName := args[0]
    37  
    38  			return cm.Reload(clusterName, gOpt, skipRestart, skipConfirm)
    39  		},
    40  		ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
    41  			switch len(args) {
    42  			case 0:
    43  				return shellCompGetClusterName(cm, toComplete)
    44  			default:
    45  				return nil, cobra.ShellCompDirectiveNoFileComp
    46  			}
    47  		},
    48  	}
    49  
    50  	cmd.Flags().StringSliceVarP(&gOpt.Roles, "role", "R", nil, "Only reload specified roles")
    51  	cmd.Flags().StringSliceVarP(&gOpt.Nodes, "node", "N", nil, "Only reload specified nodes")
    52  	cmd.Flags().BoolVar(&skipRestart, "skip-restart", false, "Only refresh configuration to remote and do not restart services")
    53  
    54  	return cmd
    55  }
    56  
    57  func validRoles(roles []string) error {
    58  	for _, r := range roles {
    59  		match := false
    60  		for _, has := range spec.AllDMComponentNames() {
    61  			if r == has {
    62  				match = true
    63  				break
    64  			}
    65  		}
    66  
    67  		if !match {
    68  			return perrs.Errorf("not valid role: %s, should be one of: %v", r, spec.AllDMComponentNames())
    69  		}
    70  	}
    71  
    72  	return nil
    73  }