使用 io.Copy 函数进行模拟调用的方法

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《使用 io.Copy 函数进行模拟调用的方法》，主要介绍了，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

尝试模拟以下函数。它基本上从 s3 (io.readcloser) 获取一个对象，并将其写入之前执行 os.open()d (io.writecloser) 的文件。

package main

import (
    "io"

    log "github.com/sirupsen/logrus"
)

func writefile(destination io.writecloser, source io.readcloser) error {
    defer destination.close()
    defer source.close()
    _, err := io.copy(destination, source)
    if err != nil {
        log.withfields(log.fields{"desc": "unable to copy contents from s3 to blahblah"}).error(err)
        return err
    }
    return nil
}

我认为我已经非常接近了，但目前我的测试挂起并且从未错误/成功...我还意识到我可以将 os.stdout 传递到我的目的地，但仍然遇到同样的问题。 io.copy 内发生了一些事情。我想这是因为我试图将空数据复制到什么都没有？

package main

import (
    "errors"
    "io"
    "reflect"
    "testing"
)

type mockReadCloser struct {}

func (m mockReadCloser) Read(p []byte) (int, error) { return 0, nil }
func (m mockReadCloser) Close() error               { return nil }

type mockWriteCloser struct{}

func (m mockWriteCloser) Close() error                      { return nil }
func (m mockWriteCloser) Write(b []byte) (n int, err error) { return 0, nil }

func Test_writeFile(t *testing.T) {
    type args struct {
        destination io.WriteCloser
        source      io.ReadCloser
    }
    tests := []struct {
        name    string
        args    args
        wantErr bool
    }{
        {
            name: "",
            args: args{
                destination: &mockWriteCloser{},
                source:      &mockReadCloser{},
            },
            wantErr: false,
        },
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            if err := writeFile(tt.args.destination, tt.args.source); (err != nil) != tt.wantErr {
                t.Errorf("writeFile() error = %v, wantErr %v", err, tt.wantErr)
            }
        })
    }
}

正确答案

这是因为 io.Copy 直到 eof 或错误才返回。

将副本从 src 复制到 dst，直到 src 达到 eof 或发生错误。它返回复制的字节数以及复制时遇到的第一个错误（如果有）。

成功的复制返回 err == nil，而不是 err == eof。因为 copy 被定义为从 src 读取直到 eof，所以它不会将 read 中的 eof 视为要报告的错误。

因此，如果您从 mockreadcloser.read 返回 eof，它不应再挂起。

func (m mockreadcloser) read(p []byte) (int, error) { return 0, io.eof }

这是因为 read 被重复调用，直到没有更多内容可供读取 (eof)。

for {
    nr, er := src.Read(buf)
    ...
}

本篇关于《使用 io.Copy 函数进行模拟调用的方法》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！