status-im
/
libp2p-dht
mirror of https://github.com/status-im/libp2p-dht.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

adding protobuf encoding

This commit is contained in:
Csaba Kiraly 2022-02-15 15:25:50 +01:00
parent af168a8a24
commit 31d99ad56e
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
eth/p2p/providers.nim
View File

@ -40,6 +40,10 @@ func getField*(pb: ProtoBuffer, field: int,
nid = readUintBE[256](buffer)
ok(true)
func write*(pb: var ProtoBuffer, field: int, nid: NodeId) =
## Write NodeId value ``nodeid`` to object ``pb`` using ProtoBuf's encoding.
write(pb, field, nid.toBytesBE())
func getField*(pb: ProtoBuffer, field: int,
pr: var PeerRecord): ProtoResult[bool] {.inline.} =
## Read ``NodeId`` from ProtoBuf's message and validate it
@ -55,6 +59,10 @@ func getField*(pb: ProtoBuffer, field: int,
else:
err(ProtoError.IncorrectBlob)
func write*(pb: var ProtoBuffer, field: int, pr: PeerRecord) =
## Write PeerRecord value ``pr`` to object ``pb`` using ProtoBuf's encoding.
write(pb, field, pr.encode())
proc decode*(
T: typedesc[AddProviderMessage],
buffer: openArray[byte]): Result[AddProviderMessage, ProtoError] =
@ -67,6 +75,15 @@ proc decode*(
ok(msg)
proc encode*(msg: AddProviderMessage): seq[byte] =
var pb = initProtoBuffer()
pb.write(1, msg.cId)
pb.write(2, msg.prov)
pb.finish()
pb.buffer
proc addProviderLocal(p: ProvidersProtocol, cId: NodeId, prov: PeerRecord) =
p.providers.mgetOrPut(cId, @[]).add(prov)