Function FunctionLoad

src/storage/scripting.cc:297–364

}

Status FunctionLoad(redis::Connection *conn, engine::Context *ctx, const std::string &script, bool need_to_store,
                    bool replace, [[maybe_unused]] std::string *lib_name) {
  std::string first_line, lua_code;
  if (auto pos = script.find('\n'); pos != std::string::npos) {
    first_line = script.substr(0, pos);
    lua_code = script.substr(pos + 1);
  } else {
    return {Status::NotOK, "Expect a Shebang statement in the first line"};
  }

  const auto libname = GET_OR_RET(ExtractLibNameFromShebang(first_line));

  if (FunctionIsLibExist(conn, ctx, libname, need_to_store)) {
    if (!replace) {
      return {Status::NotOK, "library already exists, please specify REPLACE to force load"};
    }
    auto s = FunctionDelete(*ctx, conn, libname);
    if (!s) return s;
  }

  auto srv = conn->GetServer();
  auto lua = conn->Owner()->Lua();

  ScriptRunCtx script_run_ctx;
  script_run_ctx.conn = conn;
  script_run_ctx.ctx = ctx;
  // in FunctionLoad(), the script MUST be read-only,
  // because this function can be called multiple times
  // for one script from multiple threads,
  // and it is dangerous and unexpected to allow writes here.
  script_run_ctx.flags = ScriptFlagType::kScriptNoWrites;

  SaveOnRegistry(lua, REGISTRY_SCRIPT_RUN_CTX_NAME, &script_run_ctx);

  lua_pushstring(lua, libname.c_str());
  lua_setglobal(lua, REDIS_FUNCTION_LIBNAME);
  auto libname_exit = MakeScopeExit([lua] {
    lua_pushnil(lua);
    lua_setglobal(lua, REDIS_FUNCTION_LIBNAME);
  });

  lua_pushboolean(lua, need_to_store);
  lua_setglobal(lua, REDIS_FUNCTION_NEEDSTORE);
  auto need_store_exit = MakeScopeExit([lua] {
    lua_pushnil(lua);
    lua_setglobal(lua, REDIS_FUNCTION_NEEDSTORE);
  });

  if (luaL_loadbuffer(lua, lua_code.data(), lua_code.size(), "@user_script")) {
    std::string err_msg = lua_tostring(lua, -1);
    lua_pop(lua, 1);
    return {Status::NotOK, "Error while compiling new function lib: " + err_msg};
  }

  if (lua_pcall(lua, 0, 0, 0)) {
    std::string err_msg = lua_tostring(lua, -1);
    lua_pop(lua, 1);
    return {Status::NotOK, "Error while running new function lib: " + err_msg};