static gearmand_error_t _thread_packet_read(gearman_server_con_st *con)
{
while (1)
{
if (con->packet == NULL)
{
if (! (con->packet= gearman_server_packet_create(con->thread, true)))
{
return GEARMAN_MEMORY_ALLOCATION_FAILURE;
}
}
gearmand_error_t ret;
if (gearmand_failed(ret= gearman_io_recv(con, true)))
{
if (ret == GEARMAN_IO_WAIT)
{
break;
}
gearman_server_packet_free(con->packet, con->thread, true);
con->packet= NULL;
return ret;
}
gearmand_log_debug(GEARMAN_DEFAULT_LOG_PARAM,
"Received %s %s:%u",
gearman_command_info(con->packet->packet.command)->name,
con->_host == NULL ? "-" : con->_host,
con->_port == NULL ? "-" : con->_port);
/* We read a complete packet. */
if (Server->flags.threaded)
{
/* Multi-threaded, queue for the processing thread to run. */
gearman_server_proc_packet_add(con, con->packet);
con->packet= NULL;
}
else
{
/* Single threaded, run the command here. */
gearmand_error_t rc= gearman_server_run_command(con, &(con->packet->packet));
gearmand_packet_free(&(con->packet->packet));
gearman_server_packet_free(con->packet, con->thread, true);
con->packet= NULL;
if (gearmand_failed(rc))
{
return rc;
}
}
}
return GEARMAN_SUCCESS;
}
static gearmand_error_t _thread_packet_flush(gearman_server_con_st *con)
{
/* Check to see if we've already tried to avoid excessive system calls. */
if (con->con.events & POLLOUT)
{
return GEARMAN_IO_WAIT;
}
while (con->io_packet_list)
{
gearmand_error_t ret= gearman_io_send(con, &(con->io_packet_list->packet),
con->io_packet_list->next == NULL ? true : false);
if (gearmand_failed(ret))
{
return ret;
}
gearmand_log_debug(GEARMAN_DEFAULT_LOG_PARAM,
"Sent %s to %s:%d",
gearman_command_info(con->io_packet_list->packet.command)->name,
con->_host == NULL ? "-" : con->_host,
con->_port == NULL ? "-" : con->_port);
gearman_server_io_packet_remove(con);
}
/* Clear the POLLOUT flag. */
return gearmand_io_set_events(con, POLLIN);
}
static gearmand_error_t _con_add(gearmand_thread_st *thread,
gearmand_con_st *dcon)
{
gearmand_error_t ret= GEARMAN_SUCCESS;
dcon->server_con= gearman_server_con_add(&(thread->server_thread), dcon, &ret);
assert(dcon->server_con || ret != GEARMAN_SUCCESS);
assert(! dcon->server_con || ret == GEARMAN_SUCCESS);
if (dcon->server_con == NULL)
{
gearmand_sockfd_close(dcon->fd);
return ret;
}
if (dcon->add_fn)
{
ret= (*dcon->add_fn)(dcon->server_con);
if (gearmand_failed(ret))
{
gearman_server_con_free(dcon->server_con);
gearmand_sockfd_close(dcon->fd);
return ret;
}
}
GEARMAN_LIST_ADD(thread->dcon, dcon,)
return GEARMAN_SUCCESS;
}
