mirror of https://github.com/waku-org/nwaku.git
32 lines
1.1 KiB
Nim
32 lines
1.1 KiB
Nim
|
import
|
||
|
|
json_rpc/rpcserver, stew/endians2, nimcrypto/sysrand,
|
||
|
|
eth/[p2p, async_utils], eth/p2p/rlpx_protocols/waku_protocol
|
||
|
|
|
||
|
|
proc generateTraffic(node: EthereumNode, amount = 100) {.async.} =
|
||
|
|
var topicNumber = 0'u32
|
||
|
|
let payload = @[byte 0]
|
||
|
|
for i in 0..<amount:
|
||
|
|
discard waku_protocol.postMessage(node, ttl = 10,
|
||
|
|
topic = toBytesLE(i.uint32), payload = payload)
|
||
|
|
await sleepAsync(1.milliseconds)
|
||
|
|
|
||
|
|
proc generateRandomTraffic(node: EthereumNode, amount = 100) {.async.} =
|
||
|
|
var topic: array[4, byte]
|
||
|
|
let payload = @[byte 0]
|
||
|
|
for i in 0..<amount:
|
||
|
|
while randomBytes(topic) != 4:
|
||
|
|
discard
|
||
|
|
discard waku_protocol.postMessage(node, ttl = 10, topic = topic,
|
||
|
|
payload = payload)
|
||
|
|
await sleepAsync(1.milliseconds)
|
||
|
|
|
||
|
|
proc setupWakuSimRPC*(node: EthereumNode, rpcsrv: RpcServer) =
|
||
|
|
|
||
|
|
rpcsrv.rpc("wakusim_generateTraffic") do(amount: int) -> bool:
|
||
|
|
traceAsyncErrors node.generateTraffic(amount)
|
||
|
|
return true
|
||
|
|
|
||
|
|
rpcsrv.rpc("wakusim_generateRandomTraffic") do(amount: int) -> bool:
|
||
|
|
traceAsyncErrors node.generateRandomTraffic(amount)
|
||
|
|
return true
|