Go:文件操作

时间:Feb. 2, 2020 分类:

目录:

打开和关闭文件

  • os.Open()函数能够打开一个文件,返回一个*File和一个err
  • 对得到的文件实例调用close()方法能够关闭文件,关闭文件最好写在defer中
package main

import "fmt"
import "os"

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

package main

import "fmt"
import "os"
import "io"


func main() {
        // 只读方式打开当前目录下的main.go文件
        file, err := os.Open("./main.go")
        if err != nil {
                fmt.Println("open file failed!, err:", err)
                return
        }
        defer file.Close()
        // 使用Read方法读取数据
        var tmp = make([]byte, 128)
        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]))
}

打印结果

读取了128字节数据
package main

import "fmt"
import "os"
import "io"


func main() {
    // 只读方式打开当前目录下的main.go文件
    file, 

循环读取

package main

import "fmt"
import "os"
import "io"

func main() {
    // 只读方式打开当前目录下的main.go文件
    file, err := os.Open("./main.go")
    if err != nil {
        fmt.Println("open file failed!, err:", err)
        return
    }
    defer file.Close()
    // 循环读取文件
    var content []byte
    var tmp = make([]byte, 128)
    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读取文件

bufio是在file的基础上做了封装

package main

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

// bufio按行读取示例
func main() {
    file, err := os.Open("./main.go")
    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
        }
        fmt.Print(line)
    }
}

ioutil读取整个文件

package main

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) {
    ...
}

文件打开模式flag

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

文件权限perm

Write写入和WriteString

package main

import (
    "fmt"
    "os"
)

func main() {
    file, err := os.OpenFile("demo.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"
    file.Write([]byte(str))       //写入字节切片数据
    file.WriteString("world") //直接写入字符串数据
}

查看文件

$ cat demo.txt
helloworld

bufio.NewWriter

package main

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

func main() {
    file, err := os.OpenFile("xx.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()
    writer := bufio.NewWriter(file)
    for i := 0; i < 10; i++ {
        writer.WriteString("hello world\n") //将数据先写入缓存
    }
    writer.Flush() //将缓存中的内容写入文件
}

ioutil.WriteFile

package main

import (
    "fmt"
    "io/ioutil"
)

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

实现文件拷贝

package main

import (
    "fmt"
    "io/ioutil"
)

// CopyFile 拷贝文件函数
func CopyFile(dstName, srcName string) (written int64, err error) {
    // 以读方式打开源文件
    src, err := os.Open(srcName)
    if err != nil {
        fmt.Printf("open %s failed, err:%v.\n", srcName, err)
        return
    }
    defer src.Close()
    // 以写|创建的方式打开目标文件
    dst, err := os.OpenFile(dstName, os.O_WRONLY|os.O_CREATE, 0644)
    if err != nil {
        fmt.Printf("open %s failed, err:%v.\n", dstName, err)
        return
    }
    defer dst.Close()
    return io.Copy(dst, src) //调用io.Copy()拷贝内容
}
func main() {
    _, err := CopyFile("dst.txt", "src.txt")
    if err != nil {
        fmt.Println("copy file failed, err:", err)
        return
    }
    fmt.Println("copy done!")
}