用go实现linux命令行

要求：实现selpg命令，使用Go语言

以下为该命令参数形式

-s start_page -e end_page [ -f | -l lines_per_page ] [ -d dest ] [ in_filename ]

各参数意义如下

-s start page 表示打印开始的页数
-e end page 表示打印结束页数
-f 可选，表示文件是否以f作为分页标志，与-l不能一起使用
-l 可选，表示文件几行作为一页，与-f不能一起使用
-d 可选，目标打印机
in_filename 可选，读入的文件，若不选则从标准流输入

A.关于实现过程个人觉得关键几个部分

1.关于字符串部分

根据老师教程是用os包获取命令行参数然后用flag包解析参数
这里os包只需简单调用os.Arg来获取参数即可，这里我们说一下flag包，flag包一般用来解析命令行参数
flag包主要包含一些方法和一个flagSet，后者只是对前者做了封装，这里我选择了用flagSet

/*函数原型*/
func NewFlagSet(name string, errorHandling ErrorHandling) *FlagSet
/*例子*/
flagSet   = flag.NewFlagSet(os.Args[0], flag.ExitOnError)

这是创建一个flagSet，其中第一个参数是程序名字，第二个是参数错误时的错误处理方法
这里os.Arg[0]是程序名字，然后我选择了错误时就终止程序的类型
接下来是定义几个参数

startPage = flagSet.Uint("s", 0, "tart_page, must be specified and greter than 0")
endPage   = flagSet.Uint("e", 0, "end_page, must be specified and greter than 0")
pageLen   = flagSet.String("l", "", "page_size, exclusive to -f, must be postive,default is 72")
pageType  = flagSet.Bool("f", false, "use \f to divide pages, exclusive to -l")

以上的几个方法将会解析参数并将对应指针返回，对应的几个参数分别为参数的flag，参数默认值，参数描述
这是函数原型,创建flagSet对象后通过以上可方法定义参数

func Uint(name string, value uint, usage string) *uint

定义了参数后我们可以通过调用Parse方法，flagSet就会帮我们解析命令行内输入的字符串然后赋值给对应指针，如果用户写的参数错误就会按照你选择的错误方式处理
这里我还自定义了一种类型，flag中你可以通过Var等方法来给参数赋值自定义类型，但要实现Set()和String()两个接口用于赋值和转字符串
下面是我代码中的例子

type StringArray []string

func (s *StringArray) Set(value string) error {
    *s = append(*s, value)
    return nil
}
func (s *StringArray) String() string {
    return fmt.Sprint([]string(*s))
}
func (s *StringArray) Size() int {
    return len(*s)
}

func (s *StringArray) Top() (string, error) {
    if s.Size() > 0 {
        var temp = s.String()
        temp = strings.TrimSuffix(temp, "[")
        temp = strings.TrimSuffix(temp, "]")
        return temp, nil
    }
    return "", fmt.Errorf("empty error")
}

根据官网文档建议
If you like, you can bind the flag to a variable using the Var() functions.

var flagvar int
func init() {
flag.IntVar(&flagvar, “flagname”, 1234, “help message for flagname”)
}

因此我在init函数里面对我的一个参数进行了定义

flagSet.Var(&printDest, "d", "destination")

剩下的就是获得参数后大概判断是不是合法参数，这些想必不是什么难点写写判断就可以

2.关于流和读取文件还有lp命令部分

关于流部分，首先要确定接受文件的流从哪输入过来
如果没有文件参数的话就是标准流，有的话就用文件流

var inputStream = os.Stdin
if inputFile != "" {
    inputStream, err = os.Open(inputFile)
    defer inputStream.Close()
    if err != nil {
        fmt.Fprintf(os.Stderr, "%s : ", programname)
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

这里使用os的open函数来打开文件流
然后关于这个流要输出到哪里，我们需要确定，如果存在目的打印机我们把我们读取的文件通过管道传到lp命令的输入流，这里我用bufio的reader来读文件，然后调用os/exec的StdinPipe()方法来将文件输入，当然如果用户没定义-d参数则将输出流输出到屏幕上即可
关键代码如下

stdin = os.Stdout
    if printDest.Size() > 0 && pDest != "" {
        printer = exec.Command("lp", "-d", pDest)
        //printer = exec.Command("cat")
        stdin, printErr = printer.StdinPipe()
        if printErr != nil {
            fmt.Fprintf(os.Stderr, "%s : ", programname)
            fmt.Fprintln(os.Stderr, printErr)
        }
        if *pageType {
            err = writeIntoPrintInByF(myReader, *startPage, *endPage)
        } else {
            err = writeIntoPrintInByLine(myReader, *startPage, *endPage, pLen)
        }
        if err != nil {
            fmt.Fprintf(os.Stderr, "%s : ", programname)
            fmt.Fprintln(os.Stderr, err)
        }
        stdin.Close()

然后这是将读取文件写入流中，这里我调用了Write方法

stdin.Write([]byte(line))

其中line是我调用reader的readString方法获得的字符串，当然也有其他库函数实现读取字符串

line, err := reader.ReadString('n')

最后是启动命令，这里如果命令出错我们要杀进程，这里我用timer监听实现

printer.Stdout = os.Stdout
        printer.Stderr = os.Stderr
        if err = printer.Start(); err != nil {
            fmt.Fprintf(os.Stderr, "%s : ", programname)
            fmt.Fprintln(os.Stderr, printErr)
        }
        timer := time.AfterFunc(3*time.Second, func() {
            printer.Process.Kill()
        })
        err = printer.Wait()
        timer.Stop()

3.关于源码

源码位于我的github：https://github.com/caijh23/Go/tree/master/selpg
代码分为两个文件args.go和slepg.go分别为上述讲的两个部分

4.结果展示

a.txt里面有两行数据均为123
这是没有打印机

这是输出到屏幕

这是参数错误

这是将lp命令换为cat的成功结果