feat(bitswap:msg) define interfaces

70999886 · Brian Tiger Chow · 24224979 · 70999886
Commit 70999886 authored 10 years ago by Brian Tiger Chow
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 0 deletions

bitswap/message.go bitswap/message.go +12 -0

No files found.
--- a/bitswap/message.go
+++ b/bitswap/message.go
@@ -8,6 +8,18 @@ import (
 	u "github.com/jbenet/go-ipfs/util"
 )

+type BitSwapMessage interface {
+	AppendWanted(k u.Key)
+	AppendBlock(b *blocks.Block)
+	Exportable
+}
+
+type Exportable interface {
+	ToProto() *PBMessage
+	ToSwarm(p *peer.Peer) *swarm.Message
+	ToNet(p *peer.Peer) (nm.NetMessage, error)
+}
+
 // message wraps a proto message for convenience
 type message struct {
 	pb PBMessage