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

40
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
if err != nil { fetchEventTypes := make([]string, 0, len(eventTypes))
return nil, err for _, eventType := range eventTypes {
eventTypeNID, ok := d.Cache.GetEventTypeKey(eventType)
if ok {
result[eventType] = eventTypeNID
continue
}
fetchEventTypes = append(fetchEventTypes, eventType)
} }
for eventType, nid := range nids { if len(fetchEventTypes) > 0 {
result[eventType] = nid nids, err := d.EventTypesTable.BulkSelectEventTypeNID(ctx, txn, fetchEventTypes)
if err != nil {
return nil, err
}
for eventType, nid := range nids {
result[eventType] = nid
d.Cache.StoreEventTypeKey(nid, eventType)
}
} }
return result, nil return result, nil
} }
@ -90,13 +103,15 @@ func (d *Database) EventStateKeys(
fetch = append(fetch, nid) fetch = append(fetch, nid)
} }
} }
fromDB, err := d.EventStateKeysTable.BulkSelectEventStateKey(ctx, nil, fetch) if len(fetch) > 0 {
if err != nil { fromDB, err := d.EventStateKeysTable.BulkSelectEventStateKey(ctx, nil, fetch)
return nil, err if err != nil {
} return nil, err
for nid, key := range fromDB { }
result[nid] = key for nid, key := range fromDB {
d.Cache.StoreEventStateKey(nid, key) result[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)
} }
} }