Looking for websocket issues.

Author: MihaelBercic

src/main/kotlin/chain/Chain.kt

@@ -16,7 +16,11 @@ import kotlin.concurrent.withLock
  * on 16/11/2021 at 16:21
  * using IntelliJ IDEA
  */
-class Chain(private val verifiableDelay: VerifiableDelay, private val initialDifficulty: Int, private val committeeSize: Int) {
+class Chain(
+    private val verifiableDelay: VerifiableDelay,
+    private val initialDifficulty: Int,
+    private val committeeSize: Int
+) {
 
     private val lock = ReentrantLock(true)
     private val blocks = CircularList<Block>(50) // ToDo: Remove, do not use Circular List but use persistent storage!
@@ -37,7 +41,8 @@ class Chain(private val verifiableDelay: VerifiableDelay, private val initialDif
             val lastBlock = getLastBlock()
             val lastHash = lastBlock?.hash ?: "FFFF".toByteArray()
             val difficulty = lastBlock?.difficulty ?: initialDifficulty
-            val isLegitimate = nextBlock.slot == (lastBlock?.slot ?: 0) + 1 // verifiableDelay.verifyProof(lastHash, difficulty, nextBlock.vdfProof)
+            val isLegitimate = nextBlock.slot == (lastBlock?.slot
+                ?: 0) + 1 // verifiableDelay.verifyProof(lastHash, difficulty, nextBlock.vdfProof)
             if (!isLegitimate) {
                 Logger.trace("Proof is not legitimate for block ${nextBlock.slot}!")
                 Logger.chain("Last hash: ${lastHash.asHex}")
@@ -48,6 +53,7 @@ class Chain(private val verifiableDelay: VerifiableDelay, private val initialDif
                 return false
             }
             lock.tryWithLock {
+
                 blocks.add(nextBlock)
                 // ToDo: Put chain history in some sort of storage instead of keeping in memory.
                 transaction {

src/main/kotlin/network/messaging/Server.kt

@@ -65,6 +65,7 @@ abstract class Server(val configuration: Configuration) : RPCManager(configurati
             Logger.trace("Clearing message history...")
             messageHistory.entries.removeIf { (_, value) -> currentTime - value > maximumAge }
         }
+
     }
 
     abstract fun processMessage(message: Message)
@@ -91,7 +92,10 @@ abstract class Server(val configuration: Configuration) : RPCManager(configurati
 
     /** Returns [Configuration.broadcastSpreadPercentage] number of nodes.  */
     fun pickRandomNodes(amount: Int = 0): List<Node> {
-        val toTake = if (amount > 0) amount else 5 + (configuration.broadcastSpreadPercentage * Integer.max(totalKnownNodes, 1) / 100)
+        val toTake = if (amount > 0) amount else 5 + (configuration.broadcastSpreadPercentage * Integer.max(
+            totalKnownNodes,
+            1
+        ) / 100)
         return getRandomNodes(toTake).filter { it.identifier != localNode.identifier }
     }
 
@@ -160,7 +164,11 @@ abstract class Server(val configuration: Configuration) : RPCManager(configurati
                 val message = ProtoBuf.decodeFromByteArray<Message>(data)
                 if (alreadySeen(message.uid.asHex)) return@use
                 processingQueue.add(message)
-                if (message.endpoint.transmissionType == TransmissionType.Broadcast) broadcast(TransmissionLayer.TCP, message.uid.asHex, data)
+                if (message.endpoint.transmissionType == TransmissionType.Broadcast) broadcast(
+                    TransmissionLayer.TCP,
+                    message.uid.asHex,
+                    data
+                )
             }
         }
     }
@@ -199,22 +207,30 @@ abstract class Server(val configuration: Configuration) : RPCManager(configurati
                 broadcastNodes.add(neighbour)
                 broadcastNodes.addAll(childrenKeys)
                 broadcastNodes.addAll(neighbourChildrenKeys)
-                Logger.error("[$index] [$children] Neighbour: $neighbourIndex ... Children: ${childrenKeys.joinToString(",") { "${shuffled.indexOf(it)}" }}")
+                Logger.error(
+                    "[$index] [$children] Neighbour: $neighbourIndex ... Children: ${
+                        childrenKeys.joinToString(
+                            ","
+                        ) { "${shuffled.indexOf(it)}" }
+                    }"
+                )
 
             }
 
             else -> broadcastNodes.addAll(pickRandomNodes().map { it.publicKey })
         }
 
-        val knownAndNotInSet = knownNodes.values.map(Node::publicKey).filter { !validatorSet.activeValidators.contains(it) }
+        val knownAndNotInSet =
+            knownNodes.values.map(Node::publicKey).filter { !validatorSet.activeValidators.contains(it) }
         broadcastNodes.addAll(knownAndNotInSet)
 
         Logger.trace("We have to retransmit to [total: ${shuffled.size}] --> ${broadcastNodes.size} nodes.")
 
         broadcastNodes.forEach { publicKey ->
             query(publicKey) {
                 val outgoingData = OutgoingData(it, *data)
-                val outgoingQueue = if (transmissionLayer == TransmissionLayer.UDP) udpOutgoingQueue else tcpOutgoingQueue
+                val outgoingQueue =
+                    if (transmissionLayer == TransmissionLayer.UDP) udpOutgoingQueue else tcpOutgoingQueue
                 outgoingQueue.add(outgoingData)
             }
         }
@@ -242,7 +258,11 @@ abstract class Server(val configuration: Configuration) : RPCManager(configurati
             if (alreadySeen(packetId) || alreadySeen(messageId)) return@tryAndReport
 
             val endpoint = Endpoint.byId(inputStream.read().toByte()) ?: return@tryAndReport
-            if (endpoint.transmissionType == TransmissionType.Broadcast) broadcast(TransmissionLayer.UDP, messageId, packet.data.copyOf())
+            if (endpoint.transmissionType == TransmissionType.Broadcast) broadcast(
+                TransmissionLayer.UDP,
+                messageId,
+                packet.data.copyOf()
+            )
 
             val totalSlices = inputStream.readInt()
             val currentSlice = inputStream.readInt()

src/main/kotlin/network/rpc/RPCManager.kt

@@ -77,6 +77,7 @@ open class RPCManager(configuration: Configuration) : Kademlia(configuration) {
         val serialisedData = Json.encodeToString(message)
         val clientList = subscribedClients[topic] ?: return
         clientList.forEach {
+            println("Websocket data: $serialisedData")
             try {
                 it.send(serialisedData)
             } catch (e: Exception) {