sigb.us/dendrite
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases 1 Wiki Activity

Cache all the things

Browse source
This commit is contained in:
Till Faelligen 2023-02-24 11:45:01 +01:00
parent ad07b169b8
commit 3d31b131fc
No known key found for this signature in database
GPG key ID: ACCDC9606D472758
1 changed files with 28 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -68,12 +68,25 @@ func (d *Database) eventTypeNIDs(
ctx context.Context, txn *sql.Tx, eventTypes []string, ctx context.Context, txn *sql.Tx, eventTypes []string,
) (map[string]types.EventTypeNID, error) { ) (map[string]types.EventTypeNID, error) {
result := make(map[string]types.EventTypeNID) result := make(map[string]types.EventTypeNID)
nids, err := d.EventTypesTable.BulkSelectEventTypeNID(ctx, txn, eventTypes) // first try the cache
fetchEventTypes := make([]string, 0, len(eventTypes))
for _, eventType := range eventTypes {
eventTypeNID, ok := d.Cache.GetEventTypeKey(eventType)
if ok {
result[eventType] = eventTypeNID
continue
}
fetchEventTypes = append(fetchEventTypes, eventType)
}
if len(fetchEventTypes) > 0 {
nids, err := d.EventTypesTable.BulkSelectEventTypeNID(ctx, txn, fetchEventTypes)
if err != nil { if err != nil {
return nil, err return nil, err
} }
for eventType, nid := range nids { for eventType, nid := range nids {
result[eventType] = nid result[eventType] = nid
d.Cache.StoreEventTypeKey(nid, eventType)
}
} }
return result, nil return result, nil
} }
@ -90,6 +103,7 @@ func (d *Database) EventStateKeys(
fetch = append(fetch, nid) fetch = append(fetch, nid)
} }
} }
if len(fetch) > 0 {
fromDB, err := d.EventStateKeysTable.BulkSelectEventStateKey(ctx, nil, fetch) fromDB, err := d.EventStateKeysTable.BulkSelectEventStateKey(ctx, nil, fetch)
if err != nil { if err != nil {
return nil, err return nil, err
@ -98,6 +112,7 @@ func (d *Database) EventStateKeys(
result[nid] = key result[nid] = key
d.Cache.StoreEventStateKey(nid, key) d.Cache.StoreEventStateKey(nid, key)
} }
}
return result, nil return result, nil
} }
@ -130,6 +145,7 @@ func (d *Database) eventStateKeyNIDs(
} }
for eventStateKey, nid := range nids { for eventStateKey, nid := range nids {
result[eventStateKey] = nid result[eventStateKey] = nid
d.Cache.StoreEventStateKey(nid, eventStateKey)
} }
} }