totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-19 22:15:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

add pkg

Browse Source
This commit is contained in:
Asim Aslam
2021-12-02 18:25:52 +00:00
parent bf455d169e
commit a29ccf37a0
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
pkg/cache/cache.go vendored
View File

@@ -19,7 +19,7 @@ import (
type Cache interface { type Cache interface {
// Context returns a tenant scoped Cache // Context returns a tenant scoped Cache
Context(ctx context.Context) Cache Context(ctx context.Context) Cache
Get(key string, val interface{}) error Get(key string, val interface{}) (time.Time, error)
Set(key string, val interface{}, expires time.Time) error Set(key string, val interface{}, expires time.Time) error
Delete(key string) error Delete(key string) error
Increment(key string, val int64) (int64, error) Increment(key string, val int64) (int64, error)
@@ -109,7 +109,7 @@ func (c *cache) Close() error {
return nil return nil
} }
func (c *cache) Get(key string, val interface{}) error { func (c *cache) Get(key string, val interface{}) (time.Time, error) {
k := c.Key(key) k := c.Key(key)
// try the LRU // try the LRU
@@ -121,11 +121,11 @@ func (c *cache) Get(key string, val interface{}) error {
if !i.expires.IsZero() && i.expires.Sub(time.Now()).Seconds() < 0 { if !i.expires.IsZero() && i.expires.Sub(time.Now()).Seconds() < 0 {
// remove it // remove it
c.LRU.Remove(k) c.LRU.Remove(k)
return ErrNotFound return time.Time{}, ErrNotFound
} }
// otherwise unmarshal and return it // otherwise unmarshal and return it
return json.Unmarshal(i.val, val) return i.expires, json.Unmarshal(i.val, val)
} }
logger.Infof("Cache miss for %v", k) logger.Infof("Cache miss for %v", k)
@@ -143,9 +143,9 @@ func (c *cache) Get(key string, val interface{}) error {
if err := json.Unmarshal(b, &i); err == nil { if err := json.Unmarshal(b, &i); err == nil {
if !i.expires.IsZero() && i.expires.Sub(time.Now()).Seconds() < 0 { if !i.expires.IsZero() && i.expires.Sub(time.Now()).Seconds() < 0 {
c.Disk.Erase(k) c.Disk.Erase(k)
return ErrNotFound return time.Time{}, ErrNotFound
} }
return json.Unmarshal(i.val, val) return i.expires, json.Unmarshal(i.val, val)
} }
} }
@@ -156,15 +156,15 @@ func (c *cache) Get(key string, val interface{}) error {
recs, err := c.Store.Read(k, store.ReadLimit(1)) recs, err := c.Store.Read(k, store.ReadLimit(1))
if err != nil && err == store.ErrNotFound { if err != nil && err == store.ErrNotFound {
return ErrNotFound return time.Time{}, ErrNotFound
} else if err != nil { } else if err != nil {
return err return time.Time{}, err
} }
if len(recs) == 0 { if len(recs) == 0 {
return ErrNotFound return time.Time{}, ErrNotFound
} }
if err := json.Unmarshal(recs[0].Value, val); err != nil { if err := json.Unmarshal(recs[0].Value, val); err != nil {
return err return time.Time{}, err
} }
// put it in the cache for future use // put it in the cache for future use
@@ -182,7 +182,7 @@ func (c *cache) Get(key string, val interface{}) error {
// put on disk // put on disk
c.Disk.Write(rec.Key, b) c.Disk.Write(rec.Key, b)
return nil return vi.expires, nil
} }
func (c *cache) Set(key string, val interface{}, expires time.Time) error { func (c *cache) Set(key string, val interface{}, expires time.Time) error {
@@ -234,7 +234,7 @@ func (c *cache) Increment(key string, value int64) (int64, error) {
defer c.Unlock() defer c.Unlock()
var val int64 var val int64
if err := c.Get(key, &val); err != nil && err != ErrNotFound { if _, err := c.Get(key, &val); err != nil && err != ErrNotFound {
return 0, err return 0, err
} }
val += value val += value
@@ -249,7 +249,7 @@ func (c *cache) Decrement(key string, value int64) (int64, error) {
defer c.Unlock() defer c.Unlock()
var val int64 var val int64
if err := c.Get(key, &val); err != nil && err != ErrNotFound { if _, err := c.Get(key, &val); err != nil && err != ErrNotFound {
return 0, err return 0, err
} }
val -= value val -= value
@@ -263,7 +263,7 @@ func Context(ctx context.Context) Cache {
return DefaultCache.Context(ctx) return DefaultCache.Context(ctx)
} }
func Get(key string, val interface{}) error { func Get(key string, val interface{}) (time.Time, error) {
return DefaultCache.Get(key, val) return DefaultCache.Get(key, val)
} }