Commit 581df307 authored by Kevin Atkinson's avatar Kevin Atkinson

Add WithCodec and GetCodec methods to format interface.

parent fcbf84a5
...@@ -6,6 +6,8 @@ import ( ...@@ -6,6 +6,8 @@ import (
type Format interface { type Format interface {
Sum(data []byte) (*Cid, error) Sum(data []byte) (*Cid, error)
GetCodec() uint64
WithCodec(uint64) Format
} }
type FormatV0 struct{} type FormatV0 struct{}
...@@ -34,6 +36,21 @@ func PrefixToFormat(p Prefix) Format { ...@@ -34,6 +36,21 @@ func PrefixToFormat(p Prefix) Format {
} }
} }
func (p Prefix) GetCodec() uint64 {
return p.Codec
}
func (p Prefix) WithCodec(c uint64) Format {
if c == p.Codec {
return p
}
p.Codec = c
if c != DagProtobuf {
p.Version = 1
}
return p
}
func (p FormatV0) Sum(data []byte) (*Cid, error) { func (p FormatV0) Sum(data []byte) (*Cid, error) {
hash, err := mh.Sum(data, mh.SHA2_256, -1) hash, err := mh.Sum(data, mh.SHA2_256, -1)
if err != nil { if err != nil {
...@@ -42,6 +59,17 @@ func (p FormatV0) Sum(data []byte) (*Cid, error) { ...@@ -42,6 +59,17 @@ func (p FormatV0) Sum(data []byte) (*Cid, error) {
return NewCidV0(hash), nil return NewCidV0(hash), nil
} }
func (p FormatV0) GetCodec() uint64 {
return DagProtobuf
}
func (p FormatV0) WithCodec(c uint64) Format {
if c == DagProtobuf {
return p
}
return FormatV1{Codec: c, HashFun: mh.SHA2_256}
}
func (p FormatV1) Sum(data []byte) (*Cid, error) { func (p FormatV1) Sum(data []byte) (*Cid, error) {
mhLen := p.HashLen mhLen := p.HashLen
if mhLen <= 0 { if mhLen <= 0 {
...@@ -53,3 +81,12 @@ func (p FormatV1) Sum(data []byte) (*Cid, error) { ...@@ -53,3 +81,12 @@ func (p FormatV1) Sum(data []byte) (*Cid, error) {
} }
return NewCidV1(p.Codec, hash), nil return NewCidV1(p.Codec, hash), nil
} }
func (p FormatV1) GetCodec() uint64 {
return p.Codec
}
func (p FormatV1) WithCodec(c uint64) Format {
p.Codec = c
return p
}
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