用golang实现linux下命令selpg

标签（空格分隔）： 服务计算 linux selpg golang

其实主要是参照了c语言的selpg的写法，相对有些机械的翻译了。

1.设计selpg结构体
给selpg设计一个结构体，分别保存各个参数的值，使其易于处理。

type Selpg struct {
	Begin int //起始页
	End int //结束页
	PageType bool //是否为-f类型命令，是则为真
	Length int //页长度
	Destination string //打印机名称
	Infile string//输入文件
	data []string//储存文件数据的字符串
}

2.利用flag包解析参数
首先定义一个flag。

flag.Usage = func() {
		fmt.Printf("Usage of seplg:n")
		fmt.Printf("seplg -s num1 -e num2 [-f -l num3 -d str1 file]n")
		flag.PrintDefaults()
	}

然后利用flag参数来解析各个参数

flag.IntVar(&(args.Begin), "s", -1, "the start page")
	flag.IntVar(&(args.End), "e", -1, "the end page")
	flag.IntVar(&(args.Length), "l", -1, "page length")
	flag.StringVar(&(args.Destination), "d", "", "print destionation")
	flag.BoolVar(&(args.PageType), "f", false, "type of print")

3.处理分页
分页的时候主要需要应对两种情况。一种是-f以/f为分页的标志。另一种是-l指定分页的行数。使用scanner来读取每一行获取其信息。
-f时搜寻f符号，利用split函数将文本分隔开，每个作为一页。

onSp := func(data []byte, atEOF bool) (advance int, token []byte, err error) {
			for i := 0; i < len(data); i++ {
				if data[i] == 'f' {
					return i + 1, data[:i], nil
				}
			}
			if atEOF {
				return 0, data, bufio.ErrFinalToken
			} else {
				return 0, nil, nil
			}
		}
		scanner.Split(onSp)

-l时则计算行数，当行数打到指定行数进行分页。

for scanner.Scan() {
			line := scanner.Text()
			if count / selpg.Length + 1 >= selpg.Begin && count / selpg.Length  < selpg.End {
				selpg.data = append(selpg.data, line)
			}
			count++
		}

4.重定向输入输出
这里采用的方法是将读入的内容储存到一个string字符串中，等到需要输出时再重定向到标准输出或是打印机上。不过更好的方法是通过管道把selpg输出给另一个命令。方法是调用函数 exec.Command(“命令名”, “该命令的相关参数”, …) 即可返回获得该命令结构，再调用 cmd.StdinPipe() 可以获得管道的输入管道，然后把我们的 selpg 输出到该管道即可。 这里给出范例。

if sa.printDest != "" {
    cmd := exec.Command("grep", "-nf", "keyword")
    inpipe, err = cmd.StdinPipe()
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    defer inpipe.Close()
    cmd.Stdout = fout
    cmd.Start()
}

##测试结果

selpg -s=1 -e=1 a.txt

该命令将把a.txt的第1页写至标准输出（也就是屏幕）。

selpg -s=1 -e=1 < a.txt

该命令效果同上，在本例中，selpg 读取标准输入，而标准输入已被 shell／内核重定向为来自“input_file”而不是显式命名的文件名参数。输入的第 1 页被写至屏幕。

selpg -s=1 -e=1 a.txt >b.txt 

将a.txt的第 1 页写至标准输出；标准输出被 shell／内核重定向至“b.txt"。即将a.txt中内容输入至b.txt中。

打开b.txt

重定向成功

selpg -s=1 -e=1 -l=3 a.txt  

从a.txt输入，并输出到命令行，打印前三行

selpg -s=1 -e=2 -l=3 a.txt  

从a.txt输入，并输出到命令行，打印两页，总共六行

selpg -s=2 -e=2  -l=3 a.txt  

打印出a.txt的第二页

selpg -s=1 -e=1 -f a.txt

根据换页符-f分页，打出第一页

 selpg -s=1 -e=3 -f a.txt

根据换页符-f分页，打出前三页

 selpg -s=1 -e a.txt 2>error.txt

这里故意输错命令，将错误信息写入error.txt

打开error.txt可以看到错误信息