35 lines
975 B
Nim
35 lines
975 B
Nim
|
import std/sequtils
|
||
|
|
|
||
|
|
import pkg/chronos
|
||
|
|
import pkg/libp2p
|
||
|
|
|
||
|
|
import pkg/dagger/bitswap
|
||
|
|
import pkg/dagger/stores/memorystore
|
||
|
|
import pkg/dagger/blocktype as bt
|
||
|
|
|
||
|
|
proc generateNodes*(
|
||
|
|
num: Natural,
|
||
|
|
blocks: openArray[bt.Block] = [],
|
||
|
|
secureManagers: openarray[SecureProtocol] = [
|
||
|
|
SecureProtocol.Noise,
|
||
|
|
]): seq[tuple[switch: Switch, bitswap: Bitswap]] =
|
||
|
|
|
||
|
|
for i in 0..<num:
|
||
|
|
let
|
||
|
|
switch = newStandardSwitch(transportFlags = {ServerFlags.ReuseAddr})
|
||
|
|
network = BitswapNetwork.new(switch = switch)
|
||
|
|
bitswap = Bitswap.new(MemoryStore.new(blocks), network)
|
||
|
|
|
||
|
|
switch.mount(network)
|
||
|
|
|
||
|
|
# initialize our want lists
|
||
|
|
bitswap.engine.wantList = blocks.mapIt( it.cid )
|
||
|
|
switch.mount(network)
|
||
|
|
result.add((switch, bitswap))
|
||
|
|
|
||
|
|
proc connectNodes*(nodes: seq[Switch]) {.async.} =
|
||
|
|
for dialer in nodes:
|
||
|
|
for node in nodes:
|
||
|
|
if dialer.peerInfo.peerId != node.peerInfo.peerId:
|
||
|
|
await dialer.connect(node.peerInfo.peerId, node.peerInfo.addrs)
|