Message/src/MessageTypeData.cpp

   1 /*
   2  * This file is part of TouristP2PImpl
   3  *
   4  * Copyright (C) 2007,2008 NUST Institute of Information Technology
   5  * Author: Faisal Khan <faisal.khan at [niit.edu.pk | cern.ch ]>
   6  *
   7  * TouristP2PImpl is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * TouristP2PImpl is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with TouristP2PImpl; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20 */
  21 #include "Tourist/Message/MessageTypeData.h"
  22 #include "Tourist/Message/MsgCacheXferRequest.h"
  23 #include "Tourist/Message/MsgCacheXferReply.h"
  24 #include "Tourist/Message/MsgPeerAnnouncement.h"
  25 #include "Tourist/Message/MsgBootstrapRequest.h"
  26 #include "Tourist/Message/MsgBootstrapReply.h"
  27 #include "Tourist/Message/MsgMulticast.h"
  28 #include "Tourist/Util/Util.h"
  29 #include "Poco/Exception.h"
  30 #include "Poco/Instantiator.h"
  31
  32 using Poco::Instantiator;
  33
  34 namespace Tourist {
  35 namespace Message {
  36
  37 const int MessageTypeData::PEER_ANNOUNCE  = 0;
  38 const int MessageTypeData::CACHE_XFER_REQ = 1;
  39 const int MessageTypeData::CACHE_XFER_REP = 2;
  40 const int MessageTypeData::BOOTSTRAP_REQ = 3;
  41 const int MessageTypeData::BOOTSTRAP_REP = 4;
  42 const int MessageTypeData::MULTICAST = 5;
  43
  44 MessageTypeData::MessageTypeData()
  45 {
  46   registerBuiltIns();
  47 }
  48
  49 MessageTypeData::~MessageTypeData()
  50 {
  51
  52 }
  53
  54 int MessageTypeData::registerMessageObj(int type, MessageInstantiator *instance)
  55 {
  56   int status = 0;
  57   try
  58   {
  59     dynFactory.registerClass(Util::toString(type), instance);
  60   }
  61   catch (Poco::ExistsException &e)
  62   {
  63     status = -1;
  64   }
  65   return status;
  66 }
  67
  68 AbstractMessage* MessageTypeData::createInstance(int type)
  69 {
  70   AbstractMessage * instance = NULL;
  71   try
  72   {
  73     instance = dynFactory.createInstance(Util::toString(type));
  74   }
  75   catch (Poco::NotFoundException &e)
  76   {
  77     //TODO: throw log
  78   }
  79   return instance;
  80 }
  81
  82 void MessageTypeData::registerBuiltIns()
  83 {
  84   this -> registerMessageObj(PEER_ANNOUNCE,
  85       new Instantiator<MsgPeerAnnouncement, AbstractMessage>);
  86
  87   this -> registerMessageObj(CACHE_XFER_REQ,
  88       new Instantiator<MsgCacheXferRequest, AbstractMessage>);
  89
  90   this -> registerMessageObj(CACHE_XFER_REP,
  91         new Instantiator<MsgCacheXferReply, AbstractMessage>);
  92
  93   this -> registerMessageObj(BOOTSTRAP_REQ,
  94         new Instantiator<MsgBootstrapRequest, AbstractMessage>);
  95
  96   this -> registerMessageObj(BOOTSTRAP_REP,
  97         new Instantiator<MsgBootstrapReply, AbstractMessage>);
  98
  99
 100   this -> registerMessageObj(MULTICAST,
 101         new Instantiator<MsgMulticast, AbstractMessage>);
 102 }
 103
 104 } // namespace Message
 105 } // namespace Tourist