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; }
gearman_return_t _thread_packet_read(gearman_server_con_st *con) { gearman_return_t ret; while (1) { if (con->packet == NULL) { con->packet= gearman_server_packet_create(con->thread, true); if (con->packet == NULL) return GEARMAN_MEMORY_ALLOCATION_FAILURE; } (void)gearman_connection_recv(&(con->con), &(con->packet->packet), &ret, true); if (ret != GEARMAN_SUCCESS) { if (ret == GEARMAN_IO_WAIT) break; gearman_server_packet_free(con->packet, con->thread, true); con->packet= NULL; return ret; } gearman_log_debug(con->thread->gearman, "%15s:%5s Received %s", con->host == NULL ? "-" : con->host, con->port == NULL ? "-" : con->port, gearman_command_info_list[con->packet->packet.command].name); /* We read a complete packet. */ if (con->thread->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. */ ret= gearman_server_run_command(con, &(con->packet->packet)); gearman_packet_free(&(con->packet->packet)); gearman_server_packet_free(con->packet, con->thread, true); con->packet= NULL; if (ret != GEARMAN_SUCCESS) return ret; } } return GEARMAN_SUCCESS; }
gearman_return_t gearman_server_io_packet_add(gearman_server_con_st *con, bool take_data, enum gearman_magic_t magic, gearman_command_t command, const void *arg, ...) { gearman_server_packet_st *server_packet; va_list ap; size_t arg_size; gearman_return_t ret; server_packet= gearman_server_packet_create(con->thread, false); if (server_packet == NULL) return GEARMAN_MEMORY_ALLOCATION_FAILURE; if (gearman_packet_create(con->thread->gearman, &(server_packet->packet)) == NULL) { gearman_server_packet_free(server_packet, con->thread, false); return GEARMAN_MEMORY_ALLOCATION_FAILURE; } server_packet->packet.magic= magic; server_packet->packet.command= command; va_start(ap, arg); while (arg != NULL) { arg_size = va_arg(ap, size_t); ret= gearman_packet_create_arg(&(server_packet->packet), arg, arg_size); if (ret != GEARMAN_SUCCESS) { va_end(ap); gearman_packet_free(&(server_packet->packet)); gearman_server_packet_free(server_packet, con->thread, false); return ret; } arg = va_arg(ap, void *); } va_end(ap); ret= gearman_packet_pack_header(&(server_packet->packet)); if (ret != GEARMAN_SUCCESS) { gearman_packet_free(&(server_packet->packet)); gearman_server_packet_free(server_packet, con->thread, false); return ret; } if (take_data) server_packet->packet.options.free_data= true; (void) pthread_mutex_lock(&con->thread->lock); GEARMAN_FIFO_ADD(con->io_packet, server_packet,); (void) pthread_mutex_unlock(&con->thread->lock); gearman_server_con_io_add(con); return GEARMAN_SUCCESS; }