当前位置：首页 > 编程日记 > 正文

: error c2062: 意外的类型“int”_Go 命令行解析 flag 包之扩展新类型

编程日记 2024-06-15 13:40:00

上篇文章说到，除布尔类型 Flag，flag 支持的还有整型（int、int64、uint、uint64）、浮点型（float64）、字符串（string）和时长（duration）。

flag 内置支持能满足大部分的需求，但某些场景，需要自定义解析规则。一个优秀的库肯定要支持扩展的。本文将介绍如何为 flag 扩展一个新的类型支持？

扩展目标

在 gvg 这个小工具中，list 子命令支持获取 Go 的版本列表。但版本的信息来源有多处，比如 installed（已安装）、local（本地仓库）和 remote（远程仓库）。

查看下 list 的帮助信息，如下：

NAME:gvg list - list go versionsUSAGE:gvg list [command options] [arguments...]OPTIONS:--origin value  the origin of version information , such as installed, local, remote (default: "installed")

可以看出，list 子命令支持一个 Flag 选项，--origin。它用于指定版本信息的来源，允许值的范围是 installed、local 和 remote。

如果要求不严格，用 StringVar 也可以实现。但问题是，使用 String，即使输入不在指定范围也能成功解析，不够严谨。虽说在获取后也可以检查，但还是不够灵活、可配置型也差。

接下来，我们要实现一个新的类型的 Flag，使选项的值必需在指定范围，否则要给出一定的错误提示信息。

实现思路

如何展一个新类型呢？

可以参考 flag 包内置类型的实现思路，比如 flag.DurationVar。Duration 不是基础类型，解析结果是存放到了 time.Duration 类型中，可能更有参考价值。

进入到 flag.DurationVar 查看源码，如下：

func DurationVar(p *time.Duration, name string, value time.Duration, usage string) {CommandLine.Var(newDurationValue(value, p), name, usage)
}

通过 newDurationValue 创建了一个类型为 durationValue 的变量，并传入到了 CommandLine.Var 方法中。

如果继续往下追，会根据 Value 创建一个 Flag 变量。如下：

func (f *FlagSet) Var(value Value, name string, usage string) {flag := &Flag{name, usage, value, value.String()}...
}

从 Var 的定义可以看出，它的第一个参数类型是 Value 接口类型，也就说，durationValue 是实现了 Value 接口的类型。

注意，源码中出现的 FlagSet 可以先忽略，它是下篇介绍子命令时重点关注的对象。

看下 Value 的定义，如下：

type Value interface {String() stringSet(string) error
}

那么，durationValue 的实现代码如何？

// 传入参数分别是默认值和获取 Flag 值的变量地址
func newDurationValue(val time.Duration, p *time.Duration) *durationValue {// 将默认值设置到 p 上*p = val// 使用 p 创建新的类型，保证可以获取到解析的结果return (*durationValue)(p)
}// Set 方法负责解析传入的值
func (d *durationValue) Set(s string) error {v, err := time.ParseDuration(s)if err != nil {err = errParse}*d = durationValue(v)return err
}// 获取真正的值
func (d *durationValue) String() string { return (*time.Duration)(d).String() }

核心在两个地方。

一个是创建新类型变量时，要使用传入的变量地址创建新类型变量，以实现将解析结果放到其中，让前端能获取到，二是 Set 方法中实现命令行传入字符串的解析。

逻辑梳理

看完上个小节，基本已经了解如何扩展一个新类型了。本质是是实现 Value 接口。

再看下之前提到的几个变量，分别是存放解析结果的指针、解析命令行输入的 Value 和表示一个选项的 Flag。对应于 flag.DurationVar，这个变量的类型分别是 *time.Duration、durationValue 和 Flag。

比如有 duration=1h，大致流程是首先从 os.Args 获取参数，按规则解析出选项名称 duration，查找是否存在名称为 duration 的 Flag，如果存在，使用 Flag.Value.Set 解析 1h，如果不满足 duration 的要求，将给出错误提示。

实现新类型

现在实现文章开头要求的目标。

新类型定义如下：

type stringEnumValue struct {options []stringp   *string
}

名为 StringEnumValue，即字符串枚举。它有 options 和 p 两个成员，options 指定一定范围的值，p 是 string 指针，保存解析结果的变量的地址。

