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")
}
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:]...))
}
}