Golang 基于 flag 库实现一个简单命令行工具

前言

Golang 标准库中的 flag 库提供了解析命令行选项的能力，我们可以基于此来开发命令行工具。

假设我们想做一个命令行工具，我们通过参数提供【城市】，它自动能够返回当前这个【城市】的天气状况。这样一个简单的需求，今天我们就来试一下，看怎样实现。

Package flag implements command-line flag parsing.

flag 库能够支持基础的命令行 flag 解析。使用起来并不复杂，

我们可以针对 string, integer, bool 三种类型来定义 flag，如：flag.String(), Bool(), Int()。

-nhelp message for flag n

当然，我们也可以直接将 flag 和变量绑定，这里要在上面三种方法的前面加上 Var 即可：

区别只在于首个参数是个指针，直接赋值，而不是 return 回来。签名都是类似的，我们看一个 Int64Var:

flag.Parse()

flag.Args()flag.Arg(i)

不熟悉的同学建议多看看看 go by example 的示例，讲的很清楚。

从开发者的角度看，其实我们只要定义好变量，用 flag.XXVar 来绑定，最后 flag.Parse 就可以用：

编译之后我们运行一下看看

如果没有设置 flag 的值，会取我们设置的默认值。

flag 支持的解析类型有下面四种：

有时候我们只需要一个 flag 就够了，选项本身就带着含义，不需要参数。而有些时候我们既需要 flag，也需要参数。注意区分好场景即可。如果用了第一种和第二种这种不带参数的，本质含义就是个 bool，出现就是 true，不出现就看默认值。

FlagSet 就是一组 flag 定义的集合，在 flag 库的底层是一个结构体：

注意有一个 Usage 函数，当解析 flag 出问题时就会调用这个。前面 flag 库封装的那些能力底层都是共用同一个默认的 CommandLine FlagSet实现的：

当我们调用 NewFlagSet 时需要指定这个集合的名称以及对应的错误处理。

第二个参数这个错误处理有三种选项，flag 已经提供：

ContinueOnErrorCommandLineExitOnErroros.Exit(2)PanicOnError

我们的需求很简单，提供一个 weather flag，接受输入的城市名称，随后我们返回天气数据即可。

所以，从 flag 的角度看，这里并不复杂，我们将【城市名称】绑定到一个 flag 上即可。

关键是要实现基于城市名称查天气的能力。这个也有公开的网站能实现，参照此前 Golang 教程中给出的 wttr 就可以。大家可以试一下，访问 wttr.in/wuhan 将会展示【武汉】的天气预报：

这里其实比较 trick，由于是网站，并不是公开的 open api，所以返回的数据也是 html 格式的，我们要思考一下怎么在命令行展示。

下面我们一步步来解决。

这一步基本是复用 flag 包提供的能力，这里我们用 StringVar 从命令行拿到值之后写入变量，这里相对比较通用，大家以后有需求可以直接改一下即可：

最终在 main() 函数中直接调用我们的 Execute 即可，注意我们解析到 weatherCity 不为空时，核心逻辑就在这个分支，我们留了个 TODO，下面看看怎么解。

前一节的 TODO 里本质需要我们实现的签名很简单:

fmt.Printf

而同时我们也有了 wttr 的能力，可以拿到数据，只不过是 html。关键是怎么转字符串。我们一步一步来：

一个简单的 http.Get 拿到 html 数据即可，这一步不复杂，大家直接看代码：

"github.com/JohannesKaufmann/html-to-markdown""github.com/MichaelMure/go-term-markdown"

好了，现在我们实现了两步，大家只需要把主流程里 TODO 的注释换成实际对下面 GetWeather 函数的调用即可，我们来看看运行效果。

完美，一个展示天气状况的命令行工具就做完了。这里的 opb 是我们的 package 名称，大家可以自己试一下，包名更换为你喜欢的名称即可。

其实开源社区各种能力基本都有同学研究过，大家可以打开思路，碰到知识点就思考如何能落地。笔者也是初学 flag 的时候本着实践的目的来试一试。正好发现了 html => markdown => terminal 打印这条路径，不一定是最好的，但作为一个 toy tool 足够了。