EntityRepository.cpp

Go to the documentation of this file.

/*
Copyright 2007, 2008, 2009, 2010, 2011 Instituto de Sistemas e Robotica, Instituto Superior Tecnico

This file is part of MeRMaID.

MeRMaID is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

MeRMaID 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 Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with MeRMaID.  If not, see <http://www.gnu.org/licenses/>.
*/



/**
 * @Filename EntityRepository.cpp
 * @Description EntityRepository implementation
 * @Status Work in Progress
 * @Version $Id: EntityRepository.cpp 1 2011-03-04 18:13:18Z jreis $
 * @Maintainer mafb
 */


#include "config.h"

#include "EntityRepository.hpp"
#include "EntityDescription.hpp"
#include "EntityDescriptionRepository.hpp"
#include "ServiceInterfaceDescription.hpp"
#include "ServiceTypeDescription.hpp"
#include "Service.hpp"

#include <Exception.hpp>

#include <sstream>

using namespace mermaid::support::service;
using mermaid::support::errorhandling::Exception;


void EntityRepository::addEntity (shared_ptr<Entity> entity)
{
  std::string entityName = entity->getEntityDescription()->getEntityName();
  
  if (entityMap_.find (entityName) != entityMap_.end()) {
    std::stringstream ss;
    ss << "EntityRepository::addEntity : repository already has an entity registered with entityName=\"" << entityName << "\"";
    throw Exception (ss.str());
  }
  
  entityMap_[entityName] = entity;
}; // addEntity()

shared_ptr<Entity> EntityRepository::getEntity (std::string entityName)
{
  std::map<std::string, shared_ptr<Entity> >::iterator it = entityMap_.find (entityName);
  
  if (it == entityMap_.end()) {
    std::stringstream ss;
    ss << "EntityRepository::getEntity : repository does not have an entity registered with entityName=\"" << entityName << "\"";
    throw Exception (ss.str());
  }
  
  return (*it).second;
}; // getEntity()

shared_ptr<Entity> EntityRepository::getOrCreateEntity (std::string entityName)
{

  std::map<std::string, shared_ptr<Entity> >::iterator it = entityMap_.find (entityName);
  
  if (it == entityMap_.end()) {
    shared_ptr<EntityDescription> entityDescription = EntityDescriptionRepository::getInstance()->getEntityDescription (entityName);
    shared_ptr<Entity> entity = shared_ptr<Entity> (new Entity (entityDescription));
    this->addEntity (entity);
    return entity;
  }
  
  return (*it).second;
}; // getOrCreateEntity()