Go开发命令行程序

一、开发实践要求
二、设计说明
三、代码实现
四、测试结果
五、代码地址
六、心得体会

一、开发实践要求

使用 golang 开发开发 Linux 命令行实用程序中的 selpg

提示：

请按文档 使用 selpg 章节要求测试你的程序
请使用 pflag 替代 goflag 以满足 Unix 命令行规范，参考：Golang之使用Flag和Pflag
golang 文件读写、读环境变量，请自己查 os 包
“-dXXX” 实现，请自己查 os/exec 库，例如案例 Command，管理子进程的标准输入和输出通常使用 io.Pipe，具体案例见 Pipe

二、设计说明

本次开发的是一个命令行程序，总体的流程设计如下：

在流程中重点要解决的几个问题：

1. 参数提取

本次开发的是一个命令行程序，程序的外部输入就是输入的命令及其参数，那么首先要考虑如何获得输入的命令内容并进行解析、验证，在示例c代码中是通过main函数直接获得字符串类型的输入，然后对它进行拆分，从头逐步解析同时验证。在Go中，main函数并没有传入参数，而是可以通过os包或者flag（pflag）包就可以进行快速方便地提取参数，本次开发我选择pflag，使用IntP，Parse等函数。

2. 错误处理

在程序执行流程中会产生许多错误，例如：参数使用不当、文件读取异常等，程序需要在出错时返回有用的信息并结束。因此，可以在出错流程中适时输出出错原因或正确用法，然后使用os.Exit(int)退出，在函数中使用不同的错误码来识别不同错误。但是文件处理时不可以直接使用os.Exit(int)退出，此时文件流还未关闭，此时可以使用panic函数配合defer来确保文件流的关闭（panic函数会在处理完错误后执行defer的代码），所以本代码统一使用panic处理错误。

3.输入和输出处理

来源

输入有两种来源，标准输入和指定文件，我们只需要一视同仁地转换为io.Reader接口就可以，前者通过fmt.Stdin（类型为*File）获取，后者通过os.open()获取（返回类型同样为*File）。

输出目的类似，os.Stdout和cmd.StdinPipe()，后者是使用管道进行数据流的转移，使用了os/exec，同样都把他们转换为os.Writer接口。

方式

输入有两种方式：按行和按\f分割符，bufio包中有很多处理流的方法可以帮助我们节省很多精力，bufio.Reader中，ReadLine()方法可以按行读取，ReadSlice(‘\f’)可以按\f分割符读取。
流的关闭

输入源的关闭上文已说过使用defer file.close()就可以确保关闭。

输出源的关闭主要考虑cmd中管道流的关闭。但是我们看官方Go doc中StdinPipe()的说明：

The pipe will be closed automatically after Wait sees the command exit. A caller need only call Close to force the pipe to close sooner.

也就是说pipe会自动关闭，所以可以不考虑此输出源的关闭。

三、代码实现

以下为重点实现：

命令的存储结构体：

type scom struct {
	st, ed int //start,end page.No
	isl bool //true for line based, or \f based
	lnum int //line number. Used when isl == true
	dest string //Output destination. Empty string means Stdout
	fn string //Input source. Empty string means Stdin
}

main函数的流程：

var sc scom
parseArgs(&sc)
//Show the print information read just now
printCom(&sc)
processInput(&sc)

参数提取：

st := pflag.IntP("start", "s", -1, "The to-print start page.No")
//...
pflag.Usage = usage
pflag.Parse()

输入源获取，重点：使用defer处理文件流关闭：

var reader io.Reader
var file os.File
if sc.fn != "" {
	_, errEx:= os.Stat(sc.fn)
    //...
	file, err := os.Open(sc.fn)
	//...
	reader = file
} else {
	file := os.Stdin
	reader = file
}
defer file.Close()

输出到bytes.Buffer

for {
	bs, err := rd.ReadSlice('\f')
	if cp >= sc.st && cp <= sc.ed {
		buf.Write(bs)
		buf.WriteByte('\f')
	}
	cp++
	if err == io.EOF {
		cp--
		break
	}
}

通过exec获得“-dXXX”的输出流：

cmd := exec.Command("lp", "-d"+dest)
fout, err := cmd.StdinPipe()
if err != nil {
    //...
}
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
errr := cmd.Run()
if errr != nil {
    //...
}