From 19e6c2a55d970a2203c1547b71edb8f4ef8a7851 Mon Sep 17 00:00:00 2001
From: Oskar Thoren <ot@oskarthoren.com>
Date: Mon, 15 Jun 2020 11:06:08 +0800
Subject: [PATCH] Break out sim into scenarios, one full node mesh vs light

---
 waku/node/v2/quicksim.nim      | 77 ++++++++++++++++++++++++----------
 waku/node/v2/start_network.nim | 22 ++++++----
 2 files changed, 68 insertions(+), 31 deletions(-)

diff --git a/waku/node/v2/quicksim.nim b/waku/node/v2/quicksim.nim
index ba185bb0a..4975696fd 100644
--- a/waku/node/v2/quicksim.nim
+++ b/waku/node/v2/quicksim.nim
@@ -11,11 +11,65 @@ template sourceDir: string = currentSourcePath.rsplit(DirSep, 1)[0]
 const sigWakuPath = &"{sourceDir}{DirSep}rpc{DirSep}wakucallsigs.nim"
 createRpcSigs(RpcHttpClient, sigWakuPath)
 
-const topicAmount = 100
+const topicAmount = 10 #100
 
 proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
   debug "Hit handler", topic=topic, data=data
 
+# Scenario xx1 - 16 full nodes
+#########################################
+let amount = 16
+var nodes: seq[RPCHttpClient]
+for i in 0..<amount:
+  var node = newRpcHttpClient()
+  nodes.add(node)
+  waitFor nodes[i].connect("localhost", Port(8547+i))
+  var res = waitFor nodes[i].wakuSubscribe("waku")
+
+os.sleep(2000)
+
+# TODO: Show plaintext message in log
+for i in 0..<topicAmount:
+  os.sleep(50)
+  # TODO: This would then publish on a subtopic here
+  var s = "hello " & $2
+  var res3 = waitFor nodes[0].wakuPublish("waku", s)
+
+# Scenario xx2 - 14 full nodes, two edge nodes
+# Assume one full topic
+#########################################
+#let nodea = newRpcHttpClient()
+#let nodeb = newRpcHttpClient()
+#
+#waitFor nodea.connect("localhost", Port(8545))
+#waitFor nodeb.connect("localhost", Port(8546))
+#
+#let version = waitFor nodea.wakuVersion()
+#info "Version is", version
+#
+#let res1 = waitFor nodea.wakuSubscribe("waku")
+#let res2 = waitFor nodeb.wakuSubscribe("waku")
+#
+#let amount = 14
+#var nodes: seq[RPCHttpClient]
+#for i in 0..<amount:
+#  var node = newRpcHttpClient()
+#  nodes.add(node)
+#  waitFor nodes[i].connect("localhost", Port(8547+i))
+#  var res = waitFor nodes[i].wakuSubscribe("waku")
+#
+#os.sleep(2000)
+#
+## TODO: Show plaintext message in log
+#for i in 0..<topicAmount:
+#  os.sleep(50)
+#  # TODO: This would then publish on a subtopic here
+#  var s = "hello " & $2
+#  var res3 = waitFor nodea.wakuPublish("waku", s)
+
+# Misc old scenarios
+#########################################
+
 # All full nodes connected etc
 #
 # let node1 = newRpcHttpClient()
@@ -55,19 +109,6 @@ proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
 #   os.sleep(50)
 #   let res2 = waitFor node1.wakuPublish("foobar", "hello world2")
 
-let nodea = newRpcHttpClient()
-let nodeb = newRpcHttpClient()
-
-waitFor nodea.connect("localhost", Port(8545))
-waitFor nodeb.connect("localhost", Port(8546))
-
-let version = waitFor nodea.wakuVersion()
-info "Version is", version
-
-# XXX: Unclear if we want nodea to subscribe to own topic or not
-let res1 = waitFor nodea.wakuSubscribe("foobar")
-let res2 = waitFor nodeb.wakuSubscribe("foobar")
-
 # Node 00 and 05 also subscribe
 # XXX I confirm this works. As in - with this we have A-B
 #  Now to tweak it!
@@ -77,11 +118,3 @@ let res2 = waitFor nodeb.wakuSubscribe("foobar")
 # waitFor node5.connect("localhost", Port(8552))
 # let res4 = waitFor node0.wakuSubscribe("foobar")
 # let res5 = waitFor node5.wakuSubscribe("foobar")
-
-os.sleep(2000)
-
-# XXX: Where is hello world tho?
-for i in 0..<topicAmount:
-  os.sleep(50)
-  var s = "hello " & $2
-  var res3 = waitFor nodea.wakuPublish("foobar", s)
diff --git a/waku/node/v2/start_network.nim b/waku/node/v2/start_network.nim
index ca11b112d..43588e733 100644
--- a/waku/node/v2/start_network.nim
+++ b/waku/node/v2/start_network.nim
@@ -80,7 +80,9 @@ when isMainModule:
   # TODO: WakuNetworkConf
   var nodes: seq[NodeInfo]
   let topology = FullMesh
-  let amount = 6
+
+  # Scenario xx2 14
+  let amount = 16
 
   case topology:
     of Star:
@@ -93,15 +95,17 @@ when isMainModule:
     # TODO: could also select nodes randomly
     staticnodes.add(nodes[i].address)
 
-  # TODO: Here we could add a light node, but not clear thats what we want to test?
-  # Lets call them edge nodes
+  # Scenario xx1 - 16 full nodes, one app topic, full mesh, gossip
+
+  # Scenario xx2 - 14 full nodes, two edge nodes, one app topic, full mesh, gossip
   # NOTE: Only connecting to one node here
-  var nodesubseta: seq[string]
-  var nodesubsetb: seq[string]
-  nodesubseta.add(staticnodes[0])
-  nodesubsetb.add(staticnodes[amount-1])
-  nodes.add(initNodeCmd(0, nodesubseta, label = "edge node (A)"))
-  nodes.add(initNodeCmd(1, nodesubsetb, label = "edge node (B)"))
+  #var nodesubseta: seq[string]
+  #var nodesubsetb: seq[string]
+  #nodesubseta.add(staticnodes[0])
+  #nodesubsetb.add(staticnodes[amount-1])
+  ## XXX: Let's turn them into normal nodes
+  #nodes.add(initNodeCmd(0, nodesubseta, label = "edge node (A)"))
+  #nodes.add(initNodeCmd(1, nodesubsetb, label = "edge node (B)"))
 
   var commandStr = "multitail -s 2 -M 0 -x \"Waku Simulation\""
   var count = 0