StreamWriter.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 "StreamWriter.hpp"
#include "IOManager.hpp"
#include <ace/Asynch_IO.h>
#include <ace/Message_Block.h>
#include <Exception.hpp>
#include <iostream>
#include <cassert>
#include <errno.h>

using namespace mermaid::support::io;

using mermaid::support::errorhandling::Exception;

StreamWriter::~StreamWriter()
{
}

StreamWriter::StreamWriter (Device* device) : device_ (device)
{
}

void StreamWriter::onWrite (size_t bytesWritten, size_t requestedBytes, const char* data, bool success, shared_ptr<StreamOperationHandlerMethodBase> streamOperationHandlerWrite)
{
  device_->callWriteHandler (bytesWritten, requestedBytes, data, success, streamOperationHandlerWrite);
}

int StreamWriter::getFileDescriptor() const
{
  return device_->getFileDescriptor();
}

void StreamWriter::open (ACE_Handler* streamHandler, ACE_Proactor* proactor)
{
  int ret = writeStream_.open (*streamHandler, device_->getFileDescriptor(), 0, proactor);
  if (ret != 0)
    throw Exception ("StreamWriter::open(): Error opening ACE_Asynch_Write_Stream.");
}

void StreamWriter::doWrite (size_t bytesToWrite, const char* data, shared_ptr<StreamOperationHandlerMethodBase> streamOperationHandlerWrite)
{
  ACE_Message_Block* mb = new ACE_Message_Block (bytesToWrite);
  mb->copy (data, bytesToWrite);
  
  // free data (duplicated in DeviceHandle::write)
  delete[] data;
  
  int ret = writeStream_.write (*mb, mb->length(), static_cast<void*> (new shared_ptr<StreamOperationHandlerMethodBase> (streamOperationHandlerWrite)));
  
  if (ret != 0) {
    std::cerr << "StreamWriter::doWrite: Error writing. errno: " << errno << std::endl;
    while (ret == -1)
      ret = writeStream_.write (*mb, mb->length());
      
    if (ret != 0) // ret != -1
      throw Exception ("StreamWriter::doWrite(): Error performing an asynch write operation.");
  }
}