1.命令行参数获取:
命令行获得可通过os.Args参数,Args在os中定义为:
var Args []string
官方解释:Args hold the command-line arguments, starting with the program name.
2.命令行解析
golang 对命令行解析提供了flag包,按照功能可以分为以下三类:1.基本命令解析
2.自定义参数解析
3.解析外部命令字符串
2.1 基本命令解析
命令注册:通过flag.type(int,string等),提供命令名称, 默认参数值
按照返回值的形式分为两种, 返回值为值的指针, 通过传入指针,解析值后写入传入的指针
命令解析:调用flag.Parse()解析命令行
以bool解析为例子:
func main() {
var block bool
//如果有-b参数则为真,否则为假
//使用./flag -b=false
//./flag -b=ture
flag.BoolVar(&block, "b", false, "set for block")
flag.Parse()
fmt.Println("falg", block)
}
2.自定义参数命令解析
自定义变量实现下面的接口:type Value interface {
String() string
Set(string) error
}
并定义变量,通过flag.Var传入自定义变量
3.解析外部命令字符串
通过flag.NewFlagSet创建flagset.使用返回的set注册标志,调用set.Parse(argv)解析命令行. argv为字符串数组
实现代码如下:
package main
import (
"errors"
"flag"
"fmt"
"strings"
"time"
)
//自定义结构实现接口
type args []string
func (a *args) String() string {
return fmt.Sprintf("%v", *a)
}
func (i *args) Set(value string) error {
if len(*i) > 0 {
return errors.New("interval flag already set")
}
for _, dt := range strings.Split(value, ",") {
*i = append(*i, dt)
}
return nil
}
func base_parse() {
const (
defaultGopher = "pocket"
usage = "the variety of gopher"
)
var log int
//init for init param, return int point
// param:name, deaault val, help message
//解析参数通过指针返回
var intflag = flag.Int("int", 100, "help message for int")
flag.StringVar(&gopherType, "gopher_type", defaultGopher, usage)
flag.StringVar(&gopherType, "g", defaultGopher, usage+"(shorthand)")
//指定一个传入参数的指针
flag.IntVar(&log, "log_level", 0, "log level")
flag.Parse()
fmt.Println("initFlag:", *intflag)
fmt.Println("log_level:", log)
fmt.Println("gopherType string ", gopherType)
}
/*自定义参数解析*/
func userdef_parse() {
var argFlag args
flag.Var(&intervalFlag, "deltaT", "comma-separated list of intervals to use between events")
flag.Var(&argFlag, "arg", "comma-separated list of intervals to use between events")
flag.Parse()
fmt.Println("arg string ", argFlag)
fmt.Println(intervalFlag)
}
/*解析传入的字符串*/
func parse_input(arg string) {
argv := []string{}
for _, dt := range strings.Split(arg, " ") {
argv = append(argv, dt)
}
//创建flagSet
set := flag.NewFlagSet("test", flag.ExitOnError)
//注册解析参数
varflag := set.String("ver", "myver", "version")
pflag := set.String("p", "myp", "version")
//解析参数
set.Parse(argv)
fmt.Println("input parse:", *varflag, *pflag)
}
func main() {
userdef_parse()
parse_input("-p 123 -ver 1234")
}
有疑问加站长微信联系(非本文作者)