Unverified Commit 8aa9fde8 authored by Kevin Atkinson's avatar Kevin Atkinson Committed by GitHub

Merge pull request #4 from ipfs/kevina/inliner

Add Inliner CID Builder.
parents b6f51d59 4709f60d
package cidutil
import (
cid "github.com/ipfs/go-cid"
mhash "github.com/multiformats/go-multihash"
)
// InlineBuilder is a cid.Builder that will use the id multihash when the
// size of the content is no more than limit
type InlineBuilder struct {
cid.Builder // Parent Builder
Limit int // Limit (inclusive)
}
// WithCodec implements the cid.Builder interface
func (p InlineBuilder) WithCodec(c uint64) cid.Builder {
return InlineBuilder{p.Builder.WithCodec(c), p.Limit}
}
// Sum implements the cid.Builder interface
func (p InlineBuilder) Sum(data []byte) (*cid.Cid, error) {
if len(data) > p.Limit {
return p.Builder.Sum(data)
}
return cid.V1Builder{Codec: p.GetCodec(), MhType: mhash.ID}.Sum(data)
}
package cidutil
import (
"math/rand"
"testing"
cid "github.com/ipfs/go-cid"
mhash "github.com/multiformats/go-multihash"
)
func TestInlineBuilderSmallValue(t *testing.T) {
builder := InlineBuilder{cid.V0Builder{}, 64}
c, err := builder.Sum([]byte("Hello World"))
if err != nil {
t.Fatal(err)
}
if c.Prefix().MhType != mhash.ID {
t.Fatal("Inliner builder failed to use ID Multihash on small values")
}
}
func TestInlinerBuilderLargeValue(t *testing.T) {
builder := InlineBuilder{cid.V0Builder{}, 64}
data := make([]byte, 512)
rand.Read(data)
c, err := builder.Sum(data)
if err != nil {
t.Fatal(err)
}
if c.Prefix().MhType == mhash.ID {
t.Fatal("Inliner builder used ID Multihash on large values")
}
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment