舊項目結束,新項目開始,今天開始閱讀新項目相關的代碼。公司的代碼中大量的運用了工廠技術,不過很可惜是simple factory,這樣當你新加入一種對象類型的時候都需要去工廠的實現代碼(實際上在舊代碼實現中還不止如此,乃至于公用父類也需要修改)。這是一件不太好的事情,于是我希望在自己的代碼中重新實現,以減少維護量。
我記得《C++對話系列》中有一些相關的討論,于是我找來查看。非常幸運的,我發現這正是我需要的東西,于是幾乎照抄的把代碼實現了,代碼測試通過。為了符合我自己的閱讀習慣,我把其中的一個函數及變量的命名改動了一下:
factory.hpp
#pragma once
#pragma warning (disable:4786)
/********************************************************************
created: 2008-10-22
author: lixianmin
purpose: a generic factory method lib
reference: 《C++對話系列--Abstract Factory, Template Style》
Copyright (C) 2008 - All Rights Reserved
*********************************************************************/
#include <string>
#include <map>
namespace lib
{
//通用工廠類,根據傳入的id創建產品對象
template <typename BaseType, typename KeyType=std::string>
class factory
{
private:
typedef std::auto_ptr<BaseType> (*BaseCreateFunc)();
typedef std::map<KeyType, BaseCreateFunc> FuncRegistry;
public:
static factory<BaseType, KeyType>& get_instance() //static singleton,不適用于多線程
{
static factory<BaseType, KeyType> obj;
return obj;
}
std::auto_ptr<BaseType> create(const KeyType& id) const
{
std::auto_ptr<BaseType> obj;
FuncRegistry::const_iterator regEntry = _registry.find(id);
if (regEntry != _registry.end())
{
obj = regEntry->second();
}
return obj;
}
void _register_create_function(const KeyType& id, BaseCreateFunc func){_registry[id] = func;}
private:
factory(void){}
factory(const factory& other);
factory operator=(const factory& other);
private:
FuncRegistry _registry;
};
//類型(DerivedType)注冊類,只要在DerivedType類定義(DerivedType.cpp文件)的最后聲明一個 lib::factory_register<Base, Derived> reg(id);對象即可
template <typename BaseType, typename DerivedType, typename KeyType=std::string>
class factory_register
{
public:
factory_register(const KeyType& id)
{
factory<BaseType, KeyType>::get_instance()._register_create_function(id, _create_instance);
}
private:
static std::auto_ptr<BaseType> _create_instance(){return std::auto_ptr<BaseType>(new DerivedType);}
private:
friend class factory<BaseType, KeyType>;
};
}
測試代碼:
#include <iostream>
#include "factory.hpp"
//基類
class Base
{
public:
virtual void print(void)
{
std::cout<<"base class"<<std::endl;
}
};
lib::factory_register<Base, Base> reg1("base"); //注冊Base類型,定義一個file scope的變量,通常應該置于Base.cpp文件結尾
//派生類
class Derived:public Base
{
public:
virtual void print(void)
{
std::cout<<"derived class"<<std::endl;
}
};
lib::factory_register<Base, Derived> reg2("derived"); //注冊Derived類型,定義一個file scope的變量,通常應該置于Derived.cpp文件結尾
//聲明一個全局函數(簡化操作,非必需)
lib::factory<Base>& glb_GetFactory(void)
{
return lib::factory<Base>::get_instance();
}
int main(int argc, char* argv[])
{
std::auto_ptr<Base> base=glb_GetFactory().create("base");
if (base.get())
{
base->print();
}
std::auto_ptr<Base> derived=glb_GetFactory().create("derived");
if (derived.get())
{
derived->print();
}
system("pause");
return 0;
}
輸出:
base class
derived class
請按任意鍵繼續. . .