那有更好的替换方案吗?当然是有的,我们接着往下看。
io.Copy
可以使用 io.Copy 函数来代替,源码定义如下:
src/io/io.go
func Copy(dst Writer, src Reader) (written int64, err error) {
return copyBuffer(dst, src, nil)
}
其功能是直接从 src 读取数据,并写入到 dst。
和 ioutil.ReadAll 最大的不同就是没有把所有数据一次性都取出来,而是不断读取,不断写入。
具体实现 Copy 的逻辑在 copyBuffer 函数中实现:
// src/io/io.go
func copyBuffer(dst Writer, src Reader, buf []byte) (written int64, err error) {
// 如果源实现了 WriteTo 方法,则直接调用 WriteTo
if wt, ok := src.(WriterTo); ok {
return wt.WriteTo(dst)
}
// 同样的,如果目标实现了 ReaderFrom 方法,则直接调用 ReaderFrom
if rt, ok := dst.(ReaderFrom); ok {
return rt.ReadFrom(src)
}
// 如果 buf 为空,则创建 32KB 的 buf
if buf == nil {
size := 32 * 1024
if l, ok := src.(*LimitedReader); ok && int64(size) > l.N {
if l.N < 1 {
size = 1
} else {
size = int(l.N)
}
}
buf = make([]byte, size)
}
// 循环读取数据并写入
for {
nr, er := src.Read(buf)
if nr > 0 {
nw, ew := dst.Write(buf[0:nr])
if nw < 0 || nr < nw {
nw = 0
if ew == nil {
ew = errInvalidWrite
}
}
written += int64(nw)
if ew != nil {
err = ew
break
}
if nr != nw {
err = ErrShortWrite
break
}
}
if er != nil {
if er != EOF {
err = er
}
break
}
}
return written, err
}
此函数执行步骤如下:
-
如果源实现了
WriteTo方法,则直接调用WriteTo方法; -
同样的,如果目标实现了 ReaderFrom 方法,则直接调用 ReaderFrom 方法;
-
如果
buf为空,则创建 32KB 的buf; -
最后就是循环
Read和Write;
对比之后就会发现,io.Copy 函数不会一次性读取全部数据,也不会频繁进行切片扩容,显然在数据量大时是更好的选择。
ioutil 其他函数
再看看 ioutil 包的其他函数:
-
func ReadDir(dirname string) ([]os.FileInfo, error) -
func ReadFile(filename string) ([]byte, error) -
func WriteFile(filename string, data []byte, perm os.FileMode) error -
func TempFile(dir, prefix string) (f *os.File, err error) -
func TempDir(dir, prefix string) (name string, err error) -
func NopCloser(r io.Reader) io.ReadCloser
下面举例详细说明:
ReadDir
// ReadDir 读取指定目录中的所有目录和文件(不包括子目录)。
// 返回读取到的文件信息列表和遇到的错误,列表是经过排序的。
func ReadDir(dirname string) ([]os.FileInfo, error)
举例:
package main
import (
"fmt"
"io/ioutil"
)
func main() {
dirName := "../"
fileInfos, _ := ioutil.ReadDir(dirName)
fmt.Println(len(fileInfos))
for i := 0; i < len(fileInfos); i++ {
fmt.Printf("%T\n", fileInfos[i])
fmt.Println(i, fileInfos[i].Name(), fileInfos[i].IsDir())
}
}
ReadFile
// ReadFile 读取文件中的所有数据,返回读取的数据和遇到的错误
// 如果读取成功,则 err 返回 nil,而不是 EOF
func ReadFile(filename string) ([]byte, error)
举例:
package main
import (
"fmt"
"io/ioutil"
"os"
)
func main() {
data, err := ioutil.ReadFile("./test.txt")
if err != nil {
fmt.Println("read error")
os.Exit(1)
}
fmt.Println(string(data))
}
WriteFile
// WriteFile 向文件中写入数据,写入前会清空文件。
// 如果文件不存在,则会以指定的权限创建该文件。
// 返回遇到的错误。
func WriteFile(filename string, data []byte, perm os.FileMode) error
举例:
package main
import (
"fmt"
"io/ioutil"
)
func main() {
fileName := "./text.txt"
s := "Hello AlwaysBeta"
err := ioutil.WriteFile(fileName, []byte(s), 0777)
fmt.Println(err)
}
TempFile
// TempFile 在 dir 目录中创建一个以 prefix 为前缀的临时文件,并将其以读
// 写模式打开。返回创建的文件对象和遇到的错误。
// 如果 dir 为空,则在默认的临时目录中创建文件(参见 os.TempDir),多次
// 调用会创建不同的临时文件,调用者可以通过 f.Name() 获取文件的完整路径。
// 调用本函数所创建的临时文件,应该由调用者自己删除。
func TempFile(dir, prefix string) (f *os.File, err error)
举例:
package main
import (
"fmt"
"io/ioutil"
"os"
)
func main() {
f, err := ioutil.TempFile("./", "Test")
if err != nil {
fmt.Println(err)
}
defer os.Remove(f.Name()) // 用完删除
fmt.Printf("%s\n", f.Name())
}
TempDir
// TempDir 功能同 TempFile,只不过创建的是目录,返回目录的完整路径。
func TempDir(dir, prefix string) (name string, err error)
举例:
package main
import (
"fmt"
"io/ioutil"
"os"
)
func main() {
dir, err := ioutil.TempDir("./", "Test")
if err != nil {
fmt.Println(err)
}
defer os.Remove(dir) // 用完删除
fmt.Printf("%s\n", dir)
}
NopCloser
// NopCloser 将 r 包装为一个 ReadCloser 类型,但 Close 方法不做任何事情。
func NopCloser(r io.Reader) io.ReadCloser
这个函数的使用场景是这样的:
有时候我们需要传递一个 io.ReadCloser 的实例,而我们现在有一个 io.Reader 的实例,比如:strings.Reader。
这个时候 NopCloser 就派上用场了。它包装一个 io.Reader,返回一个 io.ReadCloser,相应的 Close 方法啥也不做,只是返回 nil。
举例:
package main
import (
"fmt"
"io/ioutil"
"reflect"
"strings"
)
func main() {
//返回 *strings.Reader
reader := strings.NewReader("Hello AlwaysBeta")
r := ioutil.NopCloser(reader)
defer r.Close()
fmt.Println(reflect.TypeOf(reader))
data, _ := ioutil.ReadAll(reader)
fmt.Println(string(data))
}
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!
由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新