如何在 golang 中从多个作家那里写作？

Question

这个想法是创建一个函数，该函数将写入多个写入器并执行返回，以便：

• 将 []byte 切片写入所有写入者
• 返回每个写入器写入的n个字节，索引位置对应于写入器的索引位置。 -如果切片为空返回 ([]int{}) ...等

错误函数：

type Errors []error


func (m Errors) Error() string {
    count := 0
    for t := 0; t < len(m); t++ {
        if m[t] != nil {
            count += 1

        }
    }
    errindex := 0
    for t := 0; t < len(m); t++ {
        if m[t] != nil {
            errindex = t
            break
        }
    }

    if count == 0 {
        return fmt.Sprintf("(0 errors)")
    } else {
        if count == 1 {
            return fmt.Sprintf("%v", m[errindex])
        } else if count == 2 {
            return fmt.Sprintf("%v (and 1 other error)", m[errindex])
        } else {
            return fmt.Sprintf("%v (and %d other errors)", m[errindex], count-1)

        }

    }
}

我的新代码：

func WriteTo(b []byte, writers ...io.Writer) (n []int, errs errors.Errors) {
    if len(writers) == 0 {
        return []int{}, nil
    }
    for i := 0; i < len(writers); i++ {
        num, err := writers[i].Write(b) // write bytes to a current writer
        n = append(n, num)              // add bytes written by current write
        if err!=nil{
            switch err{
            case io.ErrShortWrite:
                errs = append(errs, err)
            default:
                errs = append(errs, err)
            }
        }
    }
    return n, errs
}

我尝试修复它，但有些测试没有通过。实在看不出来问题。当 count == 1 时不返回，而是在 count == 0 时返回，当 count == 2 时返回 count == 0

    []error{nil, io.ErrShortWrite, nil},

    []error{nil, io.ErrShortWrite, io.ErrShortWrite},

Answer 1

您可以使用io.MultiWriter()，它接受任意数量的io.Writers 并返回单个io.Writer，它将写入复制到所有提供的写入器。

例如：

buf1 := &bytes.Buffer{}
buf2 := &bytes.Buffer{}

w := io.MultiWriter(buf1, buf2)

if _, err := w.Write([]byte("Hello")); err != nil {
    panic(err)
}

fmt.Println("Buf1:", buf1.String())
fmt.Println("Buf2:", buf2.String())

这将输出（在Go Playground 上尝试）：

Buf1: Hello
Buf2: Hello