golang flag

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()解析命令行

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",usage+"(shorthand)")
	//指定一个传入参数的指针
	flag.IntVar(&log,"log_level","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")
}