快捷導(dǎo)航

Go語言讀取文本文件的三種方式總結(jié)

更新時間：2023年01月12日 11:28:42 作者：宇宙之一粟

工作中時不時需要讀取文本，文本文件是最常見的文件類型。本文將利用Go語言從逐行、逐個單詞和逐個字符三個方法讀取文件，感興趣的可以了解一下

1 逐行讀取文本文件

逐行讀取文件是最為常見的文本文件，也是最為簡單的方式。首先我們需要導(dǎo)入幾個常見的包：

bufio：緩存區(qū)讀寫文件
flag：命令行參數(shù)解析

package main

import (
  "bufio"
  "flag"
  "fmt"
  "io"
  "os"
)

func lineByLine(file string) error {

  var err error

  f, err := os.Open(file)
  if err != nil {
    return err
  }

  defer f.Close()

  r := bufio.NewReader(f)
  for {
    line, err := r.ReadString('\n')
    if err == io.EOF {
      break
    } else if err != nil {
      fmt.Printf("error reading file %s", err)
      break
    }
    fmt.Print(line)
  }
  return nil
}

func main() {
  flag.Parse()
  if len(flag.Args()) == 0 {
    fmt.Printf("usage: byLine <file1> [<file2> ...]\n")
    return
  }

  for _, file := range flag.Args() {
    err := lineByLine(file)
    if err != nil {
      fmt.Println(err)
    }
  }
}

代碼解釋：

主要通過 bufio.NewReader() 函數(shù)生成一個新的讀取器；
隨后，在 bufio.ReadString() 函數(shù)讀取字符，通知該函數(shù)持續(xù)執(zhí)行讀取任務(wù)，直到碰到該 "\n" 參數(shù)，也就是換行符。讀到換行符，執(zhí)行文本輸出。
如果讀取中斷了，即 err == io.EOF ,退出文件讀取
或者 err != nil, 打印錯誤提示，退出文件執(zhí)行

main() 函數(shù)中首先讀取命令行參數(shù)，如果命令行長度為 0，即沒有傳入要讀取的文件，如果此時執(zhí)行 byLine.go 文件的話就會給出語法提示，如下:

$ go run byLine.go
usage: byLine <file1> [<file2> ...]

我們寫一個測試的文本文件 test.txt, 寫入如下幾行數(shù)據(jù)，記得在第二行換行（加入空行）：

這是第一行
我是第二行

運(yùn)行如下命令后，結(jié)果為：

$ go run byLine.go test.txt
這是第一行
我是第二行

可以使用 cat test.txt 校驗我們的結(jié)果的準(zhǔn)確性，如下：

$ cat test.txt
這是第一行
我是第二行

2 逐個單詞讀取文本文件

package main

import (
  "bufio"
  "flag"
  "fmt"
  "os"
)

func wordByWord(file string) error {
  var err error
  f, err := os.Open(file)
  if err != nil {
    return err
  }

  defer f.Close()
  scanner := bufio.NewScanner(f)
  scanner.Split(bufio.ScanWords)
  var words []string
  for scanner.Scan() {
    words = append(words, scanner.Text())
  }

  for _, word := range words {
    fmt.Println(word)
  }
  return nil
}

func main() {
  flag.Parse()
  if len(flag.Args()) == 0 {
    fmt.Printf("usage: byWord <file1> [file2> ...]\n")
    return
  }

  for _, file := range flag.Args() {
    err := wordByWord(file)
    if err != nil {
      fmt.Println(err)
    }
  }
}

代碼解釋：

其他代碼都和 byLine.go 函數(shù)一樣，主要是利用了 bufio 中的 scanner 來掃描單詞，
scanner := bufio.NewScanner(file) 用來掃描讀取的文件
scanner.Split(bufio.ScanWords) 用來分割單詞
聲明一個單詞字符串列表，將讀取到的每一個單詞放入這個列表中
循環(huán)遍歷單詞字符串列表，打印每一個單詞

測試代碼

寫入一個 test.txt 文件：

Hello World
1 2 3

運(yùn)行代碼，結(jié)果顯示：

$ go run byWord.go test.txt
Hello
World
1
2
3

3 逐個字符讀取文本文件

逐個字符讀取文本的使用場景還是很少，除非開發(fā)一個文本編輯器。新建一個 byCharacter.go 文件，然后寫入如下代碼：

package main

import (
  "bufio"
  "flag"
  "fmt"
  "io"
  "os"
)

func charByChar(file string) error {

  var err error
  f, err := os.Open(file)
  if err != nil {
    return err
  }

  defer f.Close()

  r := bufio.NewReader(f)
  for {
    line, err := r.ReadString('\n')
    if err == io.EOF {
      break
    } else if err != nil {
      fmt.Printf("error reading file %s", err)
      return err
    }

    for _, x := range line {
      fmt.Println(string(x))
    }
  }
  return nil
}

func main() {
  flag.Parse()
  if len(flag.Args()) == 0 {
    fmt.Printf("usage: byWord <file1> [file2> ...]\n")
    return
  }

  for _, file := range flag.Args() {
    err := charByChar(file)
    if err != nil {
      fmt.Println(err)
    }
  }
}

運(yùn)行測試用例得出的最后結(jié)果為：