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 return isBlacklisted, nil
} }
func (d *fakeDatabase) GetMailserversForServer(serverName gomatrixserverlib.ServerName) ([]gomatrixserverlib.ServerName, error) {
return []gomatrixserverlib.ServerName{}, nil
}
type stubFederationRoomServerAPI struct { type stubFederationRoomServerAPI struct {
rsapi.FederationRoomserverAPI rsapi.FederationRoomserverAPI
} }

View file

@ -77,6 +77,13 @@ func (s *Statistics) ForServer(serverName gomatrixserverlib.ServerName) *ServerS
} else { } else {
server.blacklisted.Store(blacklisted) 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 return server
} }