Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Off-chain] feat: in-memory query cache(s) #994

Draft
wants to merge 15 commits into
base: main
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion pkg/client/interface.go
Original file line number Diff line number Diff line change
Expand Up @@ -363,7 +363,7 @@ type BankQueryClient interface {

// QueryCache is a key/value store style interface for a cache of a single type.
// It is intended to be used to cache query responses (or derivatives thereof),
// where each key uniquely indexes its most recent value.
// where each key uniquely indexes the most recent query response.
type QueryCache[T any] interface {
Get(key string) (T, error)
Set(key string, value T) error
Expand Down
60 changes: 40 additions & 20 deletions pkg/client/query/cache/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import (
"time"
)

// EvictionPolicy determines how items are removed when number of keys in the cache reaches MaxKeys.
// EvictionPolicy determines which items are removed when number of keys in the cache reaches maxKeys.
type EvictionPolicy int64

const (
Expand All @@ -13,57 +13,77 @@ const (
LeastFrequentlyUsed
)

// queryCacheConfig is the configuration for query caches. It is intended to be
// configured via QueryCacheOptionFn functions.
// queryCacheConfig is the configuration for query caches.
// It is intended to be configured via QueryCacheOptionFn functions.
type queryCacheConfig struct {
// MaxKeys is the maximum number of items (key/value pairs) the cache can
// maxKeys is the maximum number of items (key/value pairs) the cache can
// hold before it starts evicting.
MaxKeys int64
EvictionPolicy EvictionPolicy
// TTL is how long items should remain in the cache. Items older than the TTL
// MAY not be evicted but SHOULD not be considered as cache hits.
TTL time.Duration
maxKeys int64

// historical determines whether the cache will cache a single value for each
// key (false), or whether it will cache a history of values for each key (true).
// TODO_CONSIDERATION:
//
// maxValueSize is the maximum cumulative size of all values in the cache.
// maxValueSize int64
// maxCacheSize is the maximum cumulative size of all keys AND values in the cache.
// maxCacheSize int64

// evictionPolicy determines which items are removed when number of keys in the cache reaches maxKeys.
evictionPolicy EvictionPolicy
// ttl is how long items should remain in the cache. Items older than the ttl
// MAY NOT be evicted immediately, but are NEVER considered as cache hits.
ttl time.Duration
// historical determines whether each key will point to a single values (false)
// or a history (i.e. reverse chronological list) of values (true).
historical bool
// pruneOlderThan is the number of past blocks for which to keep historical
// numHistoricalValues is the number of past blocks for which to keep historical
// values. If 0, no historical pruning is performed. It only applies when
// historical is true.
pruneOlderThan int64
numHistoricalValues int64
}

// QueryCacheOptionFn is a function which receives a queryCacheConfig for configuration.
type QueryCacheOptionFn func(*queryCacheConfig)

// WithHistoricalMode enables historical caching with the given pruneOlderThan
// Validate ensures that the queryCacheConfig isn't configured with incompatible options.
func (cfg *queryCacheConfig) Validate() error {
switch cfg.evictionPolicy {
case FirstInFirstOut:
// TODO_IMPROVE: support LeastRecentlyUsed and LeastFrequentlyUsed policies.
default:
return ErrQueryCacheConfigValidation.Wrapf("eviction policy %d not imlemented", cfg.evictionPolicy)
}

return nil
bryanchriswhite marked this conversation as resolved.
Show resolved Hide resolved
}

// WithHistoricalMode enables historical caching with the given numHistoricalValues
// configuration; if 0, no historical pruning is performed.
func WithHistoricalMode(pruneOlderThan int64) QueryCacheOptionFn {
func WithHistoricalMode(numHistoricalBlocks int64) QueryCacheOptionFn {
bryanchriswhite marked this conversation as resolved.
Show resolved Hide resolved
return func(cfg *queryCacheConfig) {
cfg.historical = true
cfg.pruneOlderThan = pruneOlderThan
cfg.numHistoricalValues = numHistoricalBlocks
}
}

// WithMaxKeys sets the maximum number of distinct key/value pairs the cache will
Olshansk marked this conversation as resolved.
Show resolved Hide resolved
// hold before evicting according to the configured eviction policy.
func WithMaxKeys(maxKeys int64) QueryCacheOptionFn {
return func(cfg *queryCacheConfig) {
cfg.MaxKeys = maxKeys
cfg.maxKeys = maxKeys
}
}

// WithEvictionPolicy sets the eviction policy.
func WithEvictionPolicy(policy EvictionPolicy) QueryCacheOptionFn {
return func(cfg *queryCacheConfig) {
cfg.EvictionPolicy = policy
cfg.evictionPolicy = policy
}
}

// WithTTL sets the time-to-live for cached items. Items older than the TTL
bryanchriswhite marked this conversation as resolved.
Show resolved Hide resolved
// MAY not be evicted but SHOULD not be considered as cache hits.
// MAY NOT be evicted immediately, but are NEVER considered as cache hits.
func WithTTL(ttl time.Duration) QueryCacheOptionFn {
return func(cfg *queryCacheConfig) {
cfg.TTL = ttl
cfg.ttl = ttl
}
}
6 changes: 4 additions & 2 deletions pkg/client/query/cache/errors.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@ import "cosmossdk.io/errors"
const codesace = "client/query/cache"

var (
ErrCacheMiss = errors.Register(codesace, 1, "cache miss")
ErrHistoricalModeNotEnabled = errors.Register(codesace, 2, "historical mode not enabled")
ErrCacheMiss = errors.Register(codesace, 1, "cache miss")
ErrHistoricalModeNotEnabled = errors.Register(codesace, 2, "historical mode not enabled")
ErrQueryCacheConfigValidation = errors.Register(codesace, 3, "invalid query cache config")
ErrCacheInternal = errors.Register(codesace, 4, "cache internal error")
)
Loading
Loading