135 lines
3.2 KiB
Go
135 lines
3.2 KiB
Go
package caching
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
lru "github.com/hashicorp/golang-lru"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
)
|
|
|
|
func NewInMemoryLRUCache(enablePrometheus bool) (*Caches, error) {
|
|
roomVersions, err := NewInMemoryLRUCachePartition(
|
|
RoomVersionCacheName,
|
|
RoomVersionCacheMutable,
|
|
RoomVersionCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
serverKeys, err := NewInMemoryLRUCachePartition(
|
|
ServerKeyCacheName,
|
|
ServerKeyCacheMutable,
|
|
ServerKeyCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
roomServerStateKeyNIDs, err := NewInMemoryLRUCachePartition(
|
|
RoomServerStateKeyNIDsCacheName,
|
|
RoomServerStateKeyNIDsCacheMutable,
|
|
RoomServerStateKeyNIDsCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
roomServerEventTypeNIDs, err := NewInMemoryLRUCachePartition(
|
|
RoomServerEventTypeNIDsCacheName,
|
|
RoomServerEventTypeNIDsCacheMutable,
|
|
RoomServerEventTypeNIDsCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
roomServerRoomIDs, err := NewInMemoryLRUCachePartition(
|
|
RoomServerRoomIDsCacheName,
|
|
RoomServerRoomIDsCacheMutable,
|
|
RoomServerRoomIDsCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
roomInfos, err := NewInMemoryLRUCachePartition(
|
|
RoomInfoCacheName,
|
|
RoomInfoCacheMutable,
|
|
RoomInfoCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
federationEvents, err := NewInMemoryLRUCachePartition(
|
|
FederationEventCacheName,
|
|
FederationEventCacheMutable,
|
|
FederationEventCacheMaxEntries,
|
|
enablePrometheus,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Caches{
|
|
RoomVersions: roomVersions,
|
|
ServerKeys: serverKeys,
|
|
RoomServerStateKeyNIDs: roomServerStateKeyNIDs,
|
|
RoomServerEventTypeNIDs: roomServerEventTypeNIDs,
|
|
RoomServerRoomIDs: roomServerRoomIDs,
|
|
RoomInfos: roomInfos,
|
|
FederationEvents: federationEvents,
|
|
}, nil
|
|
}
|
|
|
|
type InMemoryLRUCachePartition struct {
|
|
name string
|
|
mutable bool
|
|
maxEntries int
|
|
lru *lru.Cache
|
|
}
|
|
|
|
func NewInMemoryLRUCachePartition(name string, mutable bool, maxEntries int, enablePrometheus bool) (*InMemoryLRUCachePartition, error) {
|
|
var err error
|
|
cache := InMemoryLRUCachePartition{
|
|
name: name,
|
|
mutable: mutable,
|
|
maxEntries: maxEntries,
|
|
}
|
|
cache.lru, err = lru.New(maxEntries)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if enablePrometheus {
|
|
promauto.NewGaugeFunc(prometheus.GaugeOpts{
|
|
Namespace: "dendrite",
|
|
Subsystem: "caching_in_memory_lru",
|
|
Name: name,
|
|
}, func() float64 {
|
|
return float64(cache.lru.Len())
|
|
})
|
|
}
|
|
return &cache, nil
|
|
}
|
|
|
|
func (c *InMemoryLRUCachePartition) Set(key string, value interface{}) {
|
|
if !c.mutable {
|
|
if peek, ok := c.lru.Peek(key); ok && peek != value {
|
|
panic(fmt.Sprintf("invalid use of immutable cache tries to mutate existing value of %q", key))
|
|
}
|
|
}
|
|
c.lru.Add(key, value)
|
|
}
|
|
|
|
func (c *InMemoryLRUCachePartition) Unset(key string) {
|
|
if !c.mutable {
|
|
panic(fmt.Sprintf("invalid use of immutable cache tries to unset value of %q", key))
|
|
}
|
|
c.lru.Remove(key)
|
|
}
|
|
|
|
func (c *InMemoryLRUCachePartition) Get(key string) (value interface{}, ok bool) {
|
|
return c.lru.Get(key)
|
|
}
|