void LocalSocketTest::doSelectOrPoll2(SelectPtr pAction)
{
Timespan timeout(100000);
SocketAddress sas1("/tmp/poco.server1.tcp.sock");
EchoServer echoServer1(sas1);
SocketAddress sac1("/tmp/poco.client1.tcp.sock");
StreamSocket ss1(sas1, &sac1);
SocketAddress sas2("/tmp/poco.server2.tcp.sock");
EchoServer echoServer2(sas2);
SocketAddress sac2("/tmp/poco.client2.tcp.sock");
StreamSocket ss2(sas2, &sac2);
Socket::SocketList readList;
Socket::SocketList writeList;
Socket::SocketList exceptList;
readList.push_back(ss1);
readList.push_back(ss2);
assert (pAction(readList, writeList, exceptList, timeout) == 0);
assert (readList.empty());
assert (writeList.empty());
assert (exceptList.empty());
ss1.sendBytes("hello", 5);
ss1.poll(timeout, Socket::SELECT_READ);
readList.push_back(ss1);
readList.push_back(ss2);
assert (pAction(readList, writeList, exceptList, timeout) == 1);
assert (readList.size() == 1);
assert (readList[0] == ss1);
assert (writeList.empty());
assert (exceptList.empty());
char buffer[256];
int n = ss1.receiveBytes(buffer, sizeof(buffer));
assert (n == 5);
readList.clear();
writeList.clear();
exceptList.clear();
writeList.push_back(ss1);
writeList.push_back(ss2);
assert (pAction(readList, writeList, exceptList, timeout) == 2);
assert (readList.empty());
assert (writeList.size() == 2);
assert (writeList[0] == ss1);
assert (writeList[1] == ss2);
assert (exceptList.empty());
ss1.close();
ss2.close();
}
void SocketTest::testSelect2()
{
Timespan timeout(100000);
EchoServer echoServer1;
EchoServer echoServer2;
StreamSocket ss1(SocketAddress("localhost", echoServer1.port()));
StreamSocket ss2(SocketAddress("localhost", echoServer2.port()));
Socket::SocketList readList;
Socket::SocketList writeList;
Socket::SocketList exceptList;
readList.push_back(ss1);
readList.push_back(ss2);
assert (Socket::select(readList, writeList, exceptList, timeout) == 0);
assert (readList.empty());
assert (writeList.empty());
assert (exceptList.empty());
ss1.sendBytes("hello", 5);
ss1.poll(timeout, Socket::SELECT_READ);
readList.push_back(ss1);
readList.push_back(ss2);
assert (Socket::select(readList, writeList, exceptList, timeout) == 1);
assert (readList.size() == 1);
assert (readList[0] == ss1);
assert (writeList.empty());
assert (exceptList.empty());
char buffer[256];
int n = ss1.receiveBytes(buffer, sizeof(buffer));
assert (n == 5);
readList.clear();
writeList.clear();
exceptList.clear();
writeList.push_back(ss1);
writeList.push_back(ss2);
assert (Socket::select(readList, writeList, exceptList, timeout) == 2);
assert (readList.empty());
assert (writeList.size() == 2);
assert (writeList[0] == ss1);
assert (writeList[1] == ss2);
assert (exceptList.empty());
ss1.close();
ss2.close();
}
void SocketReactor::run()
{
_pThread = Thread::current();
Socket::SocketList readable;
Socket::SocketList writable;
Socket::SocketList except;
while (!_stop)
{
try
{
readable.clear();
writable.clear();
except.clear();
int nSockets = 0;
{
FastMutex::ScopedLock lock(_mutex);
for (EventHandlerMap::iterator it = _handlers.begin(); it != _handlers.end(); ++it)
{
if (it->second->accepts(_pReadableNotification))
{
readable.push_back(it->first);
nSockets++;
}
if (it->second->accepts(_pWritableNotification))
{
writable.push_back(it->first);
nSockets++;
}
if (it->second->accepts(_pErrorNotification))
{
except.push_back(it->first);
nSockets++;
}
}
}
if (nSockets == 0)
{
onIdle();
Thread::trySleep(static_cast<long>(_timeout.totalMilliseconds()));
}
else if (Socket::select(readable, writable, except, _timeout))
{
onBusy();
for (Socket::SocketList::iterator it = readable.begin(); it != readable.end(); ++it)
dispatch(*it, _pReadableNotification);
for (Socket::SocketList::iterator it = writable.begin(); it != writable.end(); ++it)
dispatch(*it, _pWritableNotification);
for (Socket::SocketList::iterator it = except.begin(); it != except.end(); ++it)
dispatch(*it, _pErrorNotification);
}
else onTimeout();
}
catch (Exception& exc)
{
ErrorHandler::handle(exc);
}
catch (std::exception& exc)
{
ErrorHandler::handle(exc);
}
catch (...)
{
ErrorHandler::handle();
}
}
onShutdown();
}