From 24224979bb6f3facdc547b6f018707bd9e533261 Mon Sep 17 00:00:00 2001 From: Brian Tiger Chow Date: Fri, 12 Sep 2014 19:08:17 -0700 Subject: [PATCH] feat(bitswap:msg) add ToNet() method --- bitswap/message.go | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/bitswap/message.go b/bitswap/message.go index a0be726b7..d1a1db669 100644 --- a/bitswap/message.go +++ b/bitswap/message.go @@ -2,6 +2,7 @@ package bitswap import ( blocks "github.com/jbenet/go-ipfs/blocks" + nm "github.com/jbenet/go-ipfs/net/message" swarm "github.com/jbenet/go-ipfs/net/swarm" peer "github.com/jbenet/go-ipfs/peer" u "github.com/jbenet/go-ipfs/util" @@ -36,3 +37,7 @@ func (m *message) ToProto() *PBMessage { func (m *message) ToSwarm(p *peer.Peer) *swarm.Message { return swarm.NewMessage(p, m.ToProto()) } + +func (m *message) ToNet(p *peer.Peer) (nm.NetMessage, error) { + return nm.FromObject(p, m.ToProto()) +}