优化命令行，并加入 [-help] [-start node=1] [-stop] [-config path] [-pprof 0.0.0.0:6060]参数

console/command.go

@@ -1,28 +1,89 @@
 package console
 
 import (
+	"flag"
 	"fmt"
+	"os"
 )
 
-type CommandFunctionCB func(args []string)error
+type valueType int
+type CommandFunctionCB func(args interface{}) error
+var commandList []*command
+const(
+	boolType valueType = iota
+	stringType valueType = iota
+)
+
+type command struct{
+	valType valueType
+	name string
+	bValue bool
+	strValue string
+	usage string
+	fn CommandFunctionCB
+}
+
+func (cmd *command) execute() error{
+	if cmd.valType == boolType {
+		return cmd.fn(cmd.bValue)
+	}else if cmd.valType == stringType {
+		return cmd.fn(cmd.strValue)
+	}else{
+		return fmt.Errorf("unknow command type!")
+	}
+
+	return nil
+}
+
 
-var mapRegisterCmd map[string]CommandFunctionCB =  map[string]CommandFunctionCB{}
 var programName string
 
 func Run(args []string) error {
+	flag.Parse()
 	programName = args[0]
-	if len(args) <= 1 {
-		return fmt.Errorf("command not found, try `%s help` for help",args[0])
+	if flag.NFlag() <= 0 {
+		return fmt.Errorf("Command input parameter error,try `%s -help` for help",args[0])
 	}
 
-	fn,ok := mapRegisterCmd[args[1]]
-	if ok == false{
-		return fmt.Errorf("command not found, try `%s help` for help",args[0])
+	var startCmd *command
+	for _,val := range commandList {
+		if val.name == "start" {
+			startCmd = val
+			continue
+		}
+		err := val.execute()
+		if err != nil {
+			return err
+		}
 	}
 
-	return fn(args)
+	return startCmd.execute()
 }
 
-func RegisterCommand(cmd string,fn CommandFunctionCB){
-	mapRegisterCmd[cmd] = fn
+func RegisterCommandBool(cmdName string, defaultValue bool, usage string,fn CommandFunctionCB){
+	var cmd command
+	cmd.valType = boolType
+	cmd.name = cmdName
+	cmd.fn = fn
+	cmd.usage = usage
+	flag.BoolVar(&cmd.bValue, cmdName, defaultValue, usage)
+	commandList = append(commandList,&cmd)
+}
+
+func RegisterCommandString(cmdName string, defaultValue string, usage string,fn CommandFunctionCB){
+	var cmd command
+	cmd.valType = stringType
+	cmd.name = cmdName
+	cmd.fn = fn
+	cmd.usage = usage
+	flag.StringVar(&cmd.strValue, cmdName, defaultValue, usage)
+	commandList = append(commandList,&cmd)
+}
+
+func PrintDefaults(){
+	fmt.Fprintf(os.Stderr, "Options:\n")
+
+	for _,val := range commandList {
+		fmt.Fprintf(os.Stderr, "  -%-10s%10s\n",val.name,val.usage)
+	}
 }