Golang IO操作

输入输出的底层原理

终端其实是一个文件，相关实例如下

os.Stdin：标准输入的文件实例，类型为*File
os.Stdout：标准输出的文件实例，类型为*File
os.Stderr：标准错误输出的文件实例，类型为*File

以文件的方式操作终端

package main

import (
    "os"
)

func main() {
    var buf [16]byte
    // 终端输入
    os.Stdin.Read(buf[:])
    // 标准输出
    os.Stdout.WriteString(string(buf[:]))
}

文件操作相关 API

根据提供的文件名创建新的文件，返回一个文件对象，默认权限是 0666

func Create(name string) (file *File, err Error)

根据文件描述符创建相应的文件，返回一个文件对象

func NewFile(fd uintptr, name string) *File

只读方式打开一个名称为 name 的文件

func Open(name string) (file *File, err Error)

打开名称为 name 的文件，flag 是打开的方式，只读、读写等，perm 是权限

func OpenFile(name string, flag int, perm uint32) (file *File, err Error)

写入 byte 类型的信息到文件

func (file *File) Write(b []byte) (n int, err Error)

在指定位置开始写入 byte 类型的信息

func (file *File) WriteAt(b []byte, off int64) (n int, err Error)

写入 string 信息到文件

func (file *File) WriteString(s string) (ret int, err Error)

读取数据到 b 中

func (file *File) Read(b []byte) (n int, err Error)

从 off 开始读取数据到 b 中

func (file *File) ReadAt(b []byte, off int64) (n int, err Error)

删除文件名为 name 的文件

func Remove(name string) Error

写文件

package main

import (
    "fmt"
    "os"
)

func main() {
    file, err := os.Create("./test.txt")
    if err != nil {
        fmt.Println("create file error: ", err)
        return
    }

    defer file.Close()

    for i:=0;i<5;i++ {
        file.WriteString("hello
")
        file.Write([]byte("world
"))
    }
}

读文件

文件读取可以使用file.Read()和file.ReadAt()方法，读到文件末尾会返回io.EOF的错误

package main

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

func main() {
    file, err := os.Open("./test.txt")

    if err != nil {
        fmt.Println("open file err: ", err)
        return
    }

    // 读取文件缓冲区
    var buf [128]byte
    // 文件内容字节数组
    var content []byte

    for {
        n, err := file.Read(buf[:])
        if err == io.EOF {
            // 文件读取完毕
            break
        }

        if err != nil {
            fmt.Println("read file err: ", err)
            return
        }

        content = append(content, buf[:n]...)
    }

    fmt.Println(string(content))
}

文件拷贝

流程是将源文件读取出来，再将内容写入到一个新文件中

package main

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

// 文件拷贝

func readFile(filename string) (fileBodyByte []byte, err error) {
    file, err := os.Open(filename)
    if err != nil {
        return
    }
    defer file.Close()

    var buf [128]byte
    for {
        n, err := file.Read(buf[:])
        if err == io.EOF {
            break
        }

        if err != nil {
            err = errors.New("读取文件失败")
        }

        fileBodyByte = append(fileBodyByte, buf[:n]...)
    }
    return

}

func writeFile(filename string, context []byte) (status bool, err error) {
    file, err := os.Create(filename)
    if err != nil {
        return
    }
    defer file.Close()
    _, err = file.Write(context)
    if err != nil {
        return
    }
    return true, err
}

func copyFile(src string, dest string) (status bool, err error) {
    fileBodyByte, err := readFile(src)
    if err != nil {
        return
    }

    status, err = writeFile(dest, fileBodyByte)
    if err != nil {
        return
    }
    return

}

func main() {
    status, err := copyFile("./test.txt", "./hello.txt")
    if err != nil {
        fmt.Println(err)
    }
    if status {
        fmt.Println("文件拷贝成功")
    }
}

bufio

bufio 包实现了带缓冲区的读写，是对文件读写的封装
bufio 缓冲写数据
- os.O_WRONLY：只写
- os.O_CREATE：创建文件
- os.O_RDONLY：只读
- os.O_RDWR：读写
- os.O_TRUNC：清空
- os.O_APPEND：追加
bufio 读写数据

package main

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

func wr() {
    file, err := os.OpenFile("./test.txt", os.O_CREATE|os.O_WRONLY, 0666)
    if err != nil {
        fmt.Println(err)
        return
    }
    defer file.Close()

    bufWriter := bufio.NewWriter(file)
    for i :=0; i<10; i++ {
        bufWriter.Write([]byte("123
"))
        // 刷新缓冲区，强制写出
        bufWriter.Flush()
    }
}

func re() {
    file, err := os.Open("./test.txt")
    if err != nil {
        fmt.Println(err)
        return
    }

    defer file.Close()

    readBufio := bufio.NewReader(file)

    for {
        line, _, err := readBufio.ReadLine()
        if err == io.EOF {
            break
        }

        if err != nil {
            fmt.Println(err)
            return
        }
        fmt.Print(string(line))
    }
}

func main() {
    //wr()
    re()
}

ioutil 工具包

这个工具包提供了非常方便的文件读写方法

package main

import (
    "fmt"
    "io/ioutil"
)

func wr() {
    err := ioutil.WriteFile("./test123", []byte("hello world
"), 0666)
    if err != nil {
        fmt.Println(err)
        return
    }
}

func re() {
    content, err := ioutil.ReadFile("./test123")
    if err != nil {
        fmt.Println(err)
        return
    }

    fmt.Println(string(content))
}

func main() {
    //wr()
    re()
}

喜欢这篇文章？欢迎打赏~~