cluster: add option for tmpdir

components/cluster/command/check.go

@@ -71,6 +71,7 @@ it will check the new instances `,
 	cmd.Flags().BoolVar(&opt.ApplyFix, "apply", false, "Try to fix failed checks")
 	cmd.Flags().BoolVar(&opt.ExistCluster, "cluster", false, "Check existing cluster, the input is a cluster name.")
 	cmd.Flags().Uint64Var(&gOpt.APITimeout, "api-timeout", 10, "Timeout in seconds when querying PD APIs.")
+	cmd.Flags().StringVarP(&opt.TempDir, "tempdir", "t", "/tmp/tiup", "The temporary directory.")
 
 	return cmd
 }

pkg/cluster/manager/check.go

@@ -42,8 +42,9 @@ type CheckOptions struct {
 	IdentityFile string // path to the private key file
 	UsePassword  bool   // use password instead of identity file for ssh connection
 	Opr          *operator.CheckOptions
-	ApplyFix     bool // try to apply fixes of failed checks
-	ExistCluster bool // check an exist cluster
+	ApplyFix     bool   // try to apply fixes of failed checks
+	ExistCluster bool   // check an exist cluster
+	TempDir      string // tempdir
 }
 
 // CheckCluster check cluster before deploying or upgrading
@@ -188,6 +189,7 @@ func checkSystemInfo(
 		applyFixTasks      []*task.StepDisplay
 		downloadTasks      []*task.StepDisplay
 	)
+	task.CheckToolsPathDir = opt.TempDir
 	logger := ctx.Value(logprinter.ContextKeyLogger).(*logprinter.Logger)
 	insightVer := ""
 

pkg/cluster/task/check.go

@@ -41,7 +41,7 @@ var (
 )
 
 // place the check utilities are stored
-const (
+var (
 	CheckToolsPathDir = "/tmp/tiup"
 )