mirror of
https://github.com/1f349/dendrite.git
synced 2024-12-23 08:44:11 +00:00
3ea21273bc
* Try Ristretto cache * Tweak * It's beautiful * Update GMSL * More strict keyable interface * Fix that some more * Make less panicky * Don't enforce mutability checks for now * Determine mutability using deep equality * Tweaks * Namespace keys * Make federation caches mutable * Update cost estimation, add metric * Update GMSL * Estimate cost for metrics better * Reduce counters a bit * Try caching events * Some guards * Try again * Try this * Use separate caches for hopefully better hash distribution * Fix bug with admitting events into cache * Try to fix bugs * Check nil * Try that again * Preserve order jeezo this is messy * thanks VS Code for doing exactly the wrong thing * Try this again * Be more specific * aaaaargh * One more time * That might be better * Stronger sorting * Cache expiries, async publishing of EDUs * Put it back * Use a shared cache again * Cost estimation fixes * Update ristretto * Reduce counters a bit * Clean up a bit * Update GMSL * 1GB * Configurable cache sizees * Tweaks * Add `config.DataUnit` for specifying friendly cache sizes * Various tweaks * Update GMSL * Add back some lazy loading caching * Include key in cost * Include key in cost * Tweak max age handling, config key name * Only register prometheus metrics if requested * Review comments @S7evinK * Don't return errors when creating caches (it is better just to crash since otherwise we'll `nil`-pointer exception everywhere) * Review comments * Update sample configs * Update GHA Workflow * Update Complement images to Go 1.18 * Remove the cache test from the federation API as we no longer guarantee immediate cache admission * Don't check the caches in the renewal test * Possibly fix the upgrade tests * Update to matrix-org/gomatrixserverlib#322 * Update documentation to refer to Go 1.18
46 lines
1.8 KiB
Go
46 lines
1.8 KiB
Go
package caching
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
)
|
|
|
|
// ServerKeyCache contains the subset of functions needed for
|
|
// a server key cache.
|
|
type ServerKeyCache interface {
|
|
// request -> timestamp is emulating gomatrixserverlib.FetchKeys:
|
|
// https://github.com/matrix-org/gomatrixserverlib/blob/f69539c86ea55d1e2cc76fd8e944e2d82d30397c/keyring.go#L95
|
|
// The timestamp should be the timestamp of the event that is being
|
|
// verified. We will not return keys from the cache that are not valid
|
|
// at this timestamp.
|
|
GetServerKey(request gomatrixserverlib.PublicKeyLookupRequest, timestamp gomatrixserverlib.Timestamp) (response gomatrixserverlib.PublicKeyLookupResult, ok bool)
|
|
|
|
// request -> result is emulating gomatrixserverlib.StoreKeys:
|
|
// https://github.com/matrix-org/gomatrixserverlib/blob/f69539c86ea55d1e2cc76fd8e944e2d82d30397c/keyring.go#L112
|
|
StoreServerKey(request gomatrixserverlib.PublicKeyLookupRequest, response gomatrixserverlib.PublicKeyLookupResult)
|
|
}
|
|
|
|
func (c Caches) GetServerKey(
|
|
request gomatrixserverlib.PublicKeyLookupRequest,
|
|
timestamp gomatrixserverlib.Timestamp,
|
|
) (gomatrixserverlib.PublicKeyLookupResult, bool) {
|
|
key := fmt.Sprintf("%s/%s", request.ServerName, request.KeyID)
|
|
val, found := c.ServerKeys.Get(key)
|
|
if found && !val.WasValidAt(timestamp, true) {
|
|
// The key wasn't valid at the requested timestamp so don't
|
|
// return it. The caller will have to work out what to do.
|
|
c.ServerKeys.Unset(key)
|
|
return gomatrixserverlib.PublicKeyLookupResult{}, false
|
|
}
|
|
return val, found
|
|
}
|
|
|
|
func (c Caches) StoreServerKey(
|
|
request gomatrixserverlib.PublicKeyLookupRequest,
|
|
response gomatrixserverlib.PublicKeyLookupResult,
|
|
) {
|
|
key := fmt.Sprintf("%s/%s", request.ServerName, request.KeyID)
|
|
c.ServerKeys.Set(key, response)
|
|
}
|