-
Notifications
You must be signed in to change notification settings - Fork 3
/
cache.go
50 lines (39 loc) · 1.51 KB
/
cache.go
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
48
49
50
package test
import (
"github.com/alexfalkowski/go-service/cache/redis"
rem "github.com/alexfalkowski/go-service/cache/redis/telemetry/metrics"
"github.com/alexfalkowski/go-service/cache/ristretto"
rim "github.com/alexfalkowski/go-service/cache/ristretto/telemetry/metrics"
gr "github.com/alexfalkowski/go-service/redis"
"github.com/go-redis/cache/v8"
"go.opentelemetry.io/otel/metric"
"go.uber.org/fx"
"go.uber.org/zap"
)
// NewRedisCache for test.
func NewRedisCache(lc fx.Lifecycle, cfg *redis.Config, logger *zap.Logger, meter metric.Meter) (*cache.Cache, error) {
params := redis.OptionsParams{Client: NewRedisClient(lc, cfg, logger), Config: cfg, Marshaller: Marshaller, Compressor: Compressor}
opts, err := redis.NewOptions(params)
if err != nil {
return nil, err
}
cache := redis.NewCache(opts)
rem.Register(cache, Version, meter)
return cache, nil
}
// NewRedisClient for test.
func NewRedisClient(lc fx.Lifecycle, cfg *redis.Config, logger *zap.Logger) gr.Client {
tracer := NewTracer(lc)
client := redis.NewClient(redis.ClientParams{Lifecycle: lc, RingOptions: redis.NewRingOptions(cfg), Tracer: tracer, Logger: logger})
return client
}
// NewRistrettoCache for test.
func NewRistrettoCache(lc fx.Lifecycle, meter metric.Meter) ristretto.Cache {
cfg := &ristretto.Config{NumCounters: 1e7, MaxCost: 1 << 30, BufferItems: 64}
c, err := ristretto.NewCache(ristretto.CacheParams{Lifecycle: lc, Config: cfg, Version: Version})
if err != nil {
panic(err)
}
rim.Register(c, Version, meter)
return c
}