如何在 Go 中为值列表定义自定义标志类型？

在 Go 中将值列表放入 Flag 中

在 Go 中，flag 包支持解析命令行参数。然而，它最初只提供有限范围的数据类型（String、Int、Bool）。为了适应更复杂的数据，您可以定义自己的标志类型并使用 flag.Var() 来绑定它。

列表的自定义标志类型

考虑一个标志接受多个值作为列表。首先，定义自定义标志类型：

type arrayFlags []string

// String is an implementation of the flag.Value interface
func (i *arrayFlags) String() string {
    return fmt.Sprintf("%v", *i)
}

// Set is an implementation of the flag.Value interface
func (i *arrayFlags) Set(value string) error {
    *i = append(*i, value)
    return nil
}

使用自定义标志

接下来，声明此类型的变量以与自定义标志绑定：

var myFlags arrayFlags

注册自定义Flag

最后，您可以使用 flag 包注册您的自定义标志：

flag.Var(&myFlags, "list1", "Some description for this param.")

现在，您可以在运行程序时向此标志传递多个值：

go run your_file.go --list1 value1 --list1 value2

更新的代码片段

供参考，请参见此处完整的代码片段：

package main

import (
    "flag"
    "fmt"
)

type arrayFlags []string

// String is an implementation of the flag.Value interface
func (i *arrayFlags) String() string {
    return fmt.Sprintf("%v", *i)
}

// Set is an implementation of the flag.Value interface
func (i *arrayFlags) Set(value string) error {
    *i = append(*i, value)
    return nil
}

var myFlags arrayFlags

func main() {
    flag.Var(&myFlags, "list1", "Some description for this param.")
    flag.Parse()
}

以上是如何在 Go 中为值列表定义自定义标志类型？的详细内容。更多信息请关注PHP中文网其他相关文章！