redis-om-python/aredis_om/checks.py

30 lines
793 B
Python

from functools import lru_cache
from typing import List
from aredis_om.connections import get_redis_connection
@lru_cache(maxsize=None)
async def check_for_command(conn, cmd):
try:
cmd_info = await conn.execute_command("command", "info", cmd)
return True
except TypeError:
return False
@lru_cache(maxsize=None)
async def has_redis_json(conn=None):
if conn is None:
conn = get_redis_connection()
command_exists = await check_for_command(conn, "json.set")
return command_exists
@lru_cache(maxsize=None)
async def has_redisearch(conn=None):
if conn is None:
conn = get_redis_connection()
if has_redis_json(conn):
return True
command_exists = await check_for_command(conn, "ft.search")
return command_exists