go语言学习---使用os.Args获取简单参数（命令行解析）&&flag自定义参数 ...

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

go语言学习---使用os.Args获取简单参数（命令行解析）&&flag自定义参数 ...

原作者: [db:作者] 来自: [db:来源] 收藏邀请

实例1：

//main
package main

import (
	"fmt"
	"os"
)

func main() {
	fmt.Println(os.Args)

}

　　执行结果

使用flag获取复杂参数

实例2 ---flag.xxx() 例如：flag.Int, flag.String, 返回解析变量类型的指针

//main
package main

import (
	"flag"
	"fmt"
)

func main() {
	methodPtr := flag.String("method", "default", "method of sample") //return pointer
	valuePtr := flag.Int("value", -1, "value of sample")
	flag.Parse()
	fmt.Println(*methodPtr, *valuePtr)
}

执行结果：

实例3：加入错误提示（无法解析同名列表参数，eg: -cfg a -cfg b -cfg c ...）

//main
package main

import (
	"flag"
	"fmt"
)

func style() {
	//格式化定义
	methodPtr := flag.String("method", "default", "method of sample") //return pointer
	valuePtr := flag.Int("value", -1, "value of sample")

	//解析
	flag.Parse()

	fmt.Println(*methodPtr, *valuePtr)

}

func style2() {
	var method string
	var value int
	flag.StringVar(&method, "method", "default", "method of sample")
	flag.IntVar(&value, "value", -1, "value of sample")

	//解析
	flag.Parse()

	fmt.Println(method, value)

}
func main() {
	style2()

}

　　执行结果：

把解析的内容直接覆盖到自己定义的变量里： flag.XxxVar() 例如：flag.IntVar, flag.StringVar

package main

import (
    "flag"
    "fmt"
)
var host string
var port int

func init() { // 每个文件会自动执行的函数
    flag.StringVar(&host, "host", "127.0.0.1", "请输入host地址")
    flag.IntVar(&port, "port", 3306, "请输入端口号")
}

func main() {
    flag.Parse() // 解析参数
    fmt.Printf("%s:%d\n", host, port)
}

.自定义参数解析flag.Var(), 我们可以看下flag.go源码:

func Var(value Value, name string, usage string) {
    CommandLine.Var(value, name, usage)
}

type Value interface {
    String() string
    Set(string) error
}

使用flag.Var函数第一个参数我们需要传入一个Value类型的值，Value是一个接口类型，定义了两个方法，接下来我们去实现这两个方法：

package main

import (
    "flag"
    "fmt"
    "strings"
)

// 自定义类型
type HandsomeBoys []string

// 实现String()方法
func (h *HandsomeBoys) String() string {
    return fmt.Sprintf("%v", *h)
}

// 实现Set方法,Set接口决定了如何解析flag的值
func (h *HandsomeBoys) Set(s string) error {
    for _, v := range strings.Split(s, ",") {
        *h = append(*h, v)
    }
    return nil
}

// 定义一个HandsomeBoys类型的变量
var boys HandsomeBoys

func init() {
    // 绑定变量boys
    flag.Var(&boys, "boys", "请输入一组帅气的男孩名称：-boys=彭于晏,吴彦祖")
}

func main() {
    flag.Parse()
    fmt.Println(boys)
}