factorize addToSendQueue

Signed-off-by: Csaba Kiraly <csaba.kiraly@gmail.com>
This commit is contained in:
Csaba Kiraly 2023-02-15 03:18:00 +01:00
parent 0c91eff67b
commit dff0e5523a
No known key found for this signature in database
GPG Key ID: 0FE274EE8C95166E
1 changed files with 15 additions and 16 deletions

View File

@ -213,6 +213,18 @@ class Validator:
# self.statsRxDuplicateInSlot += 1
self.statsRxInSlot += 1
def addToSendQueue(self, rID, cID):
if self.perNodeQueue:
self.sendQueue.append((rID, cID))
if self.perNeighborQueue:
if rID in self.rowIDs:
for neigh in self.rowNeighbors[rID].values():
neigh.sendQueue.append(cID)
if cID in self.columnIDs:
for neigh in self.columnNeighbors[cID].values():
neigh.sendQueue.append(rID)
def receiveRowsColumns(self):
"""It receives rows and columns."""
@ -235,22 +247,9 @@ class Validator:
neigh.receiving.setall(0)
# add newly received segments to the send queue
if self.perNeighborQueue:
self.sendQueue.extend(self.receivedQueue)
if self.perNodeQueue:
while self.receivedQueue:
(rID, cID) = self.receivedQueue.popleft()
if rID in self.rowIDs:
for neigh in self.rowNeighbors[rID].values():
neigh.sendQueue.append(cID)
if cID in self.columnIDs:
for neigh in self.columnNeighbors[cID].values():
neigh.sendQueue.append(rID)
self.receivedQueue.clear()
while self.receivedQueue:
(rID, cID) = self.receivedQueue.popleft()
self.addToSendQueue(rID, cID)
def updateStats(self):
"""It updates the stats related to sent and received data."""