后端 - redis 使用 lua 的时候，怎么导入外部库?

Question

redis 使用 lua 的时候，怎么导入外部库

def get_cookie(hkey: str) -> str | None:    from core.redis.models import client    # Use Lua scripting to get a random field from the hash set    lua_script = """        local os = require("os")            math.randomseed(os.time())        local fields = redis.call('HKEYS', KEYS[1])        if #fields == 0 then            return nil        end        local random_index = math.random(1, #fields)        local field = fields[random_index]        return random_index    """    return client.eval(lua_script, 1, hkey)

我要导入 os 重置时间种子

运行会报错

    result = context.run(func, *args)             ^^^^^^^^^^^^^^^^^^^^^^^^  File "/home/pon/code/work/vobile/vt/account_console/api.py", line 75, in get_cookie_by_parser_name    return {"cookie": get_cookie(hkey)}                      ^^^^^^^^^^^^^^^^  File "/home/pon/code/work/vobile/vt/account_console/api.py", line 45, in get_cookie    return client.eval(lua_script, 1, hkey)           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^  File "/home/pon/.local/share/virtualenvs/account_console-Tc5w3Cdm/lib/python3.11/site-packages/redis/client.py", line 3131, in eval    return self.execute_command('EVAL', script, numkeys, *keys_and_args)           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^  File "/home/pon/.local/share/virtualenvs/account_console-Tc5w3Cdm/lib/python3.11/site-packages/redis/client.py", line 901, in execute_command    return self.parse_response(conn, command_name, **options)           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^  File "/home/pon/.local/share/virtualenvs/account_console-Tc5w3Cdm/lib/python3.11/site-packages/redis/client.py", line 915, in parse_response    response = connection.read_response()               ^^^^^^^^^^^^^^^^^^^^^^^^^^  File "/home/pon/.local/share/virtualenvs/account_console-Tc5w3Cdm/lib/python3.11/site-packages/redis/connection.py", line 756, in read_response    raise responseredis.exceptions.ResponseError: user_script:2: Script attempted to access nonexistent global variable 'require' script: 1243ed7643e4b791767acb3602ad454e6b9098a3, on @user_script:2.

Answer 1

看起来你在 Lua 脚本中使用了 require，这是 Node.js 的模块加载方式，而不是 Lua 的。在 Lua 中，如果你想使用外部库，你需要使用 require 的方式来加载。

然而，Redis 的 Lua 脚本环境并没有提供 require 函数来加载外部库。因此，你不能直接在 Redis 的 Lua 脚本中导入外部库。

如果你需要在 Redis 的 Lua 脚本中使用一些 Lua 库的功能，你需要做的是将这些功能集成到你的 Lua 脚本中，而不是尝试导入外部库。

在你的特定情况中，如果你需要使用 os 库的 time 函数来设置随机种子，你可以直接在 Lua 脚本中使用 os.time()，而不需要使用 require。你的代码应该像这样：

math.randomseed(os.time())

请注意，这需要在你的 Lua 环境中已经定义了 os 库。如果你在执行 Lua 脚本时遇到了问题，可能是因为你的 Lua 环境没有正确配置。在这种情况下，你可能需要检查你的 Lua 环境配置，或者考虑使用一个已经正确配置了 os 库的 Lua 环境。