mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-15 20:38:00 +00:00
cf7a906023
* fix: project by id loads project from view and from eventstore * fix: correct search key for role * feat(auth): my user changes * fix: improve error handling in change converters * fix: log-id
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package bigcache
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"reflect"
|
|
|
|
a_cache "github.com/allegro/bigcache"
|
|
"github.com/caos/logging"
|
|
"github.com/caos/zitadel/internal/errors"
|
|
)
|
|
|
|
type Bigcache struct {
|
|
cache *a_cache.BigCache
|
|
}
|
|
|
|
func NewBigcache(c *Config) (*Bigcache, error) {
|
|
cacheConfig := a_cache.DefaultConfig(c.CacheLifetime)
|
|
cacheConfig.HardMaxCacheSize = c.MaxCacheSizeInMB
|
|
cache, err := a_cache.NewBigCache(cacheConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Bigcache{
|
|
cache: cache,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Bigcache) Set(key string, object interface{}) error {
|
|
if key == "" || reflect.ValueOf(object).IsNil() {
|
|
return errors.ThrowInvalidArgument(nil, "BIGCA-du73s", "key or value should not be empty")
|
|
}
|
|
var b bytes.Buffer
|
|
enc := gob.NewEncoder(&b)
|
|
if err := enc.Encode(object); err != nil {
|
|
return errors.ThrowInvalidArgument(err, "BIGCA-RUyxI", "unable to encode object")
|
|
}
|
|
return c.cache.Set(key, b.Bytes())
|
|
}
|
|
|
|
func (c *Bigcache) Get(key string, ptrToObject interface{}) error {
|
|
if key == "" || reflect.ValueOf(ptrToObject).IsNil() {
|
|
return errors.ThrowInvalidArgument(nil, "BIGCA-dksoe", "key or value should not be empty")
|
|
}
|
|
value, err := c.cache.Get(key)
|
|
if err == a_cache.ErrEntryNotFound {
|
|
return errors.ThrowNotFound(err, "BIGCA-we32s", "not in cache")
|
|
}
|
|
if err != nil {
|
|
logging.Log("BIGCA-ftofbc").WithError(err).Info("read from cache failed")
|
|
return errors.ThrowInvalidArgument(err, "BIGCA-3idls", "error in reading from cache")
|
|
}
|
|
|
|
b := bytes.NewBuffer(value)
|
|
dec := gob.NewDecoder(b)
|
|
|
|
return dec.Decode(ptrToObject)
|
|
}
|
|
|
|
func (c *Bigcache) Delete(key string) error {
|
|
if key == "" {
|
|
return errors.ThrowInvalidArgument(nil, "BIGCA-clsi2", "key should not be empty")
|
|
}
|
|
return c.cache.Delete(key)
|
|
}
|