devel/libmdw/trunk/libmdw/src/LineBufferProtocol.cpp

/***************************************************************************
 *   Copyright (C) 2010 by Jean-Baptiste Wons   *
 *   wonsjb@gmail.com   *
 *                                                                         *
 *   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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "LineBufferProtocol.h"

namespace mdw
{

LineBufferProtocol::LineBufferProtocol (StreamServer &server) : _server (server)
{
}


LineBufferProtocol::~LineBufferProtocol()
{
}
void LineBufferProtocol::visitRead (char *buffer, int size, StreamInfo &info)
{
    std::string &currentString = _buffers[info.getStreamId()];
    int newStart = 0;
    if (currentString.empty())
    {
        // hum, we still have a chance to not copy
        for (int i = 0; i < size; ++i)
        {
            if (buffer[i] == '\n')
            {
                if (_currentSizeVisitor != 0 )
                    _currentSizeVisitor->visitRead(buffer + newStart, i - newStart + 1, 0, i - newStart + 1, info);
                if (_currentVisitor != 0 )
                    _currentVisitor->visitRead(buffer + newStart, i - newStart + 1, info);
                newStart = i+1;
            }
        }
    }
    for (int i = newStart; i < size; ++i)
    {
        currentString.push_back(buffer[i]);
        if (buffer[i] == '\n')
        {
            if (_currentSizeVisitor != 0)
                _currentSizeVisitor->visitRead((char*)currentString.data(), currentString.size(), 0, currentString.size(), info);
            if (_currentVisitor != 0)
                _currentVisitor->visitRead((char*)currentString.data(), currentString.size(), info);
            currentString.clear();
        }
    }
}
void LineBufferProtocol::visitConnect (StreamInfo &info)
{
    if (_currentVisitor != 0) _currentVisitor->visitConnect(info);
    if (_currentSizeVisitor != 0) _currentSizeVisitor->visitConnect(info);
}
void LineBufferProtocol::visitDisconnect (StreamInfo &info)
{
    if (_currentVisitor != 0) _currentVisitor->visitDisconnect(info);
    if (_currentSizeVisitor != 0) _currentSizeVisitor->visitDisconnect(info);
}

void LineBufferProtocol::accept (StreamSizeServerVisitor &visitor, int timeOutSec, int timeOutMicro)
{
    _currentSizeVisitor = &visitor;
    _currentVisitor = 0;
    _server.accept(*this, timeOutSec, timeOutMicro);
}

void LineBufferProtocol::accept (StreamServerVisitor &visitor, int timeOutSec, int timeOutMicro)
{
    _currentVisitor = &visitor;
    _currentSizeVisitor = 0;
    _server.accept(*this, timeOutSec, timeOutMicro);
}

void LineBufferProtocol::write (const char* buffer, int size, int streamId)
{
    _server.write(buffer, size, streamId);
}
Stream &LineBufferProtocol::outStream (int streamId)
{
    return _server.outStream(streamId);
}
void LineBufferProtocol::close (int streamId)
{
    _server.close(streamId);
}

}

---