mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-10 09:52:20 +08:00
40 lines
692 B
Go
40 lines
692 B
Go
package chunk
|
|
|
|
import (
|
|
"hash/fnv"
|
|
"io"
|
|
|
|
"github.com/ipfs/go-ipfs/Godeps/_workspace/src/github.com/whyrusleeping/chunker"
|
|
)
|
|
|
|
var IpfsRabinPoly = chunker.Pol(17437180132763653)
|
|
|
|
type Rabin struct {
|
|
r *chunker.Chunker
|
|
}
|
|
|
|
func NewRabin(r io.Reader, avgBlkSize uint64) *Rabin {
|
|
min := avgBlkSize / 3
|
|
max := avgBlkSize + (avgBlkSize / 2)
|
|
|
|
return NewRabinMinMax(r, min, avgBlkSize, max)
|
|
}
|
|
|
|
func NewRabinMinMax(r io.Reader, min, avg, max uint64) *Rabin {
|
|
h := fnv.New32a()
|
|
ch := chunker.New(r, IpfsRabinPoly, h, avg, min, max)
|
|
|
|
return &Rabin{
|
|
r: ch,
|
|
}
|
|
}
|
|
|
|
func (r *Rabin) NextBytes() ([]byte, error) {
|
|
ch, err := r.r.Next()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ch.Data, nil
|
|
}
|