1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker All rights reserved.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
15 * * Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following disclaimer
17 * in the documentation and/or other materials provided with the
20 * * The names of its contributors may not be used to endorse or
21 * promote products derived from this software without specific prior
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include <libmemcached/common.h>
39 #include <libmemcached/memcached/protocol_binary.h>
41 static memcached_return_t
ascii_touch(memcached_instance_st
* instance
,
42 const char *key
, size_t key_length
,
45 char expiration_buffer
[MEMCACHED_MAXIMUM_INTEGER_DISPLAY_LENGTH
+1];
46 int expiration_buffer_length
= snprintf(expiration_buffer
, sizeof(expiration_buffer
), " %llu", (unsigned long long)expiration
);
47 if (size_t(expiration_buffer_length
) >= sizeof(expiration_buffer
) or expiration_buffer_length
< 0)
49 return memcached_set_error(*instance
, MEMCACHED_MEMORY_ALLOCATION_FAILURE
, MEMCACHED_AT
,
50 memcached_literal_param("snprintf(MEMCACHED_MAXIMUM_INTEGER_DISPLAY_LENGTH)"));
53 libmemcached_io_vector_st vector
[]=
56 { memcached_literal_param("touch ") },
57 { memcached_array_string(instance
->root
->_namespace
), memcached_array_size(instance
->root
->_namespace
) },
59 { expiration_buffer
, size_t(expiration_buffer_length
) },
60 { memcached_literal_param("\r\n") }
63 memcached_return_t rc
;
64 if (memcached_failed(rc
= memcached_vdo(instance
, vector
, 6, true)))
66 return memcached_set_error(*instance
, MEMCACHED_WRITE_FAILURE
, MEMCACHED_AT
);
72 static memcached_return_t
binary_touch(memcached_instance_st
* instance
,
73 const char *key
, size_t key_length
,
76 protocol_binary_request_touch request
= {}; //{.bytes= {0}};
78 initialize_binary_request(instance
, request
.message
.header
);
80 request
.message
.header
.request
.opcode
= PROTOCOL_BINARY_CMD_TOUCH
;
81 request
.message
.header
.request
.extlen
= 4;
82 request
.message
.header
.request
.keylen
= htons((uint16_t)(key_length
+memcached_array_size(instance
->root
->_namespace
)));
83 request
.message
.header
.request
.datatype
= PROTOCOL_BINARY_RAW_BYTES
;
84 request
.message
.header
.request
.bodylen
= htonl((uint32_t)(key_length
+memcached_array_size(instance
->root
->_namespace
) +request
.message
.header
.request
.extlen
));
85 request
.message
.body
.expiration
= htonl((uint32_t) expiration
);
87 libmemcached_io_vector_st vector
[]=
90 { request
.bytes
, sizeof(request
.bytes
) },
91 { memcached_array_string(instance
->root
->_namespace
), memcached_array_size(instance
->root
->_namespace
) },
95 memcached_return_t rc
;
96 if (memcached_failed(rc
= memcached_vdo(instance
, vector
, 4, true)))
98 return memcached_set_error(*instance
, MEMCACHED_WRITE_FAILURE
, MEMCACHED_AT
);
104 memcached_return_t
memcached_touch(memcached_st
*ptr
,
105 const char *key
, size_t key_length
,
108 return memcached_touch_by_key(ptr
, key
, key_length
, key
, key_length
, expiration
);
111 memcached_return_t
memcached_touch_by_key(memcached_st
*shell
,
112 const char *group_key
, size_t group_key_length
,
113 const char *key
, size_t key_length
,
116 Memcached
* ptr
= memcached2Memcached(shell
);
117 LIBMEMCACHED_MEMCACHED_TOUCH_START();
119 memcached_return_t rc
;
120 if (memcached_failed(rc
= initialize_query(ptr
, true)))
125 if (memcached_failed(rc
= memcached_key_test(*ptr
, (const char **)&key
, &key_length
, 1)))
127 return memcached_set_error(*ptr
, rc
, MEMCACHED_AT
);
130 uint32_t server_key
= memcached_generate_hash_with_redistribution(ptr
, group_key
, group_key_length
);
131 memcached_instance_st
* instance
= memcached_instance_fetch(ptr
, server_key
);
133 if (ptr
->flags
.binary_protocol
)
135 rc
= binary_touch(instance
, key
, key_length
, expiration
);
139 rc
= ascii_touch(instance
, key
, key_length
, expiration
);
142 if (memcached_failed(rc
))
144 return memcached_set_error(*instance
, rc
, MEMCACHED_AT
, memcached_literal_param("Error occcured while writing touch command to server"));
147 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
148 rc
= memcached_response(instance
, buffer
, sizeof(buffer
), NULL
);
150 if (rc
== MEMCACHED_SUCCESS
or rc
== MEMCACHED_NOTFOUND
)
155 return memcached_set_error(*instance
, rc
, MEMCACHED_AT
, memcached_literal_param("Error occcured while reading response"));