【Golang】Go基础之ioutil包 | 十一

401 阅读3分钟

「这是我参与2022首次更文挑战的第13天,活动详情查看:2022首次更文挑战

除了io包可以读写数据,Go语言中还提供了一个辅助的工具包就是ioutil,里面的方法虽然不多,但是都还蛮好用的。

该包的介绍只有一句话:Package ioutil implements some I/O utility functions。

一、ioutil包的方法

下面我们来看一下里面的方法:

// Discard 是一个 io.Writer 接口,调用它的 Write 方法将不做任何事情
// 并且始终成功返回。
var Discard io.Writer = devNull(0)

// ReadAll 读取 r 中的所有数据,返回读取的数据和遇到的错误。
// 如果读取成功,则 err 返回 nil,而不是 EOF,因为 ReadAll 定义为读取
// 所有数据,所以不会把 EOF 当做错误处理。
func ReadAll(r io.Reader) ([]byte, error)

// ReadFile 读取文件中的所有数据,返回读取的数据和遇到的错误。
// 如果读取成功,则 err 返回 nil,而不是 EOF
func ReadFile(filename string) ([]byte, error)

// WriteFile 向文件中写入数据,写入前会清空文件。
// 如果文件不存在,则会以指定的权限创建该文件。
// 返回遇到的错误。
func WriteFile(filename string, data []byte, perm os.FileMode) error

// ReadDir 读取指定目录中的所有目录和文件(不包括子目录)。
// 返回读取到的文件信息列表和遇到的错误,列表是经过排序的。
func ReadDir(dirname string) ([]os.FileInfo, error)

// NopCloser 将 r 包装为一个 ReadCloser 类型,但 Close 方法不做任何事情。
func NopCloser(r io.Reader) io.ReadCloser

// TempFile 在 dir 目录中创建一个以 prefix 为前缀的临时文件,并将其以读
// 写模式打开。返回创建的文件对象和遇到的错误。
// 如果 dir 为空,则在默认的临时目录中创建文件(参见 os.TempDir),多次
// 调用会创建不同的临时文件,调用者可以通过 f.Name() 获取文件的完整路径。
// 调用本函数所创建的临时文件,应该由调用者自己删除。
func TempFile(dir, prefix string) (f *os.File, err error)

// TempDir 功能同 TempFile,只不过创建的是目录,返回目录的完整路径。
func TempDir(dir, prefix string) (name string, err error)

二、示例代码:

package main

import (
    "io/ioutil"
    "fmt"
    "os"
)

func main() {
    /*
    ioutil包:
        ReadFile()
        WriteFile()
        ReadDir()
        ..
     */

    //1.读取文件中的所有的数据
    //fileName1 := "/Users/ruby/Documents/pro/a/aa.txt"
    //data, err := ioutil.ReadFile(fileName1)
    //fmt.Println(err)
    //fmt.Println(string(data))

    //2.写出数据
    //fileName2:="/Users/ruby/Documents/pro/a/bbb.txt"
    //s1:="helloworld面朝大海春暖花开"
    //err:=ioutil.WriteFile(fileName2,[]byte(s1),0777)
    //fmt.Println(err)

    //3.
    //s2:="qwertyuiopsdfghjklzxcvbnm"
    //r1:=strings.NewReader(s2)
    //data,_:=ioutil.ReadAll(r1)
    //fmt.Println(data)

    //4.ReadDir(),读取一个目录下的子内容:子文件和子目录,但是仅有一层
    //dirName:="/Users/ruby/Documents/pro/a"
    //fileInfos,_:=ioutil.ReadDir(dirName)
    //fmt.Println(len(fileInfos))
    //for i:=0;i
 

三、遍历文件夹

学习io之后,尤其是文件操作,我们就可以遍历给定的目录文件夹了。可以使用ioutil包下的readDir()方法,这个方法可以获取指定目录下的内容,返回文件和子目录。

因为文件夹下还有子文件夹,而ioutil包的ReadDir()只能获取一层目录,所以我们需要自己去设计算法来实现,最容易实现的思路就是使用递归。

示例代码:

package main

import (
    "io/ioutil"
    "fmt"
    "log"
)

func main() {
    /**
    遍历文件夹:
     */

    dirname := "/Users/ruby/Documents/pro"
    listFiles(dirname, 0)

}

func listFiles(dirname string, level int) {
    // level用来记录当前递归的层次
    // 生成有层次感的空格
    s := "|--"
    for i := 0; i < level; i++ {
        s = "|   " + s
    }

    fileInfos, err := ioutil.ReadDir(dirname)
    if err != nil{
        log.Fatal(err)
    }
    for _, fi := range fileInfos {
        filename := dirname + "/" + fi.Name()
        fmt.Printf("%s%s\n", s, filename)
        if fi.IsDir() {
            //继续遍历fi这个目录
            listFiles(filename, level+1)
        }
    }
}

运行结果:

image.png