Go:从文件中读取命令行标志

时间:2018-04-05 16:34:00

标签: go command-line-interface

在Go中创建命令行程序时,是否有一种方法(由核心库提供,或者是广泛接受的实践)来实现从文件中读取其内容的命令行标志?

如果不清楚,我正在考虑命令行@程序中的curl符号。许多curl的参数允许你做这样的事情来从文件中读取标志的值

# setting the value 
curl --data-binary '{...}' http://example.com

# setting the value by reading from a file
curl --data-binary @path/to/data.txt http://example.com  

是否有自动实现这些标志的代码?我已经阅读了官方文档和didn't see anything obvious,但我仍然掌握着浏览这些文档。

如果没有正式的话,是否有一个事实上标准的“更好的标志”图书馆由go社区中的某个人提供,其中包含此功能?

或者是由个人程序员创建一个字符串标志,扫描它为@,并处理自己读取文件的内容?

1 个答案:

答案 0 :(得分:5)

flag.Value允许您实现任意标记行为:

package main

import (
        "flag"
        "fmt"
        "io/ioutil"
        "strings"
)

type T string

// String implements flag.Value
func (t *T) String() string {
        return string(*t)
}

// Set implements flag.Value
func (t *T) Set(maybeFilename string) error {
        if !strings.HasPrefix(maybeFilename, "@") {
                *t = T(maybeFilename)
                return nil
        }

        filename := maybeFilename[1:]
        b, err := ioutil.ReadFile(filename)
        if err != nil {
                return err
        }

        *t = T(string(b))
        return nil
}

func main() {
        var t T
        flag.Var(&t, "data-binary", "(description)")
        flag.Parse()
        fmt.Println(t)
}