Rename Prepare to NewStateResolution

This commit is contained in:
Neil Alexander 2020-03-19 18:06:37 +00:00
parent 5fcb394cb6
commit f6b1861c40
5 changed files with 9 additions and 8 deletions

View file

@ -157,7 +157,7 @@ func calculateAndSetState(
event gomatrixserverlib.Event,
) error {
var err error
roomState := state.Prepare(db)
roomState := state.NewStateResolution(db)
if input.HasState {
// We've been told what the state at the event is so we don't need to calculate it.

View file

@ -178,7 +178,7 @@ func (u *latestEventsUpdater) doUpdateLatestEvents() error {
func (u *latestEventsUpdater) latestState() error {
var err error
roomState := state.Prepare(u.db)
roomState := state.NewStateResolution(u.db)
latestStateAtEvents := make([]types.StateAtEvent, len(u.latest))
for i := range u.latest {

View file

@ -112,7 +112,7 @@ func (r *RoomserverQueryAPI) QueryLatestEventsAndState(
return nil
}
roomState := state.Prepare(r.DB)
roomState := state.NewStateResolution(r.DB)
response.QueryLatestEventsAndStateRequest = *request
roomNID, err := r.DB.RoomNID(ctx, request.RoomID)
@ -163,7 +163,7 @@ func (r *RoomserverQueryAPI) QueryStateAfterEvents(
return nil
}
roomState := state.Prepare(r.DB)
roomState := state.NewStateResolution(r.DB)
response.QueryStateAfterEventsRequest = *request
roomNID, err := r.DB.RoomNID(ctx, request.RoomID)
@ -354,7 +354,7 @@ func (r *RoomserverQueryAPI) QueryMembershipsForRoom(
func (r *RoomserverQueryAPI) getMembershipsBeforeEventNID(
ctx context.Context, eventNID types.EventNID, joinedOnly bool,
) ([]types.Event, error) {
roomState := state.Prepare(r.DB)
roomState := state.NewStateResolution(r.DB)
events := []types.Event{}
// Lookup the event NID
eIDs, err := r.DB.EventIDs(ctx, []types.EventNID{eventNID})
@ -456,7 +456,7 @@ func (r *RoomserverQueryAPI) QueryServerAllowedToSeeEvent(
func (r *RoomserverQueryAPI) checkServerAllowedToSeeEvent(
ctx context.Context, eventID string, serverName gomatrixserverlib.ServerName,
) (bool, error) {
roomState := state.Prepare(r.DB)
roomState := state.NewStateResolution(r.DB)
stateEntries, err := roomState.LoadStateAtEvent(ctx, eventID)
if err != nil {
return false, err
@ -676,7 +676,7 @@ func (r *RoomserverQueryAPI) QueryStateAndAuthChain(
}
func (r *RoomserverQueryAPI) loadStateAtEventIDs(ctx context.Context, eventIDs []string) ([]gomatrixserverlib.Event, error) {
roomState := state.Prepare(r.DB)
roomState := state.NewStateResolution(r.DB)
prevStates, err := r.DB.StateAtEventIDs(ctx, eventIDs)
if err != nil {
switch err.(type) {

View file

@ -0,0 +1 @@
package shared

View file

@ -35,7 +35,7 @@ type StateResolution struct {
db database.RoomStateDatabase
}
func Prepare(db database.RoomStateDatabase) StateResolution {
func NewStateResolution(db database.RoomStateDatabase) StateResolution {
return StateResolution{
db: db,
}