Cyber RT是apollo的运行环境框架,提供了模块动态加载机制。 本文基于apollo v6.0介绍Cyber RT的模块加载流程。
Cyber RT模块初探
apollo里面的很多功能都是基于Cyber RT的模块框架开发的,其生命周期由Cyber RT管理。
先介绍下两个容易混淆的概念,module(模块)和component(组件),在Cyber RT中,一个module可以由多个component组成。为避免混淆,后面我们都直接使用module和component来表述。
本文的介绍主要基于cyber/examples/common_component_example
中的例子,目录下的README.md
介绍了如何编译运行该例子。common_component_example
是一个简单的module,仅包含一个component CommonComponentSample
, 定义在common_component_example.cc
中。初始化的时候Init
函数会被框架调用,Proc
是消息回调函数,客户端发送的消息到达的时候会被框架自动调用。
细心的读者可能会发现,common_component_example.cc
中并没有main
函数。没错,它并不是一个完整的可执行程序,编译出来的成果物是动态库libcommon_component_example.so
。 通过下面的命令可以启动该动态库:
mainbooard -d cyber/examples/common_component_example/common.dag
Dag文件是module的配置文件,由protocolbuffers文件cyber/proto/dag_conf.proto
定义,每个module都要有一个配套的dag文件,启动本module时使用。 本例中的common.dag
是dag_conf.proto
的一个实例,内容如下,我们重点关注下面两点:
module_library指明了要加载的动态库。
这个module只有一个component,该component对应的class_name是CommonComponentSample
,和common_component_example.h
中的定义要一致。
# Define all coms in DAG streaming.
module_config {
module_library : "/apollo/bazel-bin/cyber/examples/common_component_example/libcommon_component_example.so"
components {
class_name : "CommonComponentSample"
config {
name : "common"
readers {
channel: "/apollo/prediction"
}
readers {
channel: "/apollo/test"
}
}
}
}
Cyber RT的模块加载机制主要分为两部分:
- 编译期进行模块注册
- 运行期加载模块并初始化
下面我们依次介绍下相关内容。
编译期工作
common_component_example.h
中调用CYBER_REGISTER_COMPONENT(CommonComponentSample)
注册component。
将宏CYBER_REGISTER_COMPONENT逐步展开后的代码如下,其中:
- 模板参数Derived的值为CommonComponentSample,即本component对应的类名
- 模板参数Base的值为apollo::cyber::ComponentBase,是CommonComponentSample的间接基类
- UniqueID是一个唯一的整数,由编译期宏__COUNTER__实现
#define CLASS_LOADER_REGISTER_CLASS_INTERNAL(Derived, Base, UniqueID)
namespace {
struct ProxyType##UniqueID {
ProxyType##UniqueID() {
apollo::cyber::class_loader::utility::RegisterClass<Derived, Base>(
#Derived, #Base);
}
};
static ProxyType##UniqueID g_register_class_##UniqueID;
}
上面的宏定义了一个名为ProxyType##UniqueID的结构体,UniqueID保证了该结构体是全局唯一的,然后用这个结构体定义了一个静态全局变量(静态全局变量的妙用我们稍后再做介绍)。
运行期工作
运行期主要负责加载动态库libcommon_component_example.so
,然后创建类CommonComponentSample
的实例对象并初始化。
可执行程序mainboard通过dag文件加载对应的component,入口为cyber/mainboard/mainboard.cc
中的main
函数。类ModuleController负责加载动态库并初始化各component的实例。
函数ModuleController::LoadModule(const std::string& path)
先将protocolbuffers文件common.dag
序列化为DagConfig对象,然后调用ModuleController::LoadModule(const DagConfig& dag_config)
,该函数主要部分解释如下:
bool ModuleController::LoadModule(const DagConfig& dag_config) {
const std::string work_root = common::WorkRoot();
for (auto module_config : dag_config.module_config()) {
...
// 1. 加载动态库libcommon_component_example.so
class_loader_manager_.LoadLibrary(load_path);
// 2. 根据配置信息初始化各个component实例
for (auto& component : module_config.components()) {
const std::string& class_name = component.class_name();
std::shared_ptr<ComponentBase> base =
class_loader_manager_.CreateClassObj<ComponentBase>(class_name);
if (base == nullptr || !base->Initialize(component.config())) {
return false;
}
component_list_.emplace_back(std::move(base));
}
...
}
return true;
}
接下来分两部分讲解上面的1和2两个步骤。
动态库加载
动态库的加载入口为ClassLoaderManager::LoadLibrary(const std::string& library_path)
, 最终在下面的函数实现具体功能:
bool LoadLibrary(const std::string& library_path, ClassLoader* loader) {
...
SharedLibraryPtr shared_library = nullptr;
static std::recursive_mutex loader_mutex;
{
std::lock_guard<std::recursive_mutex> lck(loader_mutex);
// 设置动态库加载前的上下文信息,后面会用到
SetCurActiveClassLoader(loader);
SetCurLoadingLibraryName(library_path);
// 加载动态库
shared_library = SharedLibraryPtr(new SharedLibrary(library_path));
SetCurLoadingLibraryName("");
SetCurActiveClassLoader(nullptr);
}
...
}
SharedLibrary对象构造的时候通过SharedLibrary::Load(const std::string& path, int flags)
调用dlopen
加载了libcommon_component_example.so
。
还记得前面提到的全局静态变量g_register_class_##UniqueID
吗?当libcommon_component_example.so
被加载的时候(dlopen
返回前),它是要被初始化的,也就是它的构造函数会被调用。
我们看下这个构造函数主要做了些什么。
代码语言:javascript复制// 模板参数Derived的值为CommonComponentSample
// 模板参数Base的值为apollo::cyber::ComponentBase,是CommonComponentSample的间接基类
template <typename Derived, typename Base>
void RegisterClass(const std::string& class_name,
const std::string& base_class_name) {
//创建CommonComponentSample的类工厂
utility::AbstractClassFactory<Base>* new_class_factory_obj =
new utility::ClassFactory<Derived, Base>(class_name, base_class_name);
/*
设置类工厂的上下文信息:
在so加载前,LoadLibrary中已通过SetCurActiveClassLoader和SetCurLoadingLibraryName设置了上下文信息,
下面通过GetCurActiveClassLoader和GetCurLoadingLibraryName获取相关信息,
然后和对象new_class_factory_obj做绑定。
*/
new_class_factory_obj->AddOwnedClassLoader(GetCurActiveClassLoader());
new_class_factory_obj->SetRelativeLibraryPath(GetCurLoadingLibraryName());
GetClassFactoryMapMapMutex().lock();
ClassClassFactoryMap& factory_map =
GetClassFactoryMapByBaseClass(typeid(Base).name());
/*
下面的代码将类工厂new_class_factory_obj加到factory_map中,
map的key为类的名字CommonComponentSample, value为工厂对象new_class_factory_obj。
factory_map又维护在另一个map中,该map作为静态变量由函数GetClassFactoryMapMap维护,
它的key是Base的类名,即上面的typeid(Base).name()。
*/
factory_map[class_name] = new_class_factory_obj;
GetClassFactoryMapMapMutex().unlock();
}
从上面的代码可以看出,如果要获取CommonComponentSample对应的类工厂,
- 首先要以CommonComponentSample的基类apollo::cyber::ComponentBase的名字作为key从函数GetClassFactoryMapByBaseClass获取一个map
- 然后以类名CommonComponentSample作为key从这个map中获取对应的类工厂。
上面提到的两个map是嵌套的,定义如下:
代码语言:javascript复制//map的key为字符串CommonComponentSample,value为类工厂对象
using ClassClassFactoryMap =
std::map<std::string, utility::AbstractClassFactoryBase*>;
//map的key为CommonComponentSample的基类apollo::cyber::ComponentBase的名字,value为上面的map
using BaseToClassFactoryMapMap = std::map<std::string, ClassClassFactoryMap>;
当函数LoadLibrary返回时,CommonComponentSample对应的类工厂已经维护在上述的二维map结构中。至此,动态库已加载完毕,并形成了class_name、ClassLoader和ClassFactory的绑定关系。
Component 初始化
回到上文的ModuleController::LoadModule函数,下面的代码根据类名初始化Component实例:
代码语言:javascript复制// 2. 根据配置信息初始化module中各个component实例
for (auto& component : module_config.components()) {
const std::string& class_name = component.class_name();
// class_name即CommonComponentSample,该component创建完成后,放在component_list_中统一维护
std::shared_ptr<ComponentBase> base =
class_loader_manager_.CreateClassObj<ComponentBase>(class_name);
if (base == nullptr || !base->Initialize(component.config())) {
return false;
}
component_list_.emplace_back(std::move(base));
}
函数ClassLoaderManager::CreateClassObj
首先根据类名找到对应的ClassLoader对象,然后使用ClassLoader对象创建component实例
template <typename Base>
std::shared_ptr<Base> ClassLoaderManager::CreateClassObj(
const std::string& class_name) {
//获取所有的ClassLoader对象
std::vector<ClassLoader*> class_loaders = GetAllValidClassLoaders();
//根据类名查找对应的ClassLoader对象,然后用该对象创建component实例
for (auto class_loader : class_loaders) {
if (class_loader->IsClassValid<Base>(class_name)) {
return (class_loader->CreateClassObj<Base>(class_name));
}
}
AERROR << "Invalid class name: " << class_name;
return std::shared_ptr<Base>();
}
接下来我们看下ClassLoader是如何创建component实例的。
代码语言:javascript复制template <typename Base>
std::shared_ptr<Base> ClassLoader::CreateClassObj(
const std::string& class_name) {
...
Base* class_object = utility::CreateClassObj<Base>(class_name, this);
...
}
utility::CreateClassObj中首先根据类名查找对应的类工厂,然后由该工厂创建component实例对象。
代码语言:javascript复制template <typename Base>
Base* CreateClassObj(const std::string& class_name, ClassLoader* loader) {
GetClassFactoryMapMapMutex().lock();
//根据基类名称查找由<class_name, utility::AbstractClassFactoryBase*>组成的map
ClassClassFactoryMap& factoryMap =
GetClassFactoryMapByBaseClass(typeid(Base).name());
//查找class_name对应的类工厂,即CommonComponentSample对应的类工厂
AbstractClassFactory<Base>* factory = nullptr;
if (factoryMap.find(class_name) != factoryMap.end()) {
factory = dynamic_cast<utility::AbstractClassFactory<Base>*>(
factoryMap[class_name]);
}
GetClassFactoryMapMapMutex().unlock();
//由类工厂创建CommonComponentSample实例对象
Base* classobj = nullptr;
if (factory && factory->IsOwnedBy(loader)) {
classobj = factory->CreateObj();
}
return classobj;
}
补充说明:类工厂的定义如下,在前面提到的RegisterClass函数中实例化,模板参数ClassObject的值为CommonComponentSample。所以,下面的CreateObj函数中直接new了一个CommonComponentSample对象。
代码语言:javascript复制template <typename ClassObject, typename Base>
class ClassFactory : public AbstractClassFactory<Base> {
public:
ClassFactory(const std::string& class_name,
const std::string& base_class_name)
: AbstractClassFactory<Base>(class_name, base_class_name) {}
Base* CreateObj() const { return new ClassObject; }
};
至此,CommonComponentSample已完成实例化。
再回到前面的ModuleController::LoadModule函数,它调用ComponentBase::Initialize(虚函数)初始化component。
代码语言:javascript复制// class_name即CommonComponentSample
std::shared_ptr<ComponentBase> base =
class_loader_manager_.CreateClassObj<ComponentBase>(class_name);
if (base == nullptr || !base->Initialize(component.config())) {
return false;
}
由于如下的继承关系: CommonComponentSample --> 模板类Component --> ComponentBase
。Initialize函数最终执行的是模板类Component
中的Initialize
函数,该函数又调用了bool CommonComponentSample::Init()
,至此完成CommonComponentSample的初始化。