mirror of
https://github.com/vacp2p/nim-libp2p.git
synced 2025-01-14 18:54:33 +00:00
e623e70e7b
This adds gossipsub and floodsub, as well as basic interop testing with the go libp2p daemon. * add close event * wip: gossipsub * splitting rpc message * making message handling more consistent * initial gossipsub implementation * feat: nim 1.0 cleanup * wip: gossipsub protobuf * adding encoding/decoding of gossipsub messages * add disconnect handler * add proper gossipsub msg handling * misc: cleanup for nim 1.0 * splitting floodsub and gossipsub tests * feat: add mesh rebalansing * test pubsub * add mesh rebalansing tests * testing mesh maintenance * finishing mcache implementatin * wip: commenting out broken tests * wip: don't run heartbeat for now * switchout debug for trace logging * testing gossip peer selection algorithm * test stream piping * more work around message amplification * get the peerid from message * use timed cache as backing store * allow setting timeout in constructor * several changes to improve performance * more through testing of msg amplification * prevent gc issues * allow piping to self and prevent deadlocks * improove floodsub * allow running hook on cache eviction * prevent race conditions * prevent race conditions and improove tests * use hashes as cache keys * removing useless file * don't create a new seq * re-enable pubsub tests * fix imports * reduce number of runs to speed up tests * break out control message processing * normalize sleeps between steps * implement proper transport filtering * initial interop testing * clean up floodsub publish logic * allow dialing without a protocol * adding multiple reads/writes * use protobuf varint in mplex * don't loose conn's peerInfo * initial interop pubsub tests * don't duplicate connections/peers * bring back interop tests * wip: interop * re-enable interop and daemon tests * add multiple read write tests from handlers * don't cleanup channel prematurely * use correct channel to send/receive msgs * adjust tests with latest changes * include interop tests * remove temp logging output * fix ci * use correct public key serialization * additional tests for pubsub interop
65 lines
2.3 KiB
Nim
65 lines
2.3 KiB
Nim
import options, tables
|
|
import chronos
|
|
import ../../libp2p/[switch,
|
|
peer,
|
|
connection,
|
|
multiaddress,
|
|
peerinfo,
|
|
muxers/muxer,
|
|
crypto/crypto,
|
|
muxers/mplex/mplex,
|
|
muxers/mplex/types,
|
|
protocols/identify,
|
|
transports/transport,
|
|
transports/tcptransport,
|
|
protocols/secure/secure,
|
|
protocols/secure/secio,
|
|
protocols/pubsub/pubsub,
|
|
protocols/pubsub/gossipsub,
|
|
protocols/pubsub/floodsub]
|
|
|
|
proc createMplex(conn: Connection): Muxer =
|
|
result = newMplex(conn)
|
|
|
|
proc createNode*(privKey: Option[PrivateKey] = none(PrivateKey),
|
|
address: string = "/ip4/127.0.0.1/tcp/0",
|
|
triggerSelf: bool = false,
|
|
gossip: bool = false): Switch =
|
|
var peerInfo: PeerInfo
|
|
var seckey = privKey
|
|
if privKey.isNone:
|
|
seckey = some(PrivateKey.random(RSA))
|
|
|
|
peerInfo.peerId = some(PeerID.init(seckey.get()))
|
|
peerInfo.addrs.add(Multiaddress.init(address))
|
|
|
|
let mplexProvider = newMuxerProvider(createMplex, MplexCodec)
|
|
let transports = @[Transport(newTransport(TcpTransport))]
|
|
let muxers = [(MplexCodec, mplexProvider)].toTable()
|
|
let identify = newIdentify(peerInfo)
|
|
let secureManagers = [(SecioCodec, Secure(newSecio(seckey.get())))].toTable()
|
|
|
|
var pubSub: Option[PubSub]
|
|
if gossip:
|
|
pubSub = some(PubSub(newPubSub(GossipSub, peerInfo, triggerSelf)))
|
|
else:
|
|
pubSub = some(PubSub(newPubSub(FloodSub, peerInfo, triggerSelf)))
|
|
|
|
result = newSwitch(peerInfo,
|
|
transports,
|
|
identify,
|
|
muxers,
|
|
secureManagers = secureManagers,
|
|
pubSub = pubSub)
|
|
|
|
proc generateNodes*(num: Natural, gossip: bool = false): seq[Switch] =
|
|
for i in 0..<num:
|
|
result.add(createNode(gossip = gossip))
|
|
|
|
proc subscribeNodes*(nodes: seq[Switch]) {.async.} =
|
|
for dialer in nodes:
|
|
for node in nodes:
|
|
if dialer.peerInfo.peerId != node.peerInfo.peerId:
|
|
await dialer.subscribeToPeer(node.peerInfo)
|
|
await sleepAsync(100.millis)
|