go读写文件

读取文件

读取文件一般有三种方式，不同读取模式的优缺点不同。

• 将文件整个读入内存。效率比较高，占用内存也最高
• 按字节读取。读取方式比较灵活，若文件较大，可以读取需要的部分
• 按行读取。一般用于文件日志文件读取

将文件整个读入内存

package main

import (
   "os"
   "io/ioutil"
   "fmt"
)

func main() {
   filepath := "./log.txt"
   content ,err :=ioutil.ReadFile(filepath)
   if err !=nil {
      panic(err)
   }
   fmt.Println(string(content))
}

按字节读取

package main

import (
   "fmt"
   "io"
   "os"
)

func main() {

   file := "./log.txt"
   f, err := os.Open(file)
   if err != nil {
      panic(err)
   }
   defer f.Close()

   chunks := make([]byte, 0)
   buf := make([]byte, 1024)
   for {
      n, err := f.Read(buf)
      if err != nil && err != io.EOF {
         panic(err)
      }
      if 0 == n {
         break
      }
      chunks = append(chunks, buf[:n]...)
   }
   fmt.Println(string(chunks))
}

按字行读取

package main

import (
"bufio"
"fmt"
"io"
"io/ioutil"
"os"
"strings"
)

func main() {
    filepath := "./log.txt"
    file, err := os.OpenFile(filepath, os.O_RDWR, 0666)
    if err != nil {
        fmt.Println("Open file error!", err)
        return
    }
    defer file.Close()

    stat, err := file.Stat()
    if err != nil {
        panic(err)
    }
    var size = stat.Size()
    fmt.Println("file size=", size)

    buf := bufio.NewReader(file)
    for {
        line, err := buf.ReadString('\n')
        line = strings.TrimSpace(line)
        fmt.Println(line)
        if err != nil {
            if err == io.EOF {
                fmt.Println("File read ok!")
                break
            } else {
                fmt.Println("Read file error!", err)
                return
            }
        }
    }
}

写入文件

写入文件一般分为二种：

• 写入整个文件，如果文件存在覆盖，不存在创建
• 在文件尾部添加内容，一般用于日志流等

写入整个文件

package main

import (
   "io/ioutil"
)

func main() {

   content := []byte("guoke\nzngw2\n")
   err := ioutil.WriteFile("test.txt", content, 0644)
   // 注意，如果写入的目录不存在会报错
   if err != nil {
      panic(err)
   }
}

在文件尾部添加内容

package main

import (
"fmt"
"io"
"os"
)

func checkFileIsExist(filename string) bool {
    if _, err := os.Stat(filename); os.IsNotExist(err) {
        return false
    }
    return true
}
func main() {
    var wireteString = "guoke\nzngw2\n"
    var filename = "./test.txt"
    var f *os.File
    var err1 error
    if checkFileIsExist(filename) { //如果文件存在
        f, err1 = os.OpenFile(filename, os.O_APPEND, 0666) //打开文件
        fmt.Println("文件存在")
    } else {
        f, err1 = os.Create(filename) //创建文件
        fmt.Println("文件不存在")
    }
    defer f.Close()
    n1, err1 := io.WriteString(f, wireteString) //写入文件(字符串)
    if err1 != nil {
        panic(err1)
    }
    fmt.Println("写入 %d 个字节", n1)

    n2, err2 := f.Write([]byte(wireteString)) //写入文件(字节)
    if err2 != nil {
        panic(err1)
    }
    fmt.Println("写入 %d 个字节", n2)
}