舊項目結(jié)束，新項目開始，今天開始閱讀新項目相關(guān)的代碼。公司的代碼中大量的運(yùn)用了工廠技術(shù)，不過很可惜是simple factory，這樣當(dāng)你新加入一種對象類型的時候都需要去工廠的實現(xiàn)代碼（實際上在舊代碼實現(xiàn)中還不止如此，乃至于公用父類也需要修改）。這是一件不太好的事情，于是我希望在自己的代碼中重新實現(xiàn)，以減少維護(hù)量。
我記得《C++對話系列》中有一些相關(guān)的討論，于是我找來查看。非常幸運(yùn)的，我發(fā)現(xiàn)這正是我需要的東西，于是幾乎照抄的把代碼實現(xiàn)了，代碼測試通過。為了符合我自己的閱讀習(xí)慣，我把其中的一個函數(shù)及變量的命名改動了一下：
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
{
//通用工廠類，根據(jù)傳入的id創(chuàng)建產(chǎn)品對象
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的變量，通常應(yīng)該置于Base.cpp文件結(jié)尾
//派生類
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的變量，通常應(yīng)該置于Derived.cpp文件結(jié)尾
//聲明一個全局函數(shù)（簡化操作，非必需）
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
請按任意鍵繼續(xù). . .