下面定义创建 StringEnumValue 变量的函数 newStringEnumValue，代码如下：

func newStringEnumValue(val string, p *string, options []string) *StringEnumValue {*option = valreturn &stringEnumValue{options: options, p: p}
}

除了 val 和 p 两个必要的输入外，还有一个 string 切片类型的数，名为 options，它用于范围的限定。而函数主体，首先设置默认值，然后使用 options 和 p 创建变量返回。

Set 是核心方法，解析命令行传入字符串。代码如下：

func (s *StringEnumValue) Set(v string) error {for _, option := range s.options {if v == option {*(s.p) = vreturn nil}}return fmt.Errorf("must be one of %v", s.options)
}

循环检查输入参数 v 是否满足要求。定义如下：

最后是 String() 方法，

func (s *StringEnumValue) String() string {return *(s.p)
}

返回 p 指针中的值。前面分析实现思路时，Flag 在设置默认值时就调用了它。

使用 StringEnumValue

直接看代码吧。如下：

var origin string
func init() {flag.Var(newStringEnumValue("installed",    // 默认值&origin,[]string{"installed", "local", "remote"},),"origin",`the origin of version information, such as installed, local, remote (default: "installed")`,)
}func main() {flag.Parse()fmt.Println(option)
}

重点就是 flag.Var(newStringEnumValue(...)，...)。如果觉得有点啰嗦，希望和其他类型新建过程相同，在这个基础上可以再包装。代码如下：

func StringEnumVar(p *string, name string, options []string, defVal string, usage string) {flag.Var(newStringEnumValue(defVal, p, options), name, usage)
}

编译测试下，结果如下：

$ gvg --origin=any
invalid value "any" for flag -origin: must be one of [installed local remote]
Usage of gvg:-origin valuethe origin of version information, such as installed, local, remote (default installed)
$ gvg --origin=remote
origin remote

总结

本文介绍了如何为 flag 扩展一个类型支持，通过分析源码理清实现思路。最后创建了一个只接收指定范围值 Value。

欢迎关注我的微信公众号。

https://www.dkcj.cn/info/10825.html

: error c2062: 意外的类型“int”_Go 命令行解析 flag 包之扩展新类型

扩展目标

实现思路

逻辑梳理

实现新类型

使用 StringEnumValue

总结

相关文章：

java英文字符串大小写转换必须使用_【Java基础】之字符串大小写转换不利用API....

.net core 17

vue中子组件和子组件之间怎么通信_vue.js组件之间如何通信？

以太坊RLP机制分析

Android Studio导入Eclipse项目的两种方法

mediawiki java api_维基百科 MediaWiki API 解析

elasticdump安装_elasticsearch导出、导入工具-elasticdump

mysql 无法登陆_MySQL root用户无法登录原因及解决办法

Spring Boot启动过程（二）

dom vue 加载完执行_前端面试题——Vue

查看某个存储过程

java中的分页效率考虑_面试官：数据量很大，分页查询很慢，有什么优化方案？...

dede 后台 mysql_织梦dedecms使用Mysql8.0无法登录后台的解决办法

怎样对拍、如何对拍、对拍模板

二叉树线索化示意图_103-线索化二叉树思路图解

linux环境下搭建osm_web服务器一（Postgresql配置及osm2pgsql原始数据导入）：

Java开发买低压本还是标压本_标压和低压，笔记本怎么选才最香？

php mysql 备注_php,mysql备注信息1

JSP和Servlet学习笔记1 - 访问配置

unity人物旋转移动代码_Unity3D研究院之脚本实现模型的平移与旋转（六）

博客园的第一篇博文

java后台分页插件怎么写_Java分页技术(从后台传json到前台解析显示)

c 应用程序mysql_MySQL C 语言应用程序接口开发教程

GreenPlum学习笔记：基础知识

java 套接字关联的通道_Java 通道教程 – NIO 2.0

虚拟机ubuntu14.04系统设置静态ip

高职信息安全比赛攻防思路_30.LNGZ2020-30：2020年辽宁省职业院校技能大赛(高职组)“信息安全管理与评估”赛项规程...

oracle rac对心跳要求_关于心跳网络引起的Oracle RAC的节点驱逐（不是实例驱逐）...

php 字符串数组转数组对象_php怎么将数组转成对象？

如何在Win7电脑上增加新磁盘分区？