Query mailservers when creating a new server statistics instance

This commit is contained in:
Devon Hudson 2022-11-29 17:51:04 -07:00
parent 0520a9b0ed
commit 0c86e8269d
No known key found for this signature in database
GPG key ID: CD06B18E77F6A628
2 changed files with 11 additions and 0 deletions

View file

@ -301,6 +301,10 @@ func (d *fakeDatabase) IsServerBlacklisted(serverName gomatrixserverlib.ServerNa
return isBlacklisted, nil
}
func (d *fakeDatabase) GetMailserversForServer(serverName gomatrixserverlib.ServerName) ([]gomatrixserverlib.ServerName, error) {
return []gomatrixserverlib.ServerName{}, nil
}
type stubFederationRoomServerAPI struct {
rsapi.FederationRoomserverAPI
}

View file

@ -77,6 +77,13 @@ func (s *Statistics) ForServer(serverName gomatrixserverlib.ServerName) *ServerS
} else {
server.blacklisted.Store(blacklisted)
}
knownMailservers, err := s.DB.GetMailserversForServer(serverName)
if err != nil {
logrus.WithError(err).Errorf("Failed to get mailserver list for %q", serverName)
} else {
server.knownMailservers = knownMailservers
}
}
return server
}