Golang实现一个linux命令ls命令(命令行工具构建)
编程三分钟
共 2999字，需浏览 6分钟
·
2023-01-13 09:12
希望2023可以听到这些话：
- 恭喜你得到了这份工作
- 恭喜你的建议被采用了
- 恭喜你被录取了
- 恭喜你的考试顺利通过了
- 恭喜你上岸了
- 恭喜你升职了
- 恭喜你加薪了
- 恭喜你体检结果一切正常
在这篇文章下面许个愿吧！
ls 命令
要实现ls，首先先我们复习一下ls命令。
-
ls不加任何参数，表示查询当前目录下的文件/文件夹 -
ls后面加上路径，列出所有，不包含隐藏文件/文件夹 -
ls -a列出所有，包括以.开头的隐藏文件
此命令还有很多功能，这一节的重点是命令行程度的编写，就实现上面四个。
os.Args 读取参数
定义如下，一个数组。该数组长度由参数个数决定，第一个参数为程序本身，后面为参数列表。
var Args []string
比如ls -l，Args=["ls","-l"]，我们现在实现ls+目录，或者缺省目录时默认当前目录代码如下。
func main() {
targetDirPath := "./"
if len(os.Args) > 1 {
targetDirPath = os.Args[1]
}
if dirList, err := ioutil.ReadDir(targetDirPath); err == nil {
for _, dirInfo := range dirList {
fmt.Print(dirInfo.Name() + " ")
}
} else {
fmt.Println(err.Error())
}
}
-
使用
os.Agrs来获取传入程序的参数 -
使用
ioutil.ReadDir读取目录,将返回值[] FileInfo遍历,输出目录/文件信息 -
将遇到的
error输出到 终端
输出
$ go build -o ls main.go
$ ./ls
main.go folder ls
$ ./ls folder
file1 file2
看，我们已经简单的实现了。但我们会发现，要使用-开头的参数时，用os.Args这个还需要手写更多复杂逻辑去判断输入的到底是路径，还是其他参数，于是下一个包登场了。
pflag
昨天的每日一Go已经讲了flag包，pflag 包与 flag 包的工作原理甚至是代码实现都是类似的，下面是 pflag 相对 flag 的一些优势：
-
支持更加精细的参数类型：例如，
flag只支持uint和uint64，而pflag额外支持uint8、uint16、int32等类型。 -
支持更多参数类型：
ip、ip mask、ip net、count、以及所有类型的slice类型。 -
兼容标准
flag库的Flag和FlagSet：pflag更像是对flag的扩展。 -
原生支持更丰富的功能：支持
shorthand、deprecated、hidden等高级功能。
var a = pflag.BoolP("all", "a", false, "Include directory entries whose names begin with a dot (.).")
var help = pflag.BoolP("help", "h", false, "Show this help message.")
pflag.Parse()
if *help {
pflag.Usage()
return
}
args := pflag.Args()
if len(args) == 0 {
args = append(args, "./")
}
if len(args) == 1 {
ShowPath(args[0], *a)
} else {
for _, v := range args {
fmt.Println(v + ":")
ShowPath(v, *a)
}
}
-
列出文件夹的方式写到了
ShowPath函数里，第二参数是一个bool判断是否列出全部文件（包括隐藏文件）感兴趣去看我源代码，这里不是重点。 -
pflag.BoolP定义一个bool类型的参数，支持长短参数形式--all,-a，返回值是指针。 -
pflag.Parse()调用这个函数，把参数值设置到相应指针的指向里。 -
pflag.Usage()用法打印。
这样我们就实现了ls的基本功能。测试一下。
$ go build -o ls main.go
$ ./ls
folder go.mod go.sum main.go
$ ./ls -a
. .. .git folder go.mod go.sum main.go
$ ./ls ./ folder
./:
folder go.mod go.sum main.go
folder:
file1 file2
$ ./ls -h
Usage of ./ls:
-a, --all Include directory entries whose names begin with a dot (.).
-h, --help Show this help message.
是不是相当给力啦！
小结
命令行读取一共有五个常用的包
-
os.Args系统包提供的参数解析。 -
flag：简单参数解析，昨天的每日一Go分享过。 -
pflag：命令行参数解析。 -
cobra：应用命令行框架，改天再分享。 -
viper：配置文件、环境变量、命令行、缓冲区。今天群里的每日一Go已经分享了，就不多说了。
命令行工具，你只要赋予可执行权限，再放到bin里面，他就会成为一个系统命令，方便你做更多的事情，比如备份数据库之类的动作，提高你的运维效率。
评论