File size: 954 Bytes
1e11bce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from __future__ import annotations

import json
from datetime import UTC, datetime
from typing import Any

import redis


class RedisCache:
    def __init__(self, redis_url: str):
        self.redis_url = redis_url
        self.client = redis.from_url(redis_url, decode_responses=True) if redis_url else None

    def is_enabled(self) -> bool:
        return self.client is not None

    def set_json(self, key: str, payload: dict[str, Any], ttl_seconds: int = 300) -> None:
        if self.client is None:
            return
        serialized = json.dumps(
            {
                "payload": payload,
                "cached_at": datetime.now(UTC).isoformat(),
            }
        )
        self.client.setex(key, ttl_seconds, serialized)

    def get_json(self, key: str) -> dict[str, Any] | None:
        if self.client is None:
            return None
        value = self.client.get(key)
        return json.loads(value) if value else None