Golang实现简易的命令行功能
前言
一次偶然的想法,想知道为什么在终端输入那些命令行后,就执行了对应的操作,这转化为代码,应该怎么实现呢?
既然有了问题,那我们就来解决问题吧!
首先我认为想做命令行操作,那就得先”认识“命令行(当然这里指你的代码认识),所以我认位有两个步骤:
- 解析命令行
- 实现对应命令行的功能
话不多说开干!
开始
正好在学习 Golang
,那就用它的试试吧!
首先,我们先来学习以下几个 API
flag.String
// String defines a string flag with specified name, default value, and usage string. // The return value is the address of an int variable that stores the value of the flag. // String 定义了一个字符串标志,具有指定的名称、默认值和用法字符串。返回值是存储标志值的字符串变量的地址。 func String(name string, value string, usage string) *string { return CommandLine.String(name, value, usage) }
也就是说,使用-name value
的命令参数,usage
是对这个参数的说明,返回值是这个 value 的指针
,也就是用户输入在 -name
后的 value
。
flag.Int
// Int defines an int flag with specified name, default value, and usage string. // The return value is the address of an int variable that stores the value of the flag. // Int 定义了一个具有指定名称、默认值和用法字符串的 int 标志。返回值是存储标志值的 int 变量的地址。 func Int(name string, value int, usage string) *int { return CommandLine.Int(name, value, usage) }
使用方式和 String()
一样,只是类型的区别。
flag.StringVar
// StringVar defines a string flag with specified name, default value, and usage string. // The argument p points to a string variable in which to store the value of the flag. // StringVar 定义了一个带有指定名称、默认值和用法字符串的字符串标志。参数 p 指向一个字符串变量,用于存储标志的值。 func StringVar(p *string, name string, value string, usage string) { CommandLine.Var(newStringValue(value, p), name, usage) }
这里可以看到区别就是,将返回值指针,变成了函数的第一个参数。
flag.IntVar
// IntVar defines an int flag with specified name, default value, and usage string. // The argument p points to an int variable in which to store the value of the flag. // IntVar 定义了一个具有指定名称、默认值和用法字符串的 int 标志。参数 p 指向一个存储标志值的 int 变量。 func IntVar(p *int, name string, value int, usage string) { CommandLine.Var(newIntValue(value, p), name, usage) }
了解这些后,我们就开始吧!
定义命令行参数
package main // ...import func commandStyle() { methodPtr := flag.String("method", "default", "method of sample") valuePtr := flag.Int("value", -1, "value of sample") // 解析 flag.Parse() fmt.Println(*methodPtr, *valuePtr) } func commandStyle2() { var method string var value int flag.StringVar(&method, "method", "default", "method of sample") flag.IntVar(&value, "value", -1, "value of sample") flag.Parse() fmt.Println(method, value) } func main() { commandStyle() }
在终端使用 go run . -method get -value 1
这串命令后,打印出了 get 1
。
Parse 解析来自 os.Args[1:] 的命令行标志。必须在定义所有标志之后和程序访问标志之前调用。
这里的一个重要的点就是要使用 flag.Parse()
,也就是解析go run .` 之后的标志。使用变量将标志的值接收,然后打印。
这两种方式结果都是一样,只有写法上的差距,这时候虽然我们体验了一点简单的命令行的影子了,但似乎还是感觉好像啥效果也没有呀。接下来我们就来实现一个 copy 文件内容的功能
实现 -f -v 是否强制拷贝
首先我们使用上面说过的类似方式,注册标志 f,v
,然后解析标志
func main() { var showProgress, force bool // -f 当存在时拷贝,是否强制拷贝 flag.BoolVar(&force, "f", false, "force copy when existing") flag.BoolVar(&showProgress, "v", false, "explain what is being done") flag.Parse() // 获取参数个数,必须要输入两个参数,因为copy是从这个文件到另一个文件 if flag.NArg() < 2 { flag.Usage() // 打印用途 return } copyFileAction(flag.Arg(0), flag.Arg(1), showProgress, force) }
注册标志完成后,我们就可以开始实现我们的 copy
功能了
首先我们必须后面要输入两个文件名,让最后一个文件copy到前一个文件(制定规则)
我们模拟命令行输入:go run . -f -v a.txt b.txt
,这就是我们最后需要实现的东西,f, v
是可以省略的,默认把 a.txt -> b.txt
。
-f
表示当文件存在时,强制copy覆盖里面的内容-v
表示解释正在做什么
接下来我们需要实现一个 copyFileAction
函数,来实现copy功能,以及命令行参数的效果
copyFileAction 实现
func fileExist(fileName string)bool { _, err := os.Stat(fileName) // 返回这个文件信息 // IsExist 只是错误或报告是否存在 // err == nil,表示有文件信息,os.IsExist(err),表示有文件存在 return err == nil || os.IsExist(err) } // 转化操作,命令行,与功能实现的逻辑判断 func copyFileAction(src, dst string, showProgress, force bool) { if !force { // 判断是否存在文件,若存在,是否需要覆盖它 if fileExist(dst) { fmt.Printf("%s exists override? y/n \n", dst) reader := bufio.NewReader(os.Stdin) // 读取输入内容 data, _, _ := reader.ReadLine() // 取一行的内容 // 判断输入的内容 if strings.TrimSpace(string(data)) != "y" { return } } } // copy 文件 copyFile(src, dst) }
这里我们可以看到充分利用到了 showProgress 和 force
两个命令行取的值,当文件``存在且不强制`时,会有一个询问,是否覆盖,同意就实行 copy 操作,不同意不做处理(相当于一次无效命令)。
接下来我们实现功能核心 copyFile
copyFile
func copyFile(originFile, targetFile string)(written int64, err error){ srcFile, err := os.Open(originFile) // 打开文件 if err != nil { // Error() 返回错误信息 log.Fatal(err) return } defer srcFile.Close() dstFile, err := os.Create(targetFile) // 创建文件 if err != nil { // Error() 返回错误信息 log.Fatal(err) return } defer dstFile.Close() return io.Copy(dstFile, srcFile) // 拷贝文件 }
这里我们是采取,将需要被拷贝的文件打开,拷贝到的文件名进行创建,然后将内容填充进去,这里使用了io.Copy()
的内置功能。
效果图
完整代码
package main import ( "bufio" "flag" "fmt" "io" "log" "os" "strings" ) // func commandStyle() { // methodPtr := flag.String("method", "default", "method of sample") // valuePtr := flag.Int("value", -1, "value of sample") // // 解析 // flag.Parse() // fmt.Println(*methodPtr, *valuePtr) // } // func commandStyle2() { // var method string // var value int // flag.StringVar(&method, "method", "default", "method of sample") // flag.IntVar(&value, "value", -1, "value of sample") // flag.Parse() // fmt.Println(method, value) // } func main() { // commandStyle() // commandStyle2() var showProgress, force bool // -f 当存在时拷贝,是否强制拷贝 flag.BoolVar(&force, "f", false, "force copy when existing") flag.BoolVar(&showProgress, "v", false, "explain what is being done") flag.Parse() // 获取参数个数 if flag.NArg() < 2 { flag.Usage() // 打印用途 return } copyFileAction(flag.Arg(0), flag.Arg(1), showProgress, force) } func fileExist(fileName string)bool { _, err := os.Stat(fileName) // 返回这个文件信息 // IsExist 只是错误或报告是否存在 // err == nil,表示有文件信息,os.IsExist(err),表示有文件存在 return err == nil || os.IsExist(err) } func copyFile(originFile, targetFile string)(written int64, err error){ srcFile, err := os.Open(originFile) // 打开文件 if err != nil { // Error() 返回错误信息 log.Fatal(err) return } defer srcFile.Close() dstFile, err := os.Create(targetFile) // 创建文件 if err != nil { // Error() 返回错误信息 log.Fatal(err) return } defer dstFile.Close() return io.Copy(dstFile, srcFile) // 拷贝文件 } // 拷贝文件 func copyFileAction(src, dst string, showProgress, force bool) { if !force { // 判断是否存在文件,若存在,是否需要覆盖它 if fileExist(dst) { fmt.Printf("%s exists override? y/n \n", dst) reader := bufio.NewReader(os.Stdin) // 读取输入内容 data, _, _ := reader.ReadLine() // 取一行的内容 // 判断输入的内容 if strings.TrimSpace(string(data)) != "y" { return } } } // copy 文件 copyFile(src, dst) }
以上就是Golang实现简易的命令行功能的详细内容,更多关于Golang命令行功能的资料请关注猪先飞其它相关文章!
原文出处:https://juejin.cn/post/7198707460823253047
相关文章
- 这篇文章主要介绍了golang 调用 php7详解及实例的相关资料,需要的朋友可以参考下...2017-01-15
- 这篇文章主要介绍了C#命令行编译器配置方法,本文讲解了配置C#命令行编译器、配置其它.NET命令行工具、通过csc命令行编译器来编译C#文件实例等内容,需要的朋友可以参考下...2020-06-25
- 这篇文章主要介绍了用golang实现替换某个文件中的字符串操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-25
- 这篇文章主要介绍了解决Golang json序列化字符串时多了\的情况,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-24
- 这篇文章主要介绍了golang在GRPC中设置client的超时时间,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
- 这篇文章主要介绍了golang与php实现计算两个经纬度之间距离的方法,结合实例形式对比分析了Go语言与php进行经纬度计算的相关数学运算技巧,需要的朋友可以参考下...2016-07-29
- 这篇文章主要介绍了如何使用Swift来实现一个命令行工具,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-06-30
解决golang处理http response碰到的问题和需要注意的点
这篇文章主要介绍了解决golang处理http response碰到的问题和需要注意的点,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-16Golang中的自定义类型之间的转换的实现(type conversion)
这篇文章主要介绍了Golang中的自定义类型之间的转换的实现(type conversion),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-21- 这篇文章主要介绍了golang http使用踩过的坑与填坑指南,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
- Go(又称Golang)是Google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的编程语言。这篇文章给大家介绍golang 正则表达式的相关知识,感兴趣的朋友跟随小编一起看看吧...2021-05-07
- 这篇文章主要介绍了golang文件读取-按指定BUFF大小读取方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-22
- 这次文章为大家带来的是一个比较实用的示例:利用Golang生成整数随机数,对此感兴趣的可以一起来看看。 php随机数生成一个给定范围的随机数,用 PHP 就太简单不过了,而...2017-07-06
- 这篇文章主要介绍了Jmeter如何基于命令行运行jmx脚本,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下...2020-07-22
- 这篇文章主要介绍了golang DNS服务器的简单实现操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-01
- 这篇文章主要介绍了golang中json和struct的使用说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-01
Fedora14 Linux系统安装Golang开发环境笔记
这篇文章主要介绍了Fedora14 Linux系统安装Golang开发环境笔记,本文讲解了2种安装方法,需要的朋友可以参考下...2020-05-01- 这篇文章主要介绍了基于命令行执行带参数的php脚本并取得参数的方法,分析了命令行运行PHP代码的原理、用法与相关注意事项,需要的朋友可以参考下...2016-01-26
- 这篇文章主要介绍了golang去除多余空白字符(含制表符)的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-25
golang elasticsearch Client的使用详解
这篇文章主要介绍了golang elasticsearch Client的使用详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-04