IOTestWriteService.cpp

/*
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/>.
*/



#include "config.h"

#include "IOTestWriteService.hpp"

#include <iostream>
#include <IOManager.hpp>
#include <DeviceConfigurator.hpp>
#include <StreamOperationHandlerMethod.hpp>
#include <Exception.hpp>
#include <cassert>

using namespace components::ioservice;

using mermaid::support::io::IOManager;
using mermaid::support::io::DeviceConfigurator;
using mermaid::support::io::StreamOperationHandlerMethod;
using mermaid::support::io::StreamOperationHandlerMethodBase;
using mermaid::support::errorhandling::Exception;

#define PIPE_NAME "/tmp/test_write_pipe"
#define FILE_NAME "test_file.txt"
#define CHUNK_SIZE 100

IOTestWriteService::IOTestWriteService (shared_ptr<ActiveObject> ao, shared_ptr<Entity> entity, shared_ptr<ServiceInstanceDescription> serviceInstanceDescription, shared_ptr<ServiceConfiguration> serviceConfiguration) : Service (ao, entity, serviceInstanceDescription, serviceConfiguration),
    device_ (IOManager::getInstance()->createDevice (PIPE_NAME, getActiveObject(), shared_ptr<DeviceConfigurator> (new DeviceConfigurator), 100)), file_ (FILE_NAME, fstream::in)
{
  assert (file_.is_open());
  assert (file_.good());
}; // IOTestWriteService()

void IOTestWriteService::initialize()
{
  // device_->enableWrite();
  
  char* buf = new char[CHUNK_SIZE];
  file_.read (buf, CHUNK_SIZE);
  
  // not supposed to happen with this test_file.txt
  if (file_.eof())
    croak();
    
  device_->write (file_.gcount(), buf, shared_ptr<StreamOperationHandlerMethodBase> (new StreamOperationHandlerMethod<IOTestWriteService> (this, &IOTestWriteService::handler)));
  
  delete[] buf;
}; // initialize

void IOTestWriteService::update()
{
}; // update

void IOTestWriteService::handler (size_t bytesWritten, size_t requestedBytes, const char* data, bool success)
{

  if (file_.eof())
    endTest();
    
  char *buf = new char[CHUNK_SIZE];
  
  file_.read (buf, CHUNK_SIZE);
  
  //printf("buf: %s\n", buf);
  device_->write (file_.gcount(), buf, shared_ptr<StreamOperationHandlerMethodBase> (new StreamOperationHandlerMethod<IOTestWriteService> (this, &IOTestWriteService::handler)));
  
  delete[] buf;
}

void IOTestWriteService::croak()
{
  // assert(0);
  exit (-1);
}

void IOTestWriteService::endTest()
{
  file_.close();
  device_->close();
  getActiveObject()->stop();
  IOManager::getInstance()->getIOThread()->getActiveObject()->stop();
}

IOTestWriteService::~IOTestWriteService()
{
}