coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-02-28 05:47:51 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

bitswap: add wantlist fullness to protobuf messages

Browse Source 
License: MIT
Signed-off-by: Jeromy <why@ipfs.io>
This commit is contained in:
Jeromy 2016-12-02 14:15:24 -08:00
parent a542dea5de
commit 6f8dd03e07
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
exchange/bitswap/message/message.go
View File

@ -189,6 +189,7 @@ func (m *impl) ToProtoV0() *pb.Message {
Cancel: proto.Bool(e.Cancel),
})
}
pbm.Wantlist.Full = proto.Bool(m.full)
for _, b := range m.Blocks() {
pbm.Blocks = append(pbm.Blocks, b.RawData())
}
@ -205,6 +206,7 @@ func (m *impl) ToProtoV1() *pb.Message {
Cancel: proto.Bool(e.Cancel),
})
}
pbm.Wantlist.Full = proto.Bool(m.full)
for _, b := range m.Blocks() {
blk := &pb.Message_Block{
Data: b.RawData(),

4
exchange/bitswap/message/message_test.go
View File

@ -118,6 +118,10 @@ func TestToNetFromNetPreservesWantList(t *testing.T) {
t.Fatal(err)
}
if !copied.Full() {
t.Fatal("fullness attribute got dropped on marshal")
}
keys := make(map[string]bool)
for _, k := range copied.Wantlist() {
keys[k.Cid.KeyString()] = true