31 lines
690 B
Nim
31 lines
690 B
Nim
import pkg/libp2p/varint
|
|
|
|
import pkg/dagger/chunker
|
|
import pkg/dagger/blocktype
|
|
import pkg/dagger/blockstream
|
|
|
|
import pkg/questionable
|
|
import pkg/questionable/results
|
|
|
|
export chunker
|
|
|
|
type
|
|
TestStreamProc* = proc(): ?!Block {.raises: [Defect].}
|
|
|
|
TestStream* = ref object of BlockStreamRef
|
|
handler*: TestStreamProc
|
|
|
|
method nextBlock*(b: TestStream): ?!Block =
|
|
b.handler()
|
|
|
|
proc lenPrefix*(msg: openArray[byte]): seq[byte] =
|
|
## Write `msg` with a varint-encoded length prefix
|
|
##
|
|
|
|
let vbytes = PB.toBytes(msg.len().uint64)
|
|
var buf = newSeqUninitialized[byte](msg.len() + vbytes.len)
|
|
buf[0..<vbytes.len] = vbytes.toOpenArray()
|
|
buf[vbytes.len..<buf.len] = msg
|
|
|
|
return buf
|