用Go
实现cat(1)
程序
在本节中,你将看到Go
版本的cat(1)
程序。你可能会对程序的长度感到惊讶!
cat.go
源码分为三部分。第一部分如下:
package main
import (
"bufio"
"fmt"
"io"
"os"
)
cat.go
第二部分代码如下:
func printFile(filename string) error {
f, err := os.Open(filename)
if err != nil {
return err
}
defer f.Close()
scanner := bufio.NewScanner(f)
for scanner.Scan() {
io.WriteString(os.Stdout, scanner.Text())
io.WriteString(os.Stdout, "\n")
}
return nil
}
在本部分,你会看到如何实现将文件内容打印到标准输出。
cat.go
最后一部分代码如下:
func main() {
filename := ""
arguments := os.Args
if len(arguments) == 1 {
io.Copy(os.Stdout, os.Stdin)
return
}
for i := 1; i < len(arguments); i++ {
filename = arguments[i]
err := printFile(filename)
if err != nil {
fmt.Println(err)
}
}
}
前面的代码包含了cat.go
的所有魔力,因为这是定义程序行为的地方。首先,如果在没有任何命令行参数的情况下执行cat.go
,那么程序仅仅是通过io.Copy(os.Stdout,os.Stdin)
将标准输入复制到标准输出。但是,如果有命令行参数,那么程序将按照给定的顺序处理它们。
执行cat.go
将创建如下的输出:
$go run cat.go
Mastering Go
Mastering Go
1 2 3 4
1 2 3 4
但是,如果使用Unix
管道执行cat.go
,结果会变得非常有趣:
$ go run cat.go /tmp/1.log /tmp/2.log | wc
2367 44464 279292
$ go run cat.go /tmp/1.log /tmp/2.log | go run cat.go | wc
2367 44464 279292
cat.go
还可以在屏幕上打印多个文件。
$ go run cat.go 1.txt 1 1.txt
2367 44464 279292
2367 44464 279292
open 1: no such file or dictory
2367 44464 279292
2367 44464 279292
请注意,如果你试图通过go run cat.go cat.go
来执行cat.go
,并期望在屏幕上得到cat.go
的内容,它会执行失败,并输出如下的错误信息:
package main: case-insensitive file name collision: "cat.go" and "cat.go"
原因是Go
不理解第二个cat.go
应该作为命令行参数来运行cat.go
。相反,go run
试图编译cat.go
两次,从而导致错误消息。解决方案是先执行go build cat.go
,然后使用cat.go
或任何其他go
源文件作为生成的可执行文件的参数。