mirror of
https://github.com/1f349/dendrite.git
synced 2024-12-23 00:34:07 +00:00
b891c00b09
* Add RoomInfo cache, remove RoomServerRoomNID cache, ensure caches are thread-safe * Don't panic if the roomInfo isn't known yet * LRU package is already threadsafe * Use RoomInfo cache to find room version if possible in Events() * Adding comments about RoomInfoCache safety
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
package caching
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/matrix-org/dendrite/roomserver/types"
|
|
)
|
|
|
|
const (
|
|
RoomServerStateKeyNIDsCacheName = "roomserver_statekey_nids"
|
|
RoomServerStateKeyNIDsCacheMaxEntries = 1024
|
|
RoomServerStateKeyNIDsCacheMutable = false
|
|
|
|
RoomServerEventTypeNIDsCacheName = "roomserver_eventtype_nids"
|
|
RoomServerEventTypeNIDsCacheMaxEntries = 64
|
|
RoomServerEventTypeNIDsCacheMutable = false
|
|
|
|
RoomServerRoomIDsCacheName = "roomserver_room_ids"
|
|
RoomServerRoomIDsCacheMaxEntries = 1024
|
|
RoomServerRoomIDsCacheMutable = false
|
|
)
|
|
|
|
type RoomServerCaches interface {
|
|
RoomServerNIDsCache
|
|
RoomVersionCache
|
|
RoomInfoCache
|
|
}
|
|
|
|
// RoomServerNIDsCache contains the subset of functions needed for
|
|
// a roomserver NID cache.
|
|
type RoomServerNIDsCache interface {
|
|
GetRoomServerStateKeyNID(stateKey string) (types.EventStateKeyNID, bool)
|
|
StoreRoomServerStateKeyNID(stateKey string, nid types.EventStateKeyNID)
|
|
|
|
GetRoomServerEventTypeNID(eventType string) (types.EventTypeNID, bool)
|
|
StoreRoomServerEventTypeNID(eventType string, nid types.EventTypeNID)
|
|
|
|
GetRoomServerRoomID(roomNID types.RoomNID) (string, bool)
|
|
StoreRoomServerRoomID(roomNID types.RoomNID, roomID string)
|
|
}
|
|
|
|
func (c Caches) GetRoomServerStateKeyNID(stateKey string) (types.EventStateKeyNID, bool) {
|
|
val, found := c.RoomServerStateKeyNIDs.Get(stateKey)
|
|
if found && val != nil {
|
|
if stateKeyNID, ok := val.(types.EventStateKeyNID); ok {
|
|
return stateKeyNID, true
|
|
}
|
|
}
|
|
return 0, false
|
|
}
|
|
|
|
func (c Caches) StoreRoomServerStateKeyNID(stateKey string, nid types.EventStateKeyNID) {
|
|
c.RoomServerStateKeyNIDs.Set(stateKey, nid)
|
|
}
|
|
|
|
func (c Caches) GetRoomServerEventTypeNID(eventType string) (types.EventTypeNID, bool) {
|
|
val, found := c.RoomServerEventTypeNIDs.Get(eventType)
|
|
if found && val != nil {
|
|
if eventTypeNID, ok := val.(types.EventTypeNID); ok {
|
|
return eventTypeNID, true
|
|
}
|
|
}
|
|
return 0, false
|
|
}
|
|
|
|
func (c Caches) StoreRoomServerEventTypeNID(eventType string, nid types.EventTypeNID) {
|
|
c.RoomServerEventTypeNIDs.Set(eventType, nid)
|
|
}
|
|
|
|
func (c Caches) GetRoomServerRoomID(roomNID types.RoomNID) (string, bool) {
|
|
val, found := c.RoomServerRoomIDs.Get(strconv.Itoa(int(roomNID)))
|
|
if found && val != nil {
|
|
if roomID, ok := val.(string); ok {
|
|
return roomID, true
|
|
}
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
func (c Caches) StoreRoomServerRoomID(roomNID types.RoomNID, roomID string) {
|
|
c.RoomServerRoomIDs.Set(strconv.Itoa(int(roomNID)), roomID)
|
|
}
|