首页 > itarticle > 【我说区块链】go中使用flag

【我说区块链】go中使用flag

admin 1月 23, 2021 0

os.Args

可以接受终端参数数组

1
2
3
4
5
6
7
8
9
10
11
12
 
func () {
   
 argsWithProg := os.Args
  fmt.Println(argsWithProg)


  argsWithoutProg := os.Args[1:]
   fmt.Println(argsWithoutProg)

    arg := os.Args[3]
    fmt.Println(arg)
}

flag.String, flag.Int, flag.Bool

Go提供了flag包，可以接收终端输入的参数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
 
func () {
    //基本的标记声明仅支持字符串、整数和布尔值选项。这里我们声明一个默认值为 "foo" 的字符串标志 word并带有一个简短的描述。这里的 flag.String 函数返回一个字符串指针（不是一个字符串值），在下面我们会看到是如何使用这个指针的。
 wordPtr := flag.String("word", "foo", "a string")
    //使用和声明 word 标志相同的方法来声明 numb 和 fork 标志。
 numbPtr := flag.Int("numb", 42, "an int")
    boolPtr := flag.Bool("fork", false, "a bool")

  //所有标志都声明完成以后，调用 flag.Parse() 来执行命令行解析。
 flag.Parse()


   //这里我们将仅输出解析的选项以及后面的位置参数。注意，我们需要使用类似 *wordPtr 这样的语法来对指针解引用，从而得到选项的实际值。
  fmt.Println("word:", *wordPtr)
   fmt.Println("numb:", *numbPtr)
   fmt.Println("fork:", *boolPtr)
   fmt.Println("tail:", flag.Args())

   flag.Usage()
}

所有标志声明完成，需要flag.Parse()执行命令解释，flag.Usage()可以提示帮助信息到终端

NewFlagSet

使用NewFlagSet可以实现用户自定义flag

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
 
func main() {

   addBlockCmd := flag.NewFlagSet("addblock", flag.ExitOnError)
 printChainCmd := flag.NewFlagSet("printchain", flag.ExitOnError)

    addBlockData := addBlockCmd.String("data", "", "Block data")

  switch os.Args[1] {
 case "addblock":
     err := addBlockCmd.Parse(os.Args[2:])
     if err != nil {
            log.Panic(err)
     }
    case "printchain":
       err := printChainCmd.Parse(os.Args[2:])
       if err != nil {
            log.Panic(err)
     }
    default:
     fmt.Println("No addblock and printchain!")
       os.Exit(1)
  }

   if addBlockCmd.Parsed() {
       if *addBlockData == "" {
            addBlockCmd.Usage()
          os.Exit(1)
      }
        fmt.Println("Data：" + *addBlockData)
 }

   if printChainCmd.Parsed() {
     fmt.Println("printchain,printchain,printchain!")
 }
}

使用：

./main.go addblock -data "send 2 btc to hackboy"

微海报分享

近期文章

近期评论

标签

热门

文章归档

分类目录

功能