Go 读取文件

读写文件是很多程序的基本任务,下面我们看看Go里面的文件读取。

  1. package main
  2. import (
  3. "bufio"
  4. "fmt"
  5. "io"
  6. "io/ioutil"
  7. "os"
  8. )
  9. // 读取文件的函数调用大多数都需要检查错误,
  10. // 使用下面这个错误检查方法可以方便一点
  11. func check(e error) {
  12. if e != nil {
  13. panic(e)
  14. }
  15. }
  16. func main() {
  17. // 最基本的文件读写任务就是把整个文件的内容读取到内存
  18. dat, err := ioutil.ReadFile("/tmp/dat")
  19. check(err)
  20. fmt.Print(string(dat))
  21. // 有的时候你想更多地控制到底是读取文件的哪个部分,这个
  22. // 时候你可以使用`os.Open`打开一个文件获取一个`os.File`
  23. // 对象
  24. f, err := os.Open("/tmp/dat")
  25. // 从这个文件中读取一些字节,并且由于字节数组长度所限,
  26. // 最多读取5个字节,另外还需要注意实际能够读取的字节
  27. // 数量
  28. b1 := make([]byte, 5)
  29. n1, err := f.Read(b1)
  30. check(err)
  31. fmt.Printf("%d bytes: %s\n", n1, string(b1))
  32. // 你也可以使用`Seek`来跳转到文件中的一个已知位置,并从
  33. // 那个位置开始读取数据
  34. o2, err := f.Seek(6, 0)
  35. check(err)
  36. b2 := make([]byte, 2)
  37. n2, err := f.Read(b2)
  38. check(err)
  39. fmt.Printf("%d bytes @ %d: %s\n", n2, o2, string(b2))
  40. // `io`包提供了一些帮助文件读取的函数。例如上面的方法如果
  41. // 使用方法`ReadAtLeast`函数来实现,将使得程序更健壮
  42. o3, err := f.Seek(6, 0)
  43. check(err)
  44. b3 := make([]byte, 2)
  45. n3, err := io.ReadAtLeast(f, b3, 2)
  46. check(err)
  47. fmt.Printf("%d bytes @ %d: %s\n", n3, o3, string(b3))
  48. // 没有内置的rewind方法,但是可以使用`Seek(0,0)`来实现
  49. _, err = f.Seek(0, 0)
  50. check(err)
  51. // `bufio`包提供了缓冲读取文件的方法,这将使得文件读取更加
  52. // 高效
  53. r4 := bufio.NewReader(f)
  54. b4, err := r4.Peek(5)
  55. check(err)
  56. fmt.Printf("5 bytes: %s\n", string(b4))
  57. // 最后关闭打开的文件。一般来讲这个方法会在打开文件的时候,
  58. // 使用defer来延迟关闭
  59. f.Close()
  60. }

在运行程序之前,你需要创建一个/tmp/dat文件,然后写入一些测试数据。
运行结果

  1. hello world
  2. i am jemy
  3. who are you
  4. what do you like
  5. 5 bytes: hello
  6. 2 bytes @ 6: wo
  7. 2 bytes @ 6: wo
  8. 5 bytes: hello