Hydrate destination queues at startup (#1179)

* Hydrate destination queues at startup

* Review comments
This commit is contained in:
Neil Alexander 2020-07-03 11:49:49 +01:00 committed by GitHub
parent 9dffeb9b67
commit 1773fd84b7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 89 additions and 1 deletions

View file

@ -51,7 +51,7 @@ func NewOutgoingQueues(
statistics *types.Statistics,
signing *SigningInfo,
) *OutgoingQueues {
return &OutgoingQueues{
queues := &OutgoingQueues{
db: db,
rsAPI: rsAPI,
origin: origin,
@ -60,6 +60,15 @@ func NewOutgoingQueues(
signing: signing,
queues: map[gomatrixserverlib.ServerName]*destinationQueue{},
}
// Look up which servers we have pending items for and then rehydrate those queues.
if serverNames, err := db.GetPendingServerNames(context.Background()); err == nil {
for _, serverName := range serverNames {
queues.getQueue(serverName).wakeQueueIfNeeded()
}
} else {
log.WithError(err).Error("Failed to get server names for destination queue hydration")
}
return queues
}
// TODO: Move this somewhere useful for other components as we often need to ferry these 3 variables