/****************************************************************************** * Icinga 2 * * Copyright (C) 2012 Icinga Development Team (http://www.icinga.org/) * * * * This program is free software; you can redistribute it and/or * * modify it under the terms of the GNU General Public License * * as published by the Free Software Foundation; either version 2 * * of the License, or (at your option) any later version. * * * * This program is distributed in the hope that it will be useful, * * but WITHOUT ANY WARRANTY; without even the implied warranty of * * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * * GNU General Public License for more details. * * * * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the Free Software Foundation * * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. * ******************************************************************************/ #ifndef DBTYPE_H #define DBTYPE_H #include "base/object.h" #include "base/registry.h" #include namespace icinga { class DbObject; /** * A database object type. * * @ingroup ido */ class DbType : public Object { public: DECLARE_PTR_TYPEDEFS(DbType); typedef boost::function (const boost::shared_ptr&, const String&, const String&)> ObjectFactory; typedef std::map TypeMap; typedef std::map, boost::shared_ptr, pair_string_iless> ObjectMap; DbType(const String& name, const String& table, long tid, const String& idcolumn, const ObjectFactory& factory); String GetName(void) const; String GetTable(void) const; long GetTypeID(void) const; String GetIDColumn(void) const; static void RegisterType(const DbType::Ptr& type); static DbType::Ptr GetByName(const String& name); static DbType::Ptr GetByID(long tid); boost::shared_ptr GetOrCreateObjectByName(const String& name1, const String& name2); private: String m_Name; String m_Table; long m_TypeID; String m_IDColumn; ObjectFactory m_ObjectFactory; static boost::mutex m_StaticMutex; static TypeMap& GetTypes(void); ObjectMap& GetObjects(void); }; /** * A registry for DbType objects. * * @ingroup ido */ class DbTypeRegistry : public Registry { }; /** * Helper class for registering DynamicObject implementation classes. * * @ingroup ido */ class RegisterDbTypeHelper { public: RegisterDbTypeHelper(const String& name, const String& table, long tid, const String& idcolumn, const DbType::ObjectFactory& factory) { DbType::Ptr dbtype = boost::make_shared(name, table, tid, idcolumn, factory); DbType::RegisterType(dbtype); } }; /** * Factory function for DbObject-based classes. * * @ingroup ido */ template shared_ptr DbObjectFactory(const DbType::Ptr& type, const String& name1, const String& name2) { return boost::make_shared(type, name1, name2); } #define REGISTER_DBTYPE(name, table, tid, idcolumn, type) \ I2_EXPORT icinga::RegisterDbTypeHelper g_RegisterDBT_ ## name(#name, table, tid, idcolumn, DbObjectFactory); } #endif /* DBTYPE_H */