mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-07 06:03:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

One more time

Browse source
This commit is contained in:
Neil Alexander 2022-06-15 10:45:44 +01:00
parent caccf7c663
commit 17091ee8e6
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 25 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
roomserver/storage/shared/storage.go
View file

@ -435,32 +435,25 @@ func (d *Database) Events(
func (d *Database) events(
ctx context.Context, txn *sql.Tx, inputEventNIDs []types.EventNID,
) ([]types.Event, error) {
results := make([]types.Event, len(inputEventNIDs))
positions := make(map[types.EventNID]int, len(inputEventNIDs))
inCache := make([]bool, len(inputEventNIDs))
inDatabase := make([]types.EventNID, 0, len(results))
for i, nid := range inputEventNIDs {
events := make(map[types.EventNID]*gomatrixserverlib.Event, len(inputEventNIDs))
eventNIDs := make([]types.EventNID, 0, len(inputEventNIDs))
for _, nid := range inputEventNIDs {
if event, ok := d.Cache.GetRoomServerEvent(nid); ok && event != nil {
results[i] = types.Event{
EventNID: nid,
Event: event,
}
inCache[i] = true
events[nid] = event
} else {
inDatabase = append(inDatabase, nid)
positions[nid] = i
eventNIDs = append(eventNIDs, nid)
}
}
eventJSONs, err := d.EventJSONTable.BulkSelectEventJSON(ctx, txn, inDatabase)
eventJSONs, err := d.EventJSONTable.BulkSelectEventJSON(ctx, txn, eventNIDs)
if err != nil {
return nil, err
}
eventIDs, _ := d.EventsTable.BulkSelectEventID(ctx, txn, inDatabase)
eventIDs, _ := d.EventsTable.BulkSelectEventID(ctx, txn, eventNIDs)
if err != nil {
eventIDs = map[types.EventNID]string{}
}
var roomNIDs map[types.EventNID]types.RoomNID
roomNIDs, err = d.EventsTable.SelectRoomNIDsForEventNIDs(ctx, txn, inDatabase)
roomNIDs, err = d.EventsTable.SelectRoomNIDsForEventNIDs(ctx, txn, eventNIDs)
if err != nil {
return nil, err
}
@ -487,20 +480,29 @@ func (d *Database) events(
roomVersions[n] = v
}
for _, eventJSON := range eventJSONs {
eventNID := eventJSON.EventNID
result := &results[positions[eventNID]]
result.EventNID = eventNID
roomNID := roomNIDs[eventNID]
result.Event, err = gomatrixserverlib.NewEventFromTrustedJSONWithEventID(
eventIDs[eventNID], eventJSON.EventJSON, false, roomVersions[roomNID],
roomNID := roomNIDs[eventJSON.EventNID]
roomVersion := roomVersions[roomNID]
events[eventJSON.EventNID], err = gomatrixserverlib.NewEventFromTrustedJSONWithEventID(
eventIDs[eventJSON.EventNID], eventJSON.EventJSON, false, roomVersion,
)
if err != nil {
return nil, err
}
if result.Event != nil {
d.Cache.StoreRoomServerEvent(eventNID, result.Event)
if event := events[eventJSON.EventNID]; event != nil {
d.Cache.StoreRoomServerEvent(eventJSON.EventNID, event)
}
}
results := make([]types.Event, 0, len(inputEventNIDs))
for _, nid := range inputEventNIDs {
event, ok := events[nid]
if !ok {
panic("missing event")
}
results = append(results, types.Event{
EventNID: nid,
Event: event,
})
}
if !redactionsArePermanent {
d.applyRedactions(results)
}