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

Retry room directory lookups if they fail

Browse source
This commit is contained in:
Neil Alexander 2021-03-19 14:31:37 +00:00
parent 9f2b61fb37
commit d960248ea8
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
cmd/dendrite-demo-pinecone/rooms/rooms.go
View file

@ -29,6 +29,8 @@ import (
pineconeSessions "github.com/matrix-org/pinecone/sessions"
)
const pineconeRoomAttempts = 3
type PineconeRoomProvider struct {
r *pineconeRouter.Router
s *pineconeSessions.Sessions
@ -84,12 +86,20 @@ func bulkFetchPublicRoomsFromServers(
go func(homeserverDomain gomatrixserverlib.ServerName) {
defer wg.Done()
util.GetLogger(ctx).WithField("hs", homeserverDomain).Info("Querying HS for public rooms")
fres, err := fedClient.GetPublicRooms(ctx, homeserverDomain, int(limit), "", false, "")
if err != nil {
util.GetLogger(ctx).WithError(err).WithField("hs", homeserverDomain).Warn(
"bulkFetchPublicRoomsFromServers: failed to query hs",
)
return
var fres gomatrixserverlib.RespPublicRooms
var err error
for i := 0; i < pineconeRoomAttempts; i++ {
fres, err = fedClient.GetPublicRooms(ctx, homeserverDomain, int(limit), "", false, "")
if err != nil {
util.GetLogger(ctx).WithError(err).WithField("hs", homeserverDomain).Warn(
"bulkFetchPublicRoomsFromServers: failed to query hs",
)
if i == pineconeRoomAttempts-1 {
return
}
} else {
break
}
}
for _, room := range fres.Chunk {
// atomically send a room or stop