GO – 文件相关操作

  • A+
所属分类:go语言编程 学编程

本文主要介绍了Go语言中文件读写的相关操作。

打开与关闭文件

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

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

读取文件内容

func main() {
    // 只读方式打开当前目录下的main.go文件(返回一个file和错误信息)
    file, err := os.Open("./main.go")
    // 错误处理
    if err != nil {
        fmt.Println("open file failed!, err:", err)
        return
    }
    // 用derfer 定义关闭文件
    defer file.Close()
    // 使用Read方法读取数据
    var tmp = make([]byte, 128)
    // 将文件读取到tmp这个数组里(返回值为:文件内容的总数和错误)
    n, err := file.Read(tmp)
    // 文件结尾有个EOF错误用来判断文件是否读完了
    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]))
}

循环读取

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的基础上封装了一层API,支持更多的功能。

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来关闭文件
    defer file.Close()
    // 用bufio.NewReader(file)获取一个文件读取对象
    reader := bufio.NewReader(file)
    // 死循环
    for {
        // 调用文件读取对象的ReadString方法指定读取到'\n'结束(获取到文件里一行数据和错误信息)
        line, err := reader.ReadString('\n') //注意是字符
        // 判断是不是文件结束了
        if err == io.EOF {
            fmt.Println("文件读完了")
            break
        }
        // 错误判断
        if err != nil {
            fmt.Println("read file failed, err:", err)
            return
        }
        // 每循环一次就打印一行读取的数据
        fmt.Print(line)
    }
}

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

package main

import (
    "fmt"
    "io/ioutil"
)

// ioutil.ReadFile读取整个文件
func main() {
    // 利用ioutil.ReadFile读取整个文件,返回值一个为内容,另一个为错误提示
    content, err := ioutil.ReadFile("./main.go")
    if err != nil {
        fmt.Println("read file failed, err:", err)
        return
    }
    fmt.Println(string(content))
}

从字符串里读取内容打印到终端

// 定义结构体
type Student struct {
    Name string
    Age int
    Score float32
}
func strInput() {
    var str = "stu01 18 92.2"      // 定义字符串
    var stu Student      // 定义str变量(变量类型为Student)
    fmt.Sscanf(str, "%s %d %f", &stu.Name, &stu.Age, &stu.Score)     // 从str字符里读取内容赋值给占位符(通过占位符赋值给stu的属性)
    fmt.Println(stu)      // 结果为:{stu01 18 92.2}
}

读取用户输入

func readDemo() {
    var str string      // 定义字符串类型变量
    fmt.Scanf("%s",&str)        // 获取用户输入并将用户输入的内容通过占位符+内存地址的方式赋值给str变量
    fmt.Println("output:",str)
}

文件写操作

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

参数如下:

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

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

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 Dream"
    err := ioutil.WriteFile("./xx.txt", []byte(str), 0666)
    if err != nil {
        fmt.Println("write file failed, err:", err)
        return
    }
}

将字符串写入终端

func stdoutDemo() {
    fmt.Fprintln(os.Stdout,"Read Weite test...\n")     // 往终端输入内容
}

拷贝文件

// 拷贝文件函数(从srcName拷贝到dstName)
func copyFile(dstName, srcName string) (written int64,err error) {
    // 以只读的方式打开srcName
    src, err := os.Open(srcName)
    if err != nil {
        return
    }
    // 通过defer定义关闭文件的命令
    defer src.Close()
    // 以写的方式打开dstName,os.O_CREATE:如果没有这个文件就创建这个文件,文件权限644
    dst, err := os.OpenFile(dstName,os.O_WRONLY|os.O_CREATE,0644)
    if err != nil {
        return
    }
    defer dst.Close()       // 通过defer定义关闭文件的命令
    return io.Copy(dst, src)      // 通过io.Copy实现拷贝
}

func main() {
    copyFile("dstfile.txt","srcfile.txt")          // 从srcfile拷贝到dstfile
}

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: