generic-library/poco
1
0
Fork 0
mirror of https://github.com/pocoproject/poco.git synced 2025-02-26 11:08:39 +01:00
Code Issues Projects Releases Wiki Activity

Updated SocketReactor sample with FIFOBuffer.

Browse Source
This commit is contained in:
Aleksandar Fabijanic 2012-05-30 02:56:32 +00:00
parent 1863c26cfc
commit 4bb6e0ff42
1 changed files with 53 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
Net/samples/EchoServer/src/EchoServer.cpp
View File

@ -40,6 +40,8 @@
#include "Poco/NObserver.h" #include "Poco/NObserver.h"
#include "Poco/Exception.h" #include "Poco/Exception.h"
#include "Poco/Thread.h" #include "Poco/Thread.h"
#include "Poco/FIFOBuffer.h"
#include "Poco/Delegate.h"
#include "Poco/Util/ServerApplication.h" #include "Poco/Util/ServerApplication.h"
#include "Poco/Util/Option.h" #include "Poco/Util/Option.h"
#include "Poco/Util/OptionSet.h" #include "Poco/Util/OptionSet.h"
@ -50,12 +52,15 @@
using Poco::Net::SocketReactor; using Poco::Net::SocketReactor;
using Poco::Net::SocketAcceptor; using Poco::Net::SocketAcceptor;
using Poco::Net::ReadableNotification; using Poco::Net::ReadableNotification;
using Poco::Net::WritableNotification;
using Poco::Net::ShutdownNotification; using Poco::Net::ShutdownNotification;
using Poco::Net::ServerSocket; using Poco::Net::ServerSocket;
using Poco::Net::StreamSocket; using Poco::Net::StreamSocket;
using Poco::NObserver; using Poco::NObserver;
using Poco::AutoPtr; using Poco::AutoPtr;
using Poco::Thread; using Poco::Thread;
using Poco::FIFOBuffer;
using Poco::delegate;
using Poco::Util::ServerApplication; using Poco::Util::ServerApplication;
using Poco::Util::Application; using Poco::Util::Application;
using Poco::Util::Option; using Poco::Util::Option;
@ -64,18 +69,30 @@ using Poco::Util::HelpFormatter;
class EchoServiceHandler class EchoServiceHandler
/// I/O handler class. This class (un)registers handlers for I/O based on
/// data availability. To ensure non-blocking behavior and alleviate spurious
/// socket writability callback triggering when no data to be sent is available,
/// FIFO buffers are used. I/O FIFOBuffer sends notifications on transitions
/// from [1] non-readable (i.e. empty) to readable, [2] writable to non-writable
/// (i.e. full) and [3] non-writable (i.e. full) to writable.
/// Based on these notifications, the handler member functions react by
/// enabling/disabling respective reactor framework notifications.
{ {
public: public:
EchoServiceHandler(StreamSocket& socket, SocketReactor& reactor): EchoServiceHandler(StreamSocket& socket, SocketReactor& reactor):
_socket(socket), _socket(socket),
_reactor(reactor), _reactor(reactor),
_pBuffer(new char[BUFFER_SIZE]) _fifoIn(BUFFER_SIZE, true),
_fifoOut(BUFFER_SIZE, true)
{ {
Application& app = Application::instance(); Application& app = Application::instance();
app.logger().information("Connection from " + socket.peerAddress().toString()); app.logger().information("Connection from " + socket.peerAddress().toString());
_reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onReadable)); _reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onSocketReadable));
_reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, ShutdownNotification>(*this, &EchoServiceHandler::onShutdown)); _reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, ShutdownNotification>(*this, &EchoServiceHandler::onSocketShutdown));
_fifoOut.readable += delegate(this, &EchoServiceHandler::onFIFOOutReadable);
_fifoIn.writable += delegate(this, &EchoServiceHandler::onFIFOInWritable);
} }
~EchoServiceHandler() ~EchoServiceHandler()
@ -88,21 +105,43 @@ public:
catch (...) catch (...)
{ {
} }
_reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onReadable)); _reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onSocketReadable));
_reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, ShutdownNotification>(*this, &EchoServiceHandler::onShutdown)); _reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, WritableNotification>(*this, &EchoServiceHandler::onSocketWritable));
delete [] _pBuffer; _reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, ShutdownNotification>(*this, &EchoServiceHandler::onSocketShutdown));
_fifoOut.readable -= delegate(this, &EchoServiceHandler::onFIFOOutReadable);
_fifoIn.writable -= delegate(this, &EchoServiceHandler::onFIFOInWritable);
} }
void onReadable(const AutoPtr<ReadableNotification>& pNf) void onFIFOOutReadable(bool& b)
{ {
int n = _socket.receiveBytes(_pBuffer, BUFFER_SIZE); if (b)
if (n > 0) _reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, WritableNotification>(*this, &EchoServiceHandler::onSocketWritable));
_socket.sendBytes(_pBuffer, n);
else else
delete this; _reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, WritableNotification>(*this, &EchoServiceHandler::onSocketWritable));
} }
void onShutdown(const AutoPtr<ShutdownNotification>& pNf) void onFIFOInWritable(bool& b)
{
if (b)
_reactor.addEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onSocketReadable));
else
_reactor.removeEventHandler(_socket, NObserver<EchoServiceHandler, ReadableNotification>(*this, &EchoServiceHandler::onSocketReadable));
}
void onSocketReadable(const AutoPtr<ReadableNotification>& pNf)
{
// receive bytes and transfer(echo) them to the output FIFO buffer
int len = _socket.receiveBytes(_fifoIn);
_fifoIn.drain(_fifoOut.write(_fifoIn.buffer(), len));
}
void onSocketWritable(const AutoPtr<WritableNotification>& pNf)
{
_socket.sendBytes(_fifoOut);
}
void onSocketShutdown(const AutoPtr<ShutdownNotification>& pNf)
{ {
delete this; delete this;
} }
@ -115,7 +154,8 @@ private:
StreamSocket _socket; StreamSocket _socket;
SocketReactor& _reactor; SocketReactor& _reactor;
char* _pBuffer; FIFOBuffer _fifoIn;
FIFOBuffer _fifoOut;
}; };