[C++技能提拔]类注册

最近在做AI信息在各个平台流转的框架设计，想要设计一种可以灵活扩展、不改变原有代码的框架，了解到了类注册。
详细需求是如许的：AI算法在客户本地电脑和云端都有部署，原先AI在这两个平台下的输出格式并不同一，且每个AI功能都有自己的输出格式，导致两个平台下的AI信息无法共享，带来了计算资源的浪费，管理起来也比力杂乱，因此需要一种模式将全部AI输出规范起来。
我的解决思路大概就是将全部AI信息都规范输出到同一个json里面，详细实现就是：定义一个基类，该基类有两个方法需要重载，一个是toJson用于序列化到json，一个是fromJson用于从json中解析，后续每个AI功能的输出都要继续该基类，最后用一个Manager类通过map数据结构管理全部的AI信息。
注册类机制模板实现：

思量到便于后续AI输出功能的灵活扩展，以及不改变原有代码，用到了类注册机制管理这些子类的创建，最后将类注册机制抽象为如下模板：

1. // pluginFactoryTemplate.h
2. #ifndef PLUGIN_FACTORY_TEMPLATE_H_
3. #define PLUGIN_FACTORY_TEMPLATE_H_
5. #include<map>
7. #define REGISTER_PLUGIN(Base,DerivedBase,pluginName) \
8.     static bool regPlugin_##pluginName=(PluginFatoryTemplate<Base>::register_plugin<DerivedBase>(pluginName),true);
10. #define REGISTER_PLUGIN(Base,DerivedBase) \
11.     static bool regPlugin_##DerivedBase=(PluginFatoryTemplate<Base>::register_plugin<DerivedBase>(#DerivedBase),true);
13. #define CREATE_PLUGIN(Base,pluginName) \
14.     PluginFatoryTemplate<Base>::create(pluginName)
16. template<typename Base>
17. class PluginFatoryTemplate{
18.     public:
19.     // 创建函数类型
20.     using CreateFunc=Base* (*)();
21.    
22.     // 注册插件
23.     template<typename DerivedBase>
24.     static void register_plugin(const std::string& pluginName){
25.         getRegistry()[pluginName]=[](){
26.             return (Base*) (new DerivedBase());
27.         };
28.     }
30.     // 创建插件实例
31.     static std::shared_ptr<Base> create(const std::string& pluginName){
32.         auto it=getRegistry().find(pluginName);
33.         if(it!=getRegistry().end()){
34.             return std::shared_ptr<Base>(it->second());
35.         }
36.         std::cout<<"Error:Can't find "<<pluginName<<std::endl;
37.         return nullptr;
38.     }
40.     private:
41.     static std::map<std::string,CreateFunc>& getRegistry(){
42.         static std::map<std::string,CreateFunc> registry;
43.         return registry;
44.     }
45. }
46. #endif

复制代码

使用：

定义基类： 

1. // BaseAIOutput.h
3. class BaseAIOutput{
4.     public:
5.     visual std::string toJson()=0;
6.     visual void fromJson(std::string json)=0;
7. }

复制代码

定义子类：

1. // AI1Output.h
2. #include"BaseAIOutput.h"
4. class AI1Output:public BaseAIOutput{
5.     public:
6.     virsual std::string toJson();
7.     virsual void fromJson(std::string json);
8. }

复制代码

1. // AI1Output.cpp
2. #include"AI1Output.h"
3. #include"pluginFatoryTemplate.h"
5. // 注册子类
6. REGISTER_PLUGIN(BaseAIOutput,AI1Output)
8. std::string AI1Output::toJson(){
9.     return "this is json";
10. }
12. void AI1Output::fromJson(std::string json){
13.     std::cout<<"from json"<<std::endl;
14. }

复制代码

 主函数：

1. // main.cpp
3. #include"BaseAIOutput.h"
4. #include"pluginFatoryTemplate.h"
6. void main(){
7.     // 创建子类对象
8.     auto ptrAI1Output=CREATE_PLUGIN(BaseAIOutput,"AI1Output");
9.    
10.     return 0;
11. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。