X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fdo.c;fp=libmemcached%2Fdo.c;h=dcc09afb65d80829505667279246a079a805124f;hb=e233475c354a12f9f07adbd73b13fbf719b21367;hp=34c2c95352284bfb35a9fedcc7085104407be6bf;hpb=5173b8bb8848fd9c75ee1144b2d11a5c89ac95b4;p=awesomized%2Flibmemcached diff --git a/libmemcached/do.c b/libmemcached/do.c index 34c2c953..dcc09afb 100644 --- a/libmemcached/do.c +++ b/libmemcached/do.c @@ -39,9 +39,59 @@ memcached_return_t memcached_do(memcached_server_write_instance_st ptr, const vo sent_length= memcached_io_write(ptr, command, command_length, with_flush); if (sent_length == -1 || (size_t)sent_length != command_length) + { rc= MEMCACHED_WRITE_FAILURE; + } else if ((ptr->root->flags.no_reply) == 0) + { memcached_server_response_increment(ptr); + } + + return rc; +} + +memcached_return_t memcached_vdo(memcached_server_write_instance_st ptr, + const struct __write_vector_st *vector, size_t count, + bool with_flush) +{ + memcached_return_t rc; + ssize_t sent_length; + + WATCHPOINT_ASSERT(command_length); + WATCHPOINT_ASSERT(command); + + if ((rc= memcached_connect(ptr)) != MEMCACHED_SUCCESS) + { + WATCHPOINT_ERROR(rc); + return rc; + } + + /* + ** Since non buffering ops in UDP mode dont check to make sure they will fit + ** before they start writing, if there is any data in buffer, clear it out, + ** otherwise we might get a partial write. + **/ + if (ptr->type == MEMCACHED_CONNECTION_UDP && with_flush && ptr->write_buffer_offset > UDP_DATAGRAM_HEADER_LENGTH) + { + memcached_io_write(ptr, NULL, 0, true); + } + + sent_length= memcached_io_writev(ptr, vector, count, with_flush); + + size_t command_length= 0; + for (uint32_t x= 0; x < count; ++x, vector++) + { + command_length+= vector->length; + } + + if (sent_length == -1 || (size_t)sent_length != command_length) + { + rc= MEMCACHED_WRITE_FAILURE; + } + else if ((ptr->root->flags.no_reply) == 0) + { + memcached_server_response_increment(ptr); + } return rc; }