Go语言碎片整理之文件操作

淡淡的烟草味﹌ 2022-01-12 22:25 238阅读 0赞

文件是什么？  
计算机中的文件是存储在外部介质（通常是磁盘）上的数据集合，文件分为文本文件和二进制文件。

## 打开和关闭文件 ##

os.Open()函数能够打开一个文件，返回一个\*File和一个error。对得到的文件实例调用close()方法能够关闭文件。

func main(){
        file,err := os.Open("./main.go")
        if err != nil{
            fmt.Println("open file failed!,err",err)
            return 
        }
        fmt.Println("cool!")
        file.Close()
    }

为了防止文件忘记关闭，我们通常使用defer注册文件关闭语句。

### 读取文件 ###

file.Read()

基本使用

Read的方法定义如下：  
`func(f *File) Read(b []byte)(n int, err error)`  
它接收一个字节切片，返回读取的字节数和可能的具体错误，读到文件末尾时会返回0和io.EOF。

package main
    import (
        "fmt"
        "os"
        "io"
    )
    func main(){
        file,err := os.Open("./main.go")
        if err != nil{
            fmt.Println("open file failed!,err", err)
            return 
        }
        defer file.Close()
        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]))
    }

### 循环读取 ###

使用for循环读取文件中的所有数据。

package main
    import (
        "fmt"
        "os"
        "io"
    )
    func main(){
        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的基础上封装了一层API,支持更多的功能。

package main
    import (
        "fmt"
        "os"
        "io"
        "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{
                fmt.Println("文件读完了")
                break
            }
            if err != nil{
                fmt.Println("read file failed!,err",err)
                return
            }
            fmt.Println(line)
        }
    }

#### 使用ioutil读取整个文件 ####

io/ioutil包的ReadFile方法能够读取完整的文件，只需要将文件名作为参数传入。

package main
    import (
        "fmt"
        "io/ioutil"
    )
    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是指打开文件的模式，模式有以下几种：

![Go语言 碎片整理之 文件操作][Go_ _]  
perm 文件权限，一个八进制数。r(读)04 ，w（写）02，x(执行)01。

#### Write和WriteString ####

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()
        str := "hello 沙河"
        file.Write([]byte(str))       //写入字节切片数据
        file.WriteString("hello 小王子") //直接写入字符串数据
    }

#### bufio.NewWriter ####

import (
        "io"
            "fmt"
    )
    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沙河\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
        }
    }

### 练习 ###

#### 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_CREATE|os.O_WRONLY,0644)
        if err != nil{
            fmt.Printf("open %s failed,err:%v.\n",dstName,err)
            return 
        }
        defer dst.Close()
        return io.Copy(dst,src)
    }
    func main(){
        _,err := CopyFile("dst.txt","src.txt")
        if err != nil{
            fmt.Println("copy file failed,err",err)
            return 
        }
        fmt.Println("Copy done!")
    }

转载于:https://blog.51cto.com/13766835/2404547

[Go_ _]: https://s1.51cto.com/images/blog/201906/02/af2e0990739c34bcf457c4e3de8c9597.png?x-oss-process=image/watermark,size_16,text_QDUxQ1RP5Y2a5a6i,color_FFFFFF,t_100,g_se,x_10,y_10,shadow_90,type_ZmFuZ3poZW5naGVpdGk=