12.6 用切片读写文件

最后更新于:2022-04-01 20:57:11

## 12.6 用切片读写文件 切片提供了 Go 中处理 I/O 缓冲的标准方式,下面 `cat` 函数的第二版中,在一个切片缓冲内使用无限 for 循环(直到文件尾部 EOF)读取文件,并写入到标准输出(`os.Stdout`)。 ~~~ func cat(f *os.File) { const NBUF = 512 var buf [NBUF]byte for { switch nr, err := f.Read(buf[:]); true { case nr < 0: fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error()) os.Exit(1) case nr == 0: // EOF return case nr > 0: if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr { fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error()) } } } } ~~~ 下面的代码来自于 `cat2.go`,使用了 os 包中的 `os.file` 和 `Read` 方法;`cat2.go` 与 `cat.go` 具有同样的功能。 示例 12.14 [cat2.go](https://github.com/Unknwon/the-way-to-go_ZH_CN/blob/master/eBook/examples/chapter_12/cat2.go): ~~~ package main import ( "flag" "fmt" "os" ) func cat(f *os.File) { const NBUF = 512 var buf [NBUF]byte for { switch nr, err := f.Read(buf[:]); true { case nr < 0: fmt.Fprintf(os.Stderr, "cat: error reading: %s\n", err.Error()) os.Exit(1) case nr == 0: // EOF return case nr > 0: if nw, ew := os.Stdout.Write(buf[0:nr]); nw != nr { fmt.Fprintf(os.Stderr, "cat: error writing: %s\n", ew.Error()) } } } } func main() { flag.Parse() // Scans the arg list and sets up flags if flag.NArg() == 0 { cat(os.Stdin) } for i := 0; i < flag.NArg(); i++ { f, err := os.Open(flag.Arg(i)) if f == nil { fmt.Fprintf(os.Stderr, "cat: can't open %s: error %s\n", flag.Arg(i), err) os.Exit(1) } cat(f) f.Close() } } ~~~
';