byte[] 通道使用

byte[] channel usage

我收到一个 REST 命令,想在它的主体上计算散列函数。 为此,我使用 io.TeeReader(request.Body, &writerToHash) 读取正文,其中我传递了我自己的 class 实现 io.Writer:

func (self *WriterToHash) Write(p []byte) (n int, err error) {
    n=len(p)
    fmt.println("WriterToHash  len=%v, buff=%v", n, p)  //PRINT 1
    self.BodyChannel <- p
    return n, nil
}

BodyChannel定义为:BodyChannel chan []byte

我使用这个class如下:

    writerToHash := sisutils.WriterToHash{
        BodyChannel:make(chan []byte, 1024)
    }
    writerToHash.StartListen()
    reqnew, _ := http.NewRequest("PUT", url, io.TeeReader(request.Body, &writerToHash))

听力部分:

func (wth *WriterToHash) StartListen() {
    wth.OutChannel = make(chan []byte, 1000)
    go func (self *WriterToHash) {
        done := int64(0)
        h := sha1.New()
        for done < MessageSize{
            buff := <- self.BodyChannel

            done += int64(len(buff))
            DPrint(5, "AccamulateSha1 Done=: %v, buff=%v", done, buff)  //PRINT 2

            actually_write, err := h.Write(buff)
            if err != nil ||  actually_write != len(buff) {
                log.Println("Error in sha write:" + err.Error())
                break
            }
        }
        bs := h.Sum(nil)
        self.OutChannel <- bs
    }(wth)
 }

我发送了 1000 字节的消息。在调试模式下,消息总是以相同的方式拆分:1 个字节,999 个字节——我使用 PRINT 1 看到它。在这种情况下,everythong 工作正常。 问题在于,当消息在 Write 函数中被拆分为更多部分时。在这种情况下,我在 PRINT1 中看到:

[第一个字节] : a

[接下来 ~450 字节]:b、c、d、...

[最后 ~550 字节]:w,x,y,...

但在 PRINT 2 中我看到了不同的图片:

[第一个字节] : a

[~450 字节,但从最后一部分开始]:w,x,y...

[最后 ~550 字节]:w,x,y,...

我实际上得到了两次过去的最后一次,但大小不同。

来自io.Writer documentation

Write must not modify the slice data, even temporarily. Implementations must not retain p

您不能存储或重复使用传递给您的 Write 方法的切片。如果您想在其他地方使用该数据,则需要复制它

func (self *WriterToHash) Write(p []byte) (n int, err error) {
    b := make([]byte, len(p))
    copy(b, p)
    fmt.println("WriterToHash  len=%d, buff=%v", len(p), b)
    self.BodyChannel <- b
    return len(p), nil
}