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 valueusagevalue 的指针-namevalue

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 1get 1

Parse 解析来自 os.Args[1:] 的命令行标志。必须在定义所有标志之后和程序访问标志之前调用。

flag.Parse()

这两种方式结果都是一样，只有写法上的差距，这时候虽然我们体验了一点简单的命令行的影子了，但似乎还是感觉好像啥效果也没有呀。接下来我们就来实现一个 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, va.txt -> b.txt

-f-v

copyFileAction

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

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)
}