forked from NdoleStudio/httpsms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmemory_cache.go
More file actions
47 lines (38 loc) · 1.08 KB
/
memory_cache.go
File metadata and controls
47 lines (38 loc) · 1.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package cache
import (
"context"
"fmt"
"time"
"github.com/NdoleStudio/httpsms/pkg/telemetry"
"github.com/palantir/stacktrace"
ttlCache "github.com/patrickmn/go-cache"
)
// memoryCache is the Cache implementation in memory
type memoryCache struct {
tracer telemetry.Tracer
store *ttlCache.Cache
}
// NewMemoryCache creates a new instance of memoryCache
func NewMemoryCache(tracer telemetry.Tracer, store *ttlCache.Cache) Cache {
return &memoryCache{
tracer: tracer,
store: store,
}
}
// Get an item from the redis cache
func (cache *memoryCache) Get(ctx context.Context, key string) (value string, err error) {
ctx, span := cache.tracer.Start(ctx)
defer span.End()
response, ok := cache.store.Get(key)
if !ok {
return "", stacktrace.NewError(fmt.Sprintf("no item found in cache with key [%s]", key))
}
return response.(string), nil
}
// Set an item in the redis cache
func (cache *memoryCache) Set(ctx context.Context, key string, value string, ttl time.Duration) error {
ctx, span := cache.tracer.Start(ctx)
defer span.End()
cache.store.Set(key, value, ttl)
return nil
}