1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 #include <libmemcached/common.h>
39 static memcached_return_t
ascii_exist(memcached_st
*memc
, memcached_server_write_instance_st instance
, const char *key
, size_t key_length
)
41 libmemcached_io_vector_st vector
[]=
44 { memcached_literal_param("add ") },
45 { memcached_array_string(memc
->_namespace
), memcached_array_size(memc
->_namespace
) },
47 { memcached_literal_param(" 0") },
48 { memcached_literal_param(" 2678400") },
49 { memcached_literal_param(" 0") },
50 { memcached_literal_param("\r\n") },
51 { memcached_literal_param("\r\n") }
54 /* Send command header */
55 memcached_return_t rc
= memcached_vdo(instance
, vector
, 9, true);
56 if (rc
== MEMCACHED_SUCCESS
)
58 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
59 rc
= memcached_response(instance
, buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
, NULL
);
61 if (rc
== MEMCACHED_NOTSTORED
)
63 rc
= MEMCACHED_SUCCESS
;
66 if (rc
== MEMCACHED_STORED
)
68 rc
= MEMCACHED_NOTFOUND
;
72 if (rc
== MEMCACHED_WRITE_FAILURE
)
74 memcached_io_reset(instance
);
80 static memcached_return_t
binary_exist(memcached_st
*memc
, memcached_server_write_instance_st instance
, const char *key
, size_t key_length
)
82 protocol_binary_request_set request
= {};
83 size_t send_length
= sizeof(request
.bytes
);
85 request
.message
.header
.request
.magic
= PROTOCOL_BINARY_REQ
;
86 request
.message
.header
.request
.opcode
= PROTOCOL_BINARY_CMD_ADD
;
87 request
.message
.header
.request
.keylen
= htons((uint16_t)(key_length
+ memcached_array_size(memc
->_namespace
)));
88 request
.message
.header
.request
.datatype
= PROTOCOL_BINARY_RAW_BYTES
;
89 request
.message
.header
.request
.extlen
= 8;
90 request
.message
.body
.flags
= 0;
91 request
.message
.body
.expiration
= htonl(2678400);
93 request
.message
.header
.request
.bodylen
= htonl((uint32_t) (key_length
94 +memcached_array_size(memc
->_namespace
)
95 +request
.message
.header
.request
.extlen
));
97 libmemcached_io_vector_st vector
[]=
100 { request
.bytes
, send_length
},
101 { memcached_array_string(memc
->_namespace
), memcached_array_size(memc
->_namespace
) },
105 /* write the header */
106 memcached_return_t rc
;
107 if ((rc
= memcached_vdo(instance
, vector
, 4, true)) != MEMCACHED_SUCCESS
)
109 memcached_io_reset(instance
);
110 return (rc
== MEMCACHED_SUCCESS
) ? MEMCACHED_WRITE_FAILURE
: rc
;
113 rc
= memcached_response(instance
, NULL
, 0, NULL
);
115 if (rc
== MEMCACHED_SUCCESS
)
116 rc
= MEMCACHED_NOTFOUND
;
118 if (rc
== MEMCACHED_DATA_EXISTS
)
119 rc
= MEMCACHED_SUCCESS
;
124 memcached_return_t
memcached_exist(memcached_st
*memc
, const char *key
, size_t key_length
)
126 return memcached_exist_by_key(memc
, key
, key_length
, key
, key_length
);
129 memcached_return_t
memcached_exist_by_key(memcached_st
*memc
,
130 const char *group_key
, size_t group_key_length
,
131 const char *key
, size_t key_length
)
133 memcached_return_t rc
;
134 if (memcached_failed(rc
= initialize_query(memc
, true)))
139 if (memcached_is_udp(memc
))
141 return MEMCACHED_NOT_SUPPORTED
;
144 uint32_t server_key
= memcached_generate_hash_with_redistribution(memc
, group_key
, group_key_length
);
145 memcached_server_write_instance_st instance
;
146 instance
= memcached_server_instance_fetch(memc
, server_key
);
148 if (memc
->flags
.binary_protocol
)
150 return binary_exist(memc
, instance
, key
, key_length
);
154 return ascii_exist(memc
, instance
, key
, key_length
);