void AsyncMockStreamFactory::MockStream::write(asio::const_buffer buf,
StreamHandler&& writeHandler) {
stdx::unique_lock<stdx::mutex> lk(_mutex);
auto begin = asio::buffer_cast<const uint8_t*>(buf);
auto size = asio::buffer_size(buf);
_writeQueue.push({begin, begin + size});
// Suspend execution after data is written.
_defer_inlock(kBlockedAfterWrite,
[this, writeHandler, size]() {
_io_service->post(
[writeHandler, size] { writeHandler(std::error_code(), size); });
});
}
/** @brief check select pool and process client reqeusts
*
* @param pool select pool
* @return void
*/
void SelPool::check_clients()
{
int connfd;
vector<ClientConnection*>::iterator it = clients.begin();
while ( it != clients.end() )
{
ClientConnection *client = *it;
connfd = client->getFd();
printf("Handling %d\n", connfd);
/* Client Connection State Machine */
switch ( client->getState() )
{
case ClientConnection::Ready_ForRead:
{
printf("Client State: Ready_ForRead\n");
/* read ready client socket */
if (FD_ISSET(connfd, &read_set))
{
readHandler(client);
if (client->getState() == ClientConnection::Request_Parsed)
{
processHandler(client);
/* if the client is closed after processing */
if (client->getState() == ClientConnection::Closed)
{
Socket::Close_Socket(connfd);
FD_CLR(connfd, &read_set);
delete client;
clients.erase(it);
continue;
}
writeHandler(client);
}
}
break;
}
case ClientConnection::Request_Parsed: {
printf("Client State: Request_Parsed\n");
break;
}
case ClientConnection::Writing_Response:
{
printf("Client State: Writing_Response\n");
if (FD_ISSET(connfd, &write_set))
{
if (!client->getRequest()->isCGIRequest())
{
processHandler(client);
writeHandler(client);
}
else if (FD_ISSET(client->getResponse()->getCGIout(), &read_set))
{
/* CGI request : if CGIout is also ready for reading */
pipeHandler(client);
writeHandler(client);
}
}
break;
}
case ClientConnection::Done_Response: {
printf("Client State: Done_Response\n");
break;
}
default: {
break;
}
}
it++;
}
}
