Add process subcommands for multiple nodes

cli/src/alluxio.org/cli/env/env.go

@@ -221,7 +221,7 @@ func checkAndSetJava(envVar *viper.Viper) error {
 	// check if java is available via `PATH` using `which`
 	whichJavaPath, err := exec.Command("which", "java").Output()
 	if err == nil {
-		envVar.Set(ConfJava.EnvVar, string(whichJavaPath))
+		envVar.Set(ConfJava.EnvVar, strings.TrimSpace(string(whichJavaPath)))
 		return nil
 	}
 	// cannot find java

cli/src/alluxio.org/cli/env/process_start.go

@@ -16,13 +16,15 @@ import (
 )
 
 type StartProcessCommand struct {
+	Name            string
 	AsyncStart      bool
 	SkipKillOnStart bool
 }
 
 func (c *StartProcessCommand) ToCommand() *cobra.Command {
+	c.Name = "start"
 	cmd := &cobra.Command{
-		Use:   "start",
+		Use:   c.Name,
 		Short: "Starts a process",
 	}
 	cmd.PersistentFlags().BoolVarP(&c.SkipKillOnStart, "skip-kill-prev", "N", false, "Avoid killing previous running processes when starting")

cli/src/alluxio.org/cli/env/process_stop.go

@@ -14,12 +14,14 @@ package env
 import "github.com/spf13/cobra"
 
 type StopProcessCommand struct {
+	Name     string
 	SoftKill bool
 }
 
 func (c *StopProcessCommand) ToCommand() *cobra.Command {
+	c.Name = "stop"
 	cmd := &cobra.Command{
-		Use:   "stop",
+		Use:   c.Name,
 		Short: "Stops a process",
 	}
 	cmd.PersistentFlags().BoolVarP(&c.SoftKill, "soft", "s", false, "Soft kill only, don't forcibly kill the process")

cli/src/alluxio.org/cli/main.go

@@ -34,11 +34,18 @@ import (
 
 func main() {
 	for _, p := range []env.Process{
+		processes.All,
 		processes.JobMaster,
+		processes.JobMasters,
 		processes.JobWorker,
+		processes.JobWorkers,
+		processes.Local,
 		processes.Master,
+		processes.Masters,
 		processes.Proxy,
+		processes.Proxies,
 		processes.Worker,
+		processes.Workers,
 	} {
 		env.RegisterProcess(p)
 	}

cli/src/alluxio.org/cli/processes/all.go

@@ -0,0 +1,76 @@
+/*
+ * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
+ * (the "License"). You may not use this work except in compliance with the License, which is
+ * available at www.apache.org/licenses/LICENSE-2.0
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied, as more fully set forth in the License.
+ *
+ * See the NOTICE file distributed with this work for information regarding copyright ownership.
+ */
+
+package processes
+
+import (
+	"github.com/palantir/stacktrace"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+
+	"alluxio.org/cli/env"
+)
+
+var All = &AllProcess{
+	BaseProcess: &env.BaseProcess{
+		Name: "all",
+	},
+	Processes: []env.Process{
+		Masters,
+		JobMasters,
+		Workers,
+		JobWorkers,
+		Proxies,
+	},
+}
+
+type AllProcess struct {
+	*env.BaseProcess
+	Processes []env.Process
+}
+
+func (p *AllProcess) SetEnvVars(envVar *viper.Viper) {
+	return
+}
+
+func (p *AllProcess) Base() *env.BaseProcess {
+	return p.BaseProcess
+}
+
+func (p *AllProcess) StartCmd(cmd *cobra.Command) *cobra.Command {
+	cmd.Use = p.Name
+	return cmd
+}
+
+func (p *AllProcess) StopCmd(cmd *cobra.Command) *cobra.Command {
+	cmd.Use = p.Name
+	return cmd
+}
+
+func (p *AllProcess) Start(cmd *env.StartProcessCommand) error {
+	for i := 0; i < len(p.Processes); i++ {
+		subProcess := p.Processes[i]
+		if err := subProcess.Start(cmd); err != nil {
+			return stacktrace.Propagate(err, "Error starting subprocesses for %s", p.Processes[i])
+		}
+	}
+	return nil
+}
+
+func (p *AllProcess) Stop(cmd *env.StopProcessCommand) error {
+	for i := len(p.Processes) - 1; i >= 0; i-- {
+		subProcess := p.Processes[i]
+		if err := subProcess.Stop(cmd); err != nil {
+			return stacktrace.Propagate(err, "Error stopping subprocesses for %s", p.Processes[i])
+		}
+	}
+	return nil
+}