c++11 template factory

自动注册工厂CSDN里已经有人解析过了,这里是学习过程中的总结、并详细解析。话不多说,先看代码:

oly_auto_factory.h

#pragma once
#include <functional>
#include <map>
#include <memory>
#include <string>

namespace oly {
/**
 * T 需要传入基类
 */
template <typename T>
class auto_factory {
public:
    /**
     * N 子类,这里会把对应子类添加到Map中
     */
    template <typename N>
    struct register_t {
        template <typename... Args>
        register_t(const std::string& key, Args... args)
        {
            auto_factory::instance()->_map.emplace(key, [=] { return new N(args...); });
        }
    };

    /**
     * 普通指针
     */
    static auto normal_ptr(const std::string& key) -> T*
    {
        return auto_factory::instance()->find(key);
    }

    /**
     * unique_ptr指针
     */
    static std::unique_ptr<T> unique_ptr(const std::string& key)
    {
        return std::unique_ptr<T>(normal_ptr(key));
    }

    /**
     * shared_ptr指针
     */
    static std::shared_ptr<T> shared_ptr(const std::string& key)
    {
        return std::shared_ptr<T>(normal_ptr(key));
    }

private:
    auto_factory(){};
    auto_factory(const auto_factory&) = delete;
    auto_factory(auto_factory&&)      = delete;
    auto_factory& operator=(const auto_factory&) = delete;

    /**
     * 单例模式
     */
    static auto instance() -> auto_factory<T>*
    {
        static auto_factory<T> ins;
        return &ins;
    }

    auto find(const std::string& key) -> T*
    {
        if (_map.find(key) == _map.end())
            throw std::invalid_argument("key is not exist!");

        return _map[key]();
    }
    std::map<std::string, std::function<T*()>> _map;
};
}  // namespace oly

下面我们来看具体应用,基类class B

#include <iostream>
#include "oly_auto_factory.h"

class B {
public:
    virtual void display() = 0;
};
/**
 * 注册基类
 */
#define AF_OBJECT oly::auto_factory<B>
/**
 * 注册子类,这里没有对继承关系进行判断,需要保证继承关系
 */
#define REGISTER_AF_OBJECT_CLASS(CLASS, ...) AF_OBJECT::register_t<CLASS> __##CLASS(#CLASS, ##__VA_ARGS__)

子类 class  FirstB; class SecondB ;class ThirdB 

class FirstB : public B {
public:
    void display() override
    {
        std::cout << __func__ << " FirstB\n";
    }
};
REGISTER_AF_OBJECT_CLASS(FirstB);

class SecondB : public B {
public:
    SecondB(const std::string name = "")
    {
        nameString = name;
    }

    void display() override
    {
        std::cout << __func__ << " SecondB\n";
        std::cout << nameString << "\n";
    }

private:
    std::string nameString;
};
REGISTER_AF_OBJECT_CLASS(SecondB, "Welcome");  //带参构造函数

class ThirdB : public B { 
public:
    ThirdB(int age, const std::string name = "")
    {
        nameString = name;
        ageInt     = age;
    }
    void display() override
    {
        std::cout << __func__ << " ThirdB\n";
        std::cout << nameString << ":" << ageInt << "\n";
    }

private:
    std::string nameString;
    int         ageInt;
};
REGISTER_AF_OBJECT_CLASS(ThirdB, 20, "Tom"); //带参构造函数

类下面的 REGISTER_AF_OBJECT_CLASS(FirstB); 是实现自动注册的关键。当我看到这个REGISTER_AF_OBJECT_CLASS();时一脸懵逼,函数怎么可以在函数体外执行???。

我们来分析一下,大家都知道应用#define 时替换原则,我们替换一下

REGISTER_AF_OBJECT_CLASS(FirstB); ==》 AF_OBJECT::register_t<FirstB> __FirstB('FirstB", ##__VA_ARGS__);

呵,这货就是一个变量。而我们在变量的构造函数里面把他放入std::map<std::string, std::function<T*()>> _map;Map中!

int main()
{
    AF_OBJECT::normal_ptr("SecondB")->display();
    AF_OBJECT::shared_ptr("ThirdB")->display();
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章