# 客户端库，用于与 Redis 服务器进行交互
pip3 install redis
# 安装用于在 Django 中集成 Redis 的库
pip3 install django-redis
 

# 缓存配置
CACHES = {
    'default': {
        'BACKEND': 'django_redis.cache.RedisCache',
        'LOCATION': 'redis://192.168.1.42:6379/0',  # Redis 地址和数据库编号
        'OPTIONS': {
            'CLIENT_CLASS': 'django_redis.client.DefaultClient',
            # 'PASSWORD': 'your_password',  # 如果有密码
            'SOCKET_CONNECT_TIMEOUT': 5,  # 连接超时（秒）
        }
    }
}
 

"""
redis 缓存服务
"""
from django_redis import get_redis_connection
 
class RedisService:
    def __init__(self):
        self.conn = get_redis_connection("default")
 
    def set_value(self, key, value, timeout=None):
        self.conn.set(key, value, timeout)
 
    def get_value(self, key):
        return self.conn.get(key)
 
    def delete_value(self, key):
        self.conn.delete(key)
 
    def get_all_keys(self, pattern='*'):
        return self.conn.keys(pattern)
 

from rest_framework.decorators import action
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet
from core.service.redis_service import RedisService
 
 
class DemoRedisApi(ModelViewSet):
    """
    数据存入redis
    """
 
    @action(methods=['get'], detail=False, url_path='addValue', url_name='addValue')
    def addValue(self, request):
        redis_service = RedisService()
        key = 'python-redis'
        value = 'test_value'
        redis_service.set_value(key, value, timeout=3600)  # 设置 key 的过期时间为 1 小时
        return Response(
            {
                'success': True,
                'message': 'OK',
            }
        )
 
    """
    获取redis缓存
    """
 
    @action(methods=['get'], detail=False, url_path='getValue', url_name='getValue')
    def getValue(self, request):
        redis_service = RedisService()
        key = 'python-redis'
        value = redis_service.get_value(key)
        return Response(
            {
                'success': True,
                'message': value,
            }
        )
 
    """
    删除redis缓存
    """
 
    @action(methods=['get'], detail=False, url_path='deleteValue', url_name='deleteValue')
    def deleteValue(self, request):
        redis_service = RedisService()
        key = 'python-redis'
        value = redis_service.delete_value(key)
        return Response(
            {
                'success': True,
                'message': '删除成功',
            }
        )