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

Fix #622 /get_missing_events/{roomId}

Browse source 
Signed-off-by: Alex Chen <minecnly@gmail.com>
This commit is contained in:
Cnly 2019-01-09 16:27:03 +08:00
parent 170264404e
commit 8feb65dafc
2 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/github.com/matrix-org/dendrite/federationapi/routing/routing.go
View file

@ -212,13 +212,13 @@ func Setup(
}, },
)).Methods(http.MethodGet) )).Methods(http.MethodGet)
v1fedmux.Handle("get_missing_events/{roomID}", common.MakeFedAPI( v1fedmux.Handle("/get_missing_events/{roomID}", common.MakeFedAPI(
"federation_get_missing_events", cfg.Matrix.ServerName, keys, "federation_get_missing_events", cfg.Matrix.ServerName, keys,
func(httpReq *http.Request, request *gomatrixserverlib.FederationRequest) util.JSONResponse { func(httpReq *http.Request, request *gomatrixserverlib.FederationRequest) util.JSONResponse {
vars := mux.Vars(httpReq) vars := mux.Vars(httpReq)
return GetMissingEvents(httpReq, request, query, vars["roomID"]) return GetMissingEvents(httpReq, request, query, vars["roomID"])
}, },
)).Methods(http.MethodGet) )).Methods(http.MethodPost)
v1fedmux.Handle("/backfill/{roomID}/", common.MakeFedAPI( v1fedmux.Handle("/backfill/{roomID}/", common.MakeFedAPI(
"federation_backfill", cfg.Matrix.ServerName, keys, "federation_backfill", cfg.Matrix.ServerName, keys,

15
src/github.com/matrix-org/dendrite/roomserver/query/query.go
View file

@ -438,6 +438,7 @@ func (r *RoomserverQueryAPI) QueryMissingEvents(
response *api.QueryMissingEventsResponse, response *api.QueryMissingEventsResponse,
) error { ) error {
var front []string var front []string
eventsToFilter := make(map[string]bool)
visited := make(map[string]bool, request.Limit) // request.Limit acts as a hint to size. visited := make(map[string]bool, request.Limit) // request.Limit acts as a hint to size.
for _, id := range request.EarliestEvents { for _, id := range request.EarliestEvents {
visited[id] = true visited[id] = true
@ -446,6 +447,7 @@ func (r *RoomserverQueryAPI) QueryMissingEvents(
for _, id := range request.LatestEvents { for _, id := range request.LatestEvents {
if !visited[id] { if !visited[id] {
front = append(front, id) front = append(front, id)
eventsToFilter[id] = true
} }
} }
@ -454,7 +456,18 @@ func (r *RoomserverQueryAPI) QueryMissingEvents(
return err return err
} }
response.Events, err = r.loadEvents(ctx, resultNIDs) loadedEvents, err := r.loadEvents(ctx, resultNIDs)
if err != nil {
return err
}
response.Events = make([]gomatrixserverlib.Event, 0, len(loadedEvents)-len(eventsToFilter))
for _, event := range loadedEvents {
if !eventsToFilter[event.EventID()] {
response.Events = append(response.Events, event)
}
}
return err return err
} }