Go语言文件操作 | 青训营笔记

94 阅读2分钟

这是我参与「第五届青训营 」伴学笔记创作活动的第9天

打开和关闭文件

os.Open()函数能够打开一个文件,返回一个*File和一个err。

对得到的文件实例调用close()方法能够关闭文件。

func main() {
        // 只读方式打开当前目录下的main.go文件
        file, err := os.Open("./main.go")
        if err != nil {
                fmt.Println("open file failed!, err:", err)
                return
        }
        // 关闭文件,通常使用defer注册文件关闭语句
        file.Close()}

读取文件

func (f *File) Read(b []byte) (n int, err error)

接收一个字节切片,返回读取的字节数和可能的具体错误, 读到文件末尾时会返回0和io.EOF

func main() {
    // 只读方式打开目录下的main.go文件
    file, err := os.Open("./github.com/xxx/test/main.go")
    if err != nil {
        fmt.Println("open file failed!, err:", err)
        return
    }
    defer file.Close()
    // 使用Read方法读取数据
    var tmp = make([]byte, 256)
    n, err := file.Read(tmp)
    if err == io.EOF {
        fmt.Println("文件读取完毕")
        return
    }
    if err != nil {
        fmt.Println("read file failed, err:", err)
        return
    }
    fmt.Printf("读取了%d字节数据\n", n)
    fmt.Println(string(tmp[:n]))
}

循环读取

// 循环读取文件
        var content []byte
        var tmp = make([]byte, 256)
        for {
                n, err := file.Read(tmp)
                if err == io.EOF {
                        fmt.Println("文件读取完毕")
                        break
                }
                if err != nil {
                        fmt.Println("read file failed, err:", err)
                        return
                }
                content = append(content, tmp[:n]...)
        }
        fmt.Println(string(content))

bufio读取文件

在file的基础上封装了一层API,支持更多的功能

// bufio按行读取示例
func main() {
    file, err := os.Open("C://Z/hello.txt")
    if err != nil {
        fmt.Println("open file failed, err:", err)
        return
    }
    defer file.Close()
    reader := bufio.NewReader(file)
    for {
        line, err := reader.ReadString('\n') //换行读取
        if err == io.EOF {
            if len(line) != 0 {
                fmt.Println(line)
            }
            fmt.Println("文件读取完毕")
            break
        }

        if err != nil {
            fmt.Println("read file failed, err:", err)
            return
        }
        //line=strings.TrimSuffix(line,"\r\n") //一行输出
        fmt.Print(line)
    }
}

ioutil读取整个文件

import (
        "fmt"
        "io/ioutil")
// ioutil.ReadFile读取整个文件
func main() {
        content, err := ioutil.ReadFile("./main.go")
        if err != nil {
                fmt.Println("read file failed, err:", err)
                return
        }
        fmt.Println(string(content))}

文件写入

os.OpenFile()函数能够以指定模式打开文件,从而实现文件写入相关功能。

func OpenFile(name string, flag int, perm FileMode) (*File, error) { }

name:要打开的文件名 

flag:打开文件的模式

os.O_WRONLY    只写
os.O_CREATE    创建文件
os.O_RDONLY    只读
os.O_RDWR      读写
os.O_TRUNC     清空
os.O_APPEND    追加

perm:文件权限,一个八进制数。r 04,w 02,x 01。

func main() {
    file, err := os.OpenFile("C://Z/hello.txt", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
    if err != nil {
        fmt.Println("open file failed, err:", err)
        return
    }

    defer file.Close()
    str := "hello 123\nThis is hello.txt\n"
    file.Write([]byte(str))       //写入字节切片数据
    file.WriteString("hello END!/\\") //直接写入字符串数据
}
bufio.NewWriter
   defer file.Close() 
   writer := bufio.NewWriter(file)
   for i := 0; i < 10; i++ {
       writer.WriteString("hello\n") //将数据先写入缓存
   }
   writer.Flush() //将缓存中的内容写入文件

ioutil.WriteFile

func main() {
        str := "hello"
        err := ioutil.WriteFile("./xx.txt", []byte(str), 0666)
        if err != nil {
                fmt.Println("write file failed, err:", err)
                return
        }}