Go实现cat(1)程序

在本节中,你将看到Go版本的cat(1)程序。你可能会对程序的长度感到惊讶!

cat.go源码分为三部分。第一部分如下:

  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "io"
  6. "os"
  7. )

cat.go第二部分代码如下:

  1. func printFile(filename string) error {
  2. f, err := os.Open(filename)
  3. if err != nil {
  4. return err
  5. }
  6. defer f.Close()
  7. scanner := bufio.NewScanner(f)
  8. for scanner.Scan() {
  9. io.WriteString(os.Stdout, scanner.Text())
  10. io.WriteString(os.Stdout, "\n")
  11. }
  12. return nil
  13. }

在本部分,你会看到如何实现将文件内容打印到标准输出。

cat.go最后一部分代码如下:

  1. func main() {
  2. filename := ""
  3. arguments := os.Args
  4. if len(arguments) == 1 {
  5. io.Copy(os.Stdout, os.Stdin)
  6. return
  7. }
  8. for i := 1; i < len(arguments); i++ {
  9. filename = arguments[i]
  10. err := printFile(filename)
  11. if err != nil {
  12. fmt.Println(err)
  13. }
  14. }
  15. }

前面的代码包含了cat.go的所有魔力,因为这是定义程序行为的地方。首先,如果在没有任何命令行参数的情况下执行cat.go,那么程序仅仅是通过io.Copy(os.Stdout,os.Stdin)将标准输入复制到标准输出。但是,如果有命令行参数,那么程序将按照给定的顺序处理它们。

执行cat.go将创建如下的输出:

  1. $go run cat.go
  2. Mastering Go
  3. Mastering Go
  4. 1 2 3 4
  5. 1 2 3 4

但是,如果使用Unix管道执行cat.go,结果会变得非常有趣:

  1. $ go run cat.go /tmp/1.log /tmp/2.log | wc
  2. 2367 44464 279292
  3. $ go run cat.go /tmp/1.log /tmp/2.log | go run cat.go | wc
  4. 2367 44464 279292

cat.go还可以在屏幕上打印多个文件。

  1. $ go run cat.go 1.txt 1 1.txt
  2. 2367 44464 279292
  3. 2367 44464 279292
  4. open 1: no such file or dictory
  5. 2367 44464 279292
  6. 2367 44464 279292

请注意,如果你试图通过go run cat.go cat.go来执行cat.go,并期望在屏幕上得到cat.go的内容,它会执行失败,并输出如下的错误信息:

  1. 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源文件作为生成的可执行文件的参数。