73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"github.com/go-redis/redis/v8"
|
||
|
|
"time"
|
||
|
|
"uauth/internal/interfaces"
|
||
|
|
)
|
||
|
|
|
||
|
|
type _redis struct {
|
||
|
|
client *redis.Client
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) Get(ctx context.Context, key string) ([]byte, error) {
|
||
|
|
result, err := r.client.Get(ctx, key).Result()
|
||
|
|
if err != nil {
|
||
|
|
if errors.Is(err, redis.Nil) {
|
||
|
|
return nil, ErrorKeyNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return []byte(result), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) GetScan(ctx context.Context, key string) interfaces.Scanner {
|
||
|
|
return newScanner(r.Get(ctx, key))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) GetEx(ctx context.Context, key string, duration time.Duration) ([]byte, error) {
|
||
|
|
result, err := r.client.GetEx(ctx, key, duration).Result()
|
||
|
|
if err != nil {
|
||
|
|
if errors.Is(err, redis.Nil) {
|
||
|
|
return nil, ErrorKeyNotFound
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return []byte(result), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) GetExScan(ctx context.Context, key string, duration time.Duration) interfaces.Scanner {
|
||
|
|
return newScanner(r.GetEx(ctx, key, duration))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) Set(ctx context.Context, key string, value any) error {
|
||
|
|
bs, err := handleValue(value)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = r.client.Set(ctx, key, bs, redis.KeepTTL).Result()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) SetEx(ctx context.Context, key string, value any, duration time.Duration) error {
|
||
|
|
bs, err := handleValue(value)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = r.client.SetEX(ctx, key, bs, duration).Result()
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *_redis) Del(ctx context.Context, keys ...string) error {
|
||
|
|
return r.client.Del(ctx, keys...).Err()
|
||
|
|
}
|