Method load

src/mgr/PyModule.cc:339–461

}

int PyModule::load(PyThreadState *pMainThreadState)
{
  ceph_assert(pMainThreadState != nullptr);

  const auto subinterpreter_modules_opt = g_conf().get_val<std::string>(
    "mgr_subinterpreter_modules"
  );
  auto subinterpreter_modules = ceph::split(subinterpreter_modules_opt);
  use_main_interpreter = std::count(
    subinterpreter_modules.begin(), subinterpreter_modules.end(), "*"
  ) == 0 && std::count(
    subinterpreter_modules.begin(), subinterpreter_modules.end(), module_name
  ) == 0;

  if (use_main_interpreter) {
    // Use main interpreter
    derr << "Loading module " << module_name << " in main interpreter" << dendl;
    pMyThreadState.set(pMainThreadState);
  } else {
    // Configure sub-interpreter
    derr << "Loading module " << module_name << " in sub-interpreter" << dendl;
    SafeThreadState sts(pMainThreadState);
    Gil gil(sts);

    auto thread_state = Py_NewInterpreter();
    if (thread_state == nullptr) {
      derr << "Failed to create python sub-interpreter for '" << module_name << '"' << dendl;
      return -EINVAL;
    } else {
      pMyThreadState.set(thread_state);
    }
  }

  // Environment is all good, import the external module
  {
    Gil gil(pMyThreadState);

    int r;

    pPickleModule = PyImport_ImportModule("pickle");
    if (!pPickleModule) {
      derr << "Unable to load pickle" << dendl;
      return -EINVAL;
    }

    r = load_subclass_of("MgrModule", &pClass);
    if (r) {
      derr << "Class not found in module '" << module_name << "'" << dendl;
      return r;
    }

    r = load_commands();
    if (r != 0) {
      derr << "Missing or invalid COMMANDS attribute in module '"
          << module_name << "'" << dendl;
      error_string = "Missing or invalid COMMANDS attribute";
      return r;
    }