Esempio n. 1
0
func ConfigAction(c *cli.Context) {
	key := c.String("k")
	value := c.String("v")
	if key != "zkhosts" && key != "historyfile" && key != "display" {
		fmt.Printf("key %s not exists\n", key)
		os.Exit(-1)
	}
	u, err := user.Current()
	if err != nil {
		fmt.Println(err)
		os.Exit(-1)
	}
	conf, err := context.LoadConfig(u.HomeDir + context.DEFAULT_CONFIG_FILE)
	if err != nil {
		fmt.Println(err)
		os.Exit(-1)
	}
	switch key {
	case "zkhosts":
		conf.Zkhosts = value
		break
	case "historyfile":
		conf.HistoryFile = value
		break
	case "display":
		conf.Display = value
		break
	default:
		break
	}
	err = context.SaveConfig(u.HomeDir+context.DEFAULT_CONFIG_FILE, conf)
	if err != nil {
		fmt.Println(err)
		os.Exit(-1)
	}
	fmt.Println("Update config done")

}
Esempio n. 2
0
func main() {
	//load config
	user, err := user.Current()
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	conf, err := context.LoadConfig(user.HomeDir + context.DEFAULT_CONFIG_FILE)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	context.SetConfigContext(conf)

	if len(os.Args) > 1 {
		app := cli.NewApp()
		app.Name = "cli"
		app.Usage = "cluster manage tool"

		app.Commands = []cli.Command{
			initialize.Command,
			initialize.Command_node,
			c.AppAddCommand,
			c.AppDelCommand,
			c.AppModCommand,
			c.AppListCommand,
			c.ConfigCommand,
			c.UserAddCommand,
			c.UserDelCommand,
			c.UserGetCommand,
			c.ListFailoverRecordCommand,
			c.GetFailoverRecordCommand,
		}
		arg := append(os.Args)
		for _, cmd := range app.Commands {
			if os.Args[1] == cmd.Name {
				app.Run(arg)
				os.Exit(0)
			}
		}
	}
	if (len(os.Args) == 2 && (string(os.Args[1]) == "-h" || string(os.Args[1]) == "--help")) || (len(os.Args) == 1) {
		help := `Usage:
        cli init [options], -h for more details
        cli assign slot_range, -h for more details
        cli config -k <key> -v <value>, -h for more details
        cli appadd [options], -h for more details
        cli appdel [options], -h for more details
        cli appmod [options], -h for more details
        cli applist, -h for more details
        cli useradd [options], -h for more details
        cli userdel -u <username>, -h  for more details
        cli listfailover, -h for more details
        cli getfailover, -h for more details
        cli <AppName> [<Command>] [options], -h for more details
        `
		fmt.Println(help)
		os.Exit(1)
	}

	// Set context
	appName := os.Args[1]
	err = context.SetApp(appName, conf.Zkhosts)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	if conf.HistoryFile == "" {
		conf.HistoryFile = user.HomeDir + DEFAULT_HISTORY_FILE
	}
	_, err = os.Stat(conf.HistoryFile)
	if err != nil && os.IsNotExist(err) {
		_, err = os.Create(conf.HistoryFile)
		if err != nil {
			fmt.Println(conf.HistoryFile + "create failed")
		}
	}

	// REPL
	if len(os.Args) == 2 {
		err = linenoise.LoadHistory(conf.HistoryFile)
		if err != nil {
			fmt.Println(err)
		}
		for {
			str, err := linenoise.Line(appName + "> ")
			if err != nil {
				if err == linenoise.KillSignalError {
					os.Exit(1)
				}
				fmt.Printf("Unexpected error: %s\n", err)
				os.Exit(1)
			}
			fields := strings.Fields(str)

			linenoise.AddHistory(str)
			err = linenoise.SaveHistory(conf.HistoryFile)
			if err != nil {
				fmt.Println(err)
			}

			if len(fields) == 0 {
				continue
			}

			switch fields[0] {
			case "help":
				showHelp()
				continue
			case "quit":
				os.Exit(0)
			}

			cmd, ok := cmdmap[fields[0]]
			if !ok {
				fmt.Println("Error: unknown command.")
			}
			app := cli.NewApp()
			app.Name = cmd.Name
			app.Commands = []cli.Command{cmd}
			app.Run(append(os.Args[:1], fields...))
		}
	}

	// Command line
	if len(os.Args) > 2 {
		app := cli.NewApp()
		app.Name = "cli"
		app.Usage = "redis cluster cli"
		app.Commands = cmds
		app.Run(append(os.Args[:1], os.Args[2:]...))
	}
}