我正在构建一个 blob 存储系统,我选择 Go 作为编程语言。 我创建了一个流来执行从客户端到 Blob 服务器的多部分文件上传。
流工作正常,但我想从请求正文中创建一个 sha1 哈希。我需要 io.Copy body 两次。 sha1 被创建,但之后多部分流 0 个字节。
- 用于创建哈希
- 用于将正文作为多部分流式传输
知道我该怎么做吗?
客户端上传
func (c *Client) Upload(h *UploadHandle) (*PutResult, error) {
body, bodySize, err := h.Read()
if err != nil {
return nil, err
}
// Creating a sha1 hash from the bytes of body
dropRef, err := drop.Sha1FromReader(body)
if err != nil {
return nil, err
}
bodyReader, bodyWriter := io.Pipe()
writer := multipart.NewWriter(bodyWriter)
errChan := make(chan error, 1)
go func() {
defer bodyWriter.Close()
part, err := writer.CreateFormFile(dropRef, dropRef)
if err != nil {
errChan <- err
return
}
if _, err := io.Copy(part, body); err != nil {
errChan <- err
return
}
if err = writer.Close(); err != nil {
errChan <- err
}
}()
req, err := http.NewRequest("POST", c.Server+"/drops/upload", bodyReader)
req.Header.Add("Content-Type", writer.FormDataContentType())
resp, err := c.Do(req)
if err != nil {
return nil, err
}
.....
}
sha1 函数
func Sha1FromReader(src io.Reader) (string, error) {
hash := sha1.New()
_, err := io.Copy(hash, src)
if err != nil {
return "", err
}
return hex.EncodeToString(hash.Sum(nil)), nil
}
上传句柄
func (h *UploadHandle) Read() (io.Reader, int64, error) {
var b bytes.Buffer
hw := &Hasher{&b, sha1.New()}
n, err := io.Copy(hw, h.Contents)
if err != nil {
return nil, 0, err
}
return &b, n, nil
}
最佳答案
我建议使用 io.TeeReader
如果您想同时通过 sha1 推送来自 blob 的所有读取。
bodyReader := io.TeeReader(body, hash)
现在由于在上传过程中消耗了 bodyReader,哈希值会自动更新。
关于io - Golang io.copy 在请求正文上两次,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25671305/