Add: CLI tool to improve cluster management

Signed-off-by: Mofei Zhang <mofei2816@gmail.com>

Author: mervinkid

cli/cli-sample.json

@@ -0,0 +1,5 @@
+{
+  "masterAddr": [
+    "master.chubao.io"
+  ]
+}

cli/cli.go

@@ -0,0 +1,92 @@
+// Copyright 2018 The Chubao Authors.
+//
+// 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,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+// implied. See the License for the specific language governing
+// permissions and limitations under the License.
+
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"path"
+
+	"github.com/chubaofs/chubaofs/cli/cmd"
+	"github.com/chubaofs/chubaofs/sdk/master"
+	"github.com/spf13/cobra"
+)
+
+var (
+	CommitID   string
+	BranchName string
+	BuildTime  string
+)
+
+var (
+	defaultHomeDir, _ = os.UserHomeDir()
+	defaultConfigName = ".cfs-cli.json"
+	defaultConfigPath = path.Join(defaultHomeDir, defaultConfigName)
+	defaultConfigData = []byte(`
+{
+  "masterAddr": [
+    "master.chubao.io"
+  ]
+}
+`)
+)
+
+type config struct {
+	MasterAddr []string `json:"masterAddr"`
+}
+
+func runCLI() (err error) {
+	var cfg *config
+	if cfg, err = loadConfig(); err != nil {
+		return
+	}
+	err = setupCommands(cfg).Execute()
+	return
+}
+
+func loadConfig() (*config, error) {
+	var err error
+	var configData []byte
+	if configData, err = ioutil.ReadFile(defaultConfigPath); err != nil && !os.IsNotExist(err) {
+		return nil, err
+	}
+	if os.IsNotExist(err) {
+		if err = ioutil.WriteFile(defaultConfigPath, defaultConfigData, 0600); err != nil {
+			return nil, err
+		}
+		configData = defaultConfigData
+	}
+	var config = &config{}
+	if err = json.Unmarshal(configData, config); err != nil {
+		return nil, err
+	}
+	return config, nil
+}
+
+func setupCommands(cfg *config) *cobra.Command {
+	fmt.Printf("Using master address: %v\n", cfg.MasterAddr)
+	var mc = master.NewMasterClient(cfg.MasterAddr, false)
+	return cmd.NewRootCmd(mc)
+}
+
+func main() {
+	var err error
+	if err = runCLI(); err != nil {
+		_, _ = fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+}

cli/cmd/cluster.go

@@ -0,0 +1,75 @@
+// Copyright 2018 The Chubao Authors.
+//
+// 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,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+// implied. See the License for the specific language governing
+// permissions and limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/chubaofs/chubaofs/proto"
+	"github.com/chubaofs/chubaofs/sdk/master"
+	"github.com/spf13/cobra"
+)
+
+const (
+	cmdClusterUse   = "cluster [COMMAND]"
+	cmdClusterShort = "Manage cluster components"
+)
+
+func newClusterCmd(client *master.MasterClient) *cobra.Command {
+	var cmd = &cobra.Command{
+		Use:   cmdClusterUse,
+		Short: cmdClusterShort,
+	}
+	cmd.AddCommand(
+		newClusterInfoCmd(client),
+	)
+	return cmd
+}
+
+const (
+	cmdClusterInfoUse   = "info"
+	cmdClusterInfoShort = "Show cluster summary information"
+)
+
+func newClusterInfoCmd(client *master.MasterClient) *cobra.Command {
+	var cmd = &cobra.Command{
+		Use:   cmdClusterInfoUse,
+		Short: cmdClusterInfoShort,
+		Run: func(cmd *cobra.Command, args []string) {
+			var err error
+			var cv *proto.ClusterView
+			if cv, err = client.AdminAPI().GetCluster(); err != nil {
+				errout("Get cluster info fail:\n%v\n", err)
+				os.Exit(1)
+			}
+			stdout("\n[Summary]\n")
+			stdout(formatClusterView(cv))
+			stdout("\n")
+		},
+	}
+	return cmd
+}
+
+func formatClusterView(cv *proto.ClusterView) string {
+	var sb = strings.Builder{}
+	sb.WriteString(fmt.Sprintf("  Name          : %v\n", cv.Name))
+	sb.WriteString(fmt.Sprintf("  Auto allocate : %v\n", !cv.DisableAutoAlloc))
+	sb.WriteString(fmt.Sprintf("  MetaNode count: %v\n", len(cv.MetaNodes)))
+	sb.WriteString(fmt.Sprintf("  DataNode count: %v\n", len(cv.DataNodes)))
+	sb.WriteString(fmt.Sprintf("  Volume count  : %v\n", len(cv.VolStatInfo)))
+	return sb.String()
+}

cli/cmd/root.go

@@ -0,0 +1,50 @@
+// Copyright 2018 The Chubao Authors.
+//
+// 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,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+// implied. See the License for the specific language governing
+// permissions and limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"path"
+
+	"github.com/chubaofs/chubaofs/sdk/master"
+	"github.com/spf13/cobra"
+)
+
+const (
+	cmdRootShort = "ChubaoFS Command Line Interface (CLI)"
+)
+
+func NewRootCmd(client *master.MasterClient) *cobra.Command {
+	var cmd = &cobra.Command{
+		Use:   path.Base(os.Args[0]),
+		Short: cmdRootShort,
+		Args:  cobra.MinimumNArgs(0),
+	}
+	cmd.AddCommand(
+		newClusterCmd(client),
+		newVolCmd(client),
+		newUserCmd(client),
+	)
+	return cmd
+}
+
+func stdout(format string, a ...interface{}) {
+	_, _ = fmt.Fprintf(os.Stdout, format, a...)
+}
+
+func errout(format string, a ...interface{}) {
+	_, _ = fmt.Fprintf(os.Stderr, format, a...)
+}