Golang实现简易的命令行功能
哈喽!今天心血来潮给大家带来了《Golang实现简易的命令行功能》,想必大家应该对Golang都不陌生吧,那么阅读本文就都不会很困难,以下内容主要涉及到命令行,若是你正在学习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() 注册标志完成后,我们就可以开始实现我们的 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() 以上就是《Golang实现简易的命令行功能》的详细内容,更多关于golang的资料请关注golang学习网公众号!
Mysql查询优化之IN子查询优化方法详解
- 上一篇
- Mysql查询优化之IN子查询优化方法详解
- 下一篇
- Mac上安装Mysql的详细步骤及配置
-
- Golang · Go教程 | 3小时前 | 格式化输出 printf fmt库 格式化动词 Stringer接口
- Golangfmt库用法与格式化技巧解析
- 140浏览 收藏
-
- Golang · Go教程 | 3小时前 |
- Golang配置Protobuf安装教程
- 147浏览 收藏
-
- Golang · Go教程 | 3小时前 |
- Golang中介者模式实现与通信解耦技巧
- 378浏览 收藏
-
- Golang · Go教程 | 4小时前 |
- Golang多协程通信技巧分享
- 255浏览 收藏
-
- Golang · Go教程 | 4小时前 |
- Golang如何判断变量类型?
- 393浏览 收藏
-
- Golang · Go教程 | 4小时前 |
- Golang云原生微服务实战教程
- 310浏览 收藏
-
- Golang · Go教程 | 5小时前 |
- Golang迭代器与懒加载结合应用
- 110浏览 收藏
-
- Golang · Go教程 | 5小时前 | 性能优化 并发安全 Golangslicemap 预设容量 指针拷贝
- Golangslicemap优化技巧分享
- 412浏览 收藏
-
- Golang · Go教程 | 5小时前 |
- Golang代理模式与访问控制实现解析
- 423浏览 收藏
-
- Golang · Go教程 | 5小时前 |
- Golang事件管理模块实现教程
- 274浏览 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 543次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 516次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 500次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 485次学习
-
- ChatExcel酷表
- ChatExcel酷表是由北京大学团队打造的Excel聊天机器人,用自然语言操控表格,简化数据处理,告别繁琐操作,提升工作效率!适用于学生、上班族及政府人员。
- 3164次使用
-
- Any绘本
- 探索Any绘本(anypicturebook.com/zh),一款开源免费的AI绘本创作工具,基于Google Gemini与Flux AI模型,让您轻松创作个性化绘本。适用于家庭、教育、创作等多种场景,零门槛,高自由度,技术透明,本地可控。
- 3376次使用
-
- 可赞AI
- 可赞AI,AI驱动的办公可视化智能工具,助您轻松实现文本与可视化元素高效转化。无论是智能文档生成、多格式文本解析,还是一键生成专业图表、脑图、知识卡片,可赞AI都能让信息处理更清晰高效。覆盖数据汇报、会议纪要、内容营销等全场景,大幅提升办公效率,降低专业门槛,是您提升工作效率的得力助手。
- 3405次使用
-
- 星月写作
- 星月写作是国内首款聚焦中文网络小说创作的AI辅助工具,解决网文作者从构思到变现的全流程痛点。AI扫榜、专属模板、全链路适配,助力新人快速上手,资深作者效率倍增。
- 4509次使用
-
- MagicLight
- MagicLight.ai是全球首款叙事驱动型AI动画视频创作平台,专注于解决从故事想法到完整动画的全流程痛点。它通过自研AI模型,保障角色、风格、场景高度一致性,让零动画经验者也能高效产出专业级叙事内容。广泛适用于独立创作者、动画工作室、教育机构及企业营销,助您轻松实现创意落地与商业化。
- 3785次使用
-
- Golang操作命令行的几种方式总结
- 2022-12-27 426浏览
-
- Golang执行cmd命令行的方法
- 2023-01-20 412浏览
-
- golang执行命令行的实现
- 2023-01-07 109浏览
-
- Go语言中使用urfave/cli命令行框架
- 2022-12-31 386浏览
-
- golang实现命令行程序的使用帮助功能
- 2023-01-09 269浏览

