Changed the way that modules are detected to work with Redis Enterprise and OSS.
This commit is contained in:
parent
8a661c9f4c
commit
e9fbd56e4e
1 changed files with 10 additions and 9 deletions
|
@ -5,18 +5,19 @@ from aredis_om.connections import get_redis_connection
|
|||
|
||||
|
||||
@lru_cache(maxsize=None)
|
||||
async def get_modules(conn) -> List[str]:
|
||||
modules = await conn.execute_command("module", "list")
|
||||
return [m[1] for m in modules]
|
||||
|
||||
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()
|
||||
names = await get_modules(conn)
|
||||
return b"ReJSON" in names or "ReJSON" in names
|
||||
|
||||
command_exists = await check_for_command(conn, "json.set")
|
||||
return command_exists
|
||||
|
||||
@lru_cache(maxsize=None)
|
||||
async def has_redisearch(conn=None):
|
||||
|
@ -24,5 +25,5 @@ async def has_redisearch(conn=None):
|
|||
conn = get_redis_connection()
|
||||
if has_redis_json(conn):
|
||||
return True
|
||||
names = await get_modules(conn)
|
||||
return b"search" in names or "search" in names
|
||||
command_exists = await check_for_command(conn, "ft.search")
|
||||
return command_exists
|
||||
|
|
Loading…
Reference in a new issue