1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-28 08:47:42 +08:00

feat(bitswap:msg) impl FromSwarm method

This commit is contained in:
Brian Tiger Chow
2014-09-12 19:08:52 -07:00
parent 70999886c5
commit 21639564e7

View File

@ -1,6 +1,8 @@
package bitswap
import (
proto "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/goprotobuf/proto"
blocks "github.com/jbenet/go-ipfs/blocks"
nm "github.com/jbenet/go-ipfs/net/message"
swarm "github.com/jbenet/go-ipfs/net/swarm"
@ -41,6 +43,15 @@ func (m *message) AppendBlock(b *blocks.Block) {
m.pb.Blocks = append(m.pb.Blocks, b.Data)
}
func FromSwarm(sms swarm.Message) (BitSwapMessage, error) {
var protoMsg PBMessage
err := proto.Unmarshal(sms.Data, &protoMsg)
if err != nil {
return nil, err
}
return newMessageFromProto(protoMsg), nil
}
func (m *message) ToProto() *PBMessage {
cp := m.pb
return &cp