12.6 用切片读写文件

切片提供了 Go 中处理 I/O 缓冲的标准方式,下面 cat 函数的第二版中,在一个切片缓冲内使用无限 for 循环(直到文件尾部 EOF)读取文件,并写入到标准输出(os.Stdout)。

  1. func cat(f *os.File) {
  2. const NBUF = 512
  3. var buf [NBUF]byte
  4. for {
  5. switch nr, err := f.Read(buf[:]); {
  6. case nr < 0:
  7. fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error())
  8. os.Exit(1)
  9. case nr == 0: // EOF
  10. return
  11. case nr > 0:
  12. if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {
  13. fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error())
  14. }
  15. }
  16. }
  17. }

上面的代码来自于 cat2.go,使用了 os 包中的 os.FileRead 方法;cat2.gocat.go 具有同样的功能。

示例 12.14 cat2.go

  1. package main
  2. import (
  3. "flag"
  4. "fmt"
  5. "os"
  6. )
  7. func cat(f *os.File) {
  8. const NBUF = 512
  9. var buf [NBUF]byte
  10. for {
  11. switch nr, err := f.Read(buf[:]); true {
  12. case nr < 0:
  13. fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error())
  14. os.Exit(1)
  15. case nr == 0: // EOF
  16. return
  17. case nr > 0:
  18. if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr {
  19. fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error())
  20. }
  21. }
  22. }
  23. }
  24. func main() {
  25. flag.Parse() // Scans the arg list and sets up flags
  26. if flag.NArg() == 0 {
  27. cat(os.Stdin)
  28. }
  29. for i := 0; i < flag.NArg(); i++ {
  30. f, err := os.Open(flag.Arg(i))
  31. if f == nil {
  32. fmt.Fprintf(os.Stderr, "cat: can't open %s: error %s\n", flag.Arg(i), err)
  33. os.Exit(1)
  34. }
  35. cat(f)
  36. f.Close()
  37. }
  38. }

链接