static inline void memcached_mark_server_for_timeout(memcached_server_write_instance_st server)
{
if (server->state != MEMCACHED_SERVER_STATE_IN_TIMEOUT)
{
struct timeval next_time;
if (gettimeofday(&next_time, NULL) == 0)
{
server->next_retry= next_time.tv_sec +server->root->retry_timeout;
}
else
{
server->next_retry= 1; // Setting the value to 1 causes the timeout to occur immediatly
}
server->state= MEMCACHED_SERVER_STATE_IN_TIMEOUT;
if (server->server_failure_counter_query_id != server->root->query_id)
{
server->server_failure_counter++;
server->server_failure_counter_query_id= server->root->query_id;
}
set_last_disconnected_host(server);
}
}
void memcached_quit_server(memcached_server_st *ptr, bool io_death)
{
if (ptr->fd != -1)
{
if (io_death == false && ptr->type != MEMCACHED_CONNECTION_UDP && ptr->options.is_shutting_down == false)
{
memcached_return_t rc;
char buffer[MEMCACHED_MAX_BUFFER];
ptr->options.is_shutting_down= true;
if (ptr->root->flags.binary_protocol)
{
protocol_binary_request_quit request;
memset(&request, 0, sizeof(request));
request.message.header.request.magic = PROTOCOL_BINARY_REQ;
request.message.header.request.opcode = PROTOCOL_BINARY_CMD_QUIT;
request.message.header.request.datatype = PROTOCOL_BINARY_RAW_BYTES;
rc= memcached_do(ptr, request.bytes, sizeof(request.bytes), true);
}
else
{
rc= memcached_do(ptr, "quit\r\n", strlen("quit\r\n"), true);
}
WATCHPOINT_ASSERT(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_FETCH_NOTFINISHED);
(void)rc; /* Shut up ICC */
/* read until socket is closed, or there is an error
* closing the socket before all data is read
* results in server throwing away all data which is
* not read
*
* In .40 we began to only do this if we had been doing buffered
* requests of had replication enabled.
*/
if (ptr->root->flags.buffer_requests || ptr->root->number_of_replicas)
{
ssize_t nread;
while (memcached_io_read(ptr, buffer, sizeof(buffer)/sizeof(*buffer),
&nread) == MEMCACHED_SUCCESS);
}
/*
* memcached_io_read may call memcached_quit_server with io_death if
* it encounters problems, but we don't care about those occurences.
* The intention of that loop is to drain the data sent from the
* server to ensure that the server processed all of the data we
* sent to the server.
*/
ptr->server_failure_counter= 0;
}
memcached_io_close(ptr);
}
ptr->fd= -1;
ptr->io_bytes_sent= 0;
ptr->write_buffer_offset= (size_t) ((ptr->type == MEMCACHED_CONNECTION_UDP) ? UDP_DATAGRAM_HEADER_LENGTH : 0);
ptr->read_buffer_length= 0;
ptr->read_ptr= ptr->read_buffer;
ptr->options.is_shutting_down= false;
memcached_server_response_reset(ptr);
/* We reset the version so that if we end up talking to a different server */
/* we don't have stale server version information. */
ptr->major_version= ptr->minor_version= ptr->micro_version= UINT8_MAX;
if (io_death)
{
ptr->server_failure_counter++;
set_last_disconnected_host(ptr);
}
}
