mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-26 00:03:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

Comments

Browse source
This commit is contained in:
Neil Alexander 2020-12-08 16:46:07 +00:00
parent 3a15792b31
commit b64e79e3ac
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 16 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
federationsender/queue/destinationqueue.go
View file

@ -204,6 +204,7 @@ func (oq *destinationQueue) getPendingFromDatabase() {
}
} else {
logrus.WithError(err).Errorf("Failed to get pending PDUs for %q", oq.destination)
return
}
}
if eduCapacity := maxEDUsInMemory - len(oq.pendingEDUs); eduCapacity > 0 {
@ -218,6 +219,7 @@ func (oq *destinationQueue) getPendingFromDatabase() {
}
} else {
logrus.WithError(err).Errorf("Failed to get pending EDUs for %q", oq.destination)
return
}
}
// If we've retrieved all of the events from the database with room to spare
@ -326,20 +328,24 @@ func (oq *destinationQueue) backgroundSend() {
// If we successfully sent the transaction then clear out
// the pending events and EDUs, and wipe our transaction ID.
oq.statistics.Success()
oq.pendingMutex.Lock()
for i := range oq.pendingPDUs[:pc] {
oq.pendingPDUs[i] = nil
}
for i := range oq.pendingEDUs[:ec] {
oq.pendingEDUs[i] = nil
}
oq.pendingPDUs = oq.pendingPDUs[pc:]
oq.pendingEDUs = oq.pendingEDUs[ec:]
oq.pendingMutex.Unlock()
oq.deleteUpto(pc, ec)
}
}
}
func (oq *destinationQueue) deleteUpto(pc, ec int) {
oq.pendingMutex.Lock()
for i := range oq.pendingPDUs[:pc] {
oq.pendingPDUs[i] = nil
}
for i := range oq.pendingEDUs[:ec] {
oq.pendingEDUs[i] = nil
}
oq.pendingPDUs = nil
oq.pendingEDUs = nil
oq.pendingMutex.Unlock()
}
// nextTransaction creates a new transaction from the pending event
// queue and sends it. Returns true if a transaction was sent or
// false otherwise.