restic/internal/hashing/reader.go

30 lines
626 B
Go
Raw Normal View History

2017-01-22 11:43:36 +00:00
package hashing
import (
"hash"
"io"
)
// Reader hashes all data read from the underlying reader.
type Reader struct {
r io.Reader
2017-01-22 11:43:36 +00:00
h hash.Hash
}
// NewReader returns a new Reader that uses the hash h. If the underlying
2021-01-01 11:42:33 +00:00
// reader supports WriteTo then the returned reader will do so too.
func NewReader(r io.Reader, h hash.Hash) *Reader {
return &Reader{r: r, h: h}
}
2021-01-01 11:42:33 +00:00
func (h *Reader) Read(p []byte) (int, error) {
n, err := h.r.Read(p)
_, _ = h.h.Write(p[:n]) // Never returns an error.
return n, err
2017-01-22 11:43:36 +00:00
}
// Sum returns the hash of the data read so far.
func (h *Reader) Sum(d []byte) []byte {
2017-01-22 11:43:36 +00:00
return h.h.Sum(d)
}