首页 > 编程笔记 > Go语言笔记

Go语言使用切片读写文件

切片提供了 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[:]); true {
  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. }
下面的代码使用了 os 包中的 os.file 和 Read 方法；与上一节示例代码具有同样的功能。
  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() // 扫描arg列表并设置标志
  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. }

所有教程

优秀文章