11 #include "memcached_io.h"
20 } memcached_storage_action
;
23 char *storage_op_string(memcached_storage_action verb
)
44 static inline memcached_return
memcached_send(memcached_st
*ptr
,
45 char *key
, size_t key_length
,
46 char *value
, size_t value_length
,
50 memcached_storage_action verb
)
56 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
57 unsigned int server_key
;
59 WATCHPOINT_ASSERT(!(value
== NULL
&& value_length
> 0));
60 WATCHPOINT_ASSERT(!(value
&& value_length
== 0));
63 return MEMCACHED_NO_KEY_PROVIDED
;
65 /* Leaving this WATCHPOINT_ASSERT in since only a library fubar could blow this */
67 if (!(ptr
->flags
& MEM_NO_BLOCK
) && ptr
->write_buffer_offset
!= 0)
71 if (ptr
->hosts
== NULL
|| ptr
->number_of_hosts
== 0)
72 return MEMCACHED_NO_SERVERS
;
74 server_key
= memcached_generate_hash(ptr
, key
, key_length
);
76 rc
= memcached_connect(ptr
, server_key
);
77 if (rc
!= MEMCACHED_SUCCESS
)
82 write_length
= snprintf(buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
,
83 "%s %.*s %u %llu %zu\r\n", storage_op_string(verb
),
84 (int)key_length
, key
, flags
,
85 (unsigned long long)expiration
, value_length
);
87 write_length
= snprintf(buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
,
88 "%s %.*s %u %llu %zu %llu\r\n", storage_op_string(verb
),
89 (int)key_length
, key
, flags
,
90 (unsigned long long)expiration
, value_length
,
91 (unsigned long long)cas
);
93 if (write_length
>= MEMCACHED_DEFAULT_COMMAND_SIZE
)
95 rc
= MEMCACHED_WRITE_FAILURE
;
100 We have to flush after sending the command. Memcached is not smart enough
101 to just keep reading from the socket :(
103 if ((sent_length
= memcached_io_write(ptr
, server_key
, buffer
, write_length
, 0)) == -1)
105 rc
= MEMCACHED_WRITE_FAILURE
;
109 if ((sent_length
= memcached_io_write(ptr
, server_key
, value
, value_length
, 0)) == -1)
111 rc
= MEMCACHED_WRITE_FAILURE
;
115 if ((ptr
->flags
& MEM_NO_BLOCK
) && verb
== SET_OP
)
120 if ((sent_length
= memcached_io_write(ptr
, server_key
, "\r\n", 2, to_write
)) == -1)
122 memcached_quit_server(ptr
, server_key
);
123 rc
= MEMCACHED_WRITE_FAILURE
;
127 if ((ptr
->flags
& MEM_NO_BLOCK
) && verb
== SET_OP
)
129 rc
= MEMCACHED_SUCCESS
;
130 memcached_server_response_increment(ptr
, server_key
);
134 rc
= memcached_response(ptr
, buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
, server_key
);
136 if (rc
== MEMCACHED_STORED
)
137 return MEMCACHED_SUCCESS
;
142 memcached_io_reset(ptr
, server_key
);
147 memcached_return
memcached_set(memcached_st
*ptr
, char *key
, size_t key_length
,
148 char *value
, size_t value_length
,
153 LIBMEMCACHED_MEMCACHED_SET_START();
154 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
155 expiration
, flags
, 0, SET_OP
);
156 LIBMEMCACHED_MEMCACHED_SET_END();
160 memcached_return
memcached_add(memcached_st
*ptr
,
161 char *key
, size_t key_length
,
162 char *value
, size_t value_length
,
167 LIBMEMCACHED_MEMCACHED_ADD_START();
168 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
169 expiration
, flags
, 0, ADD_OP
);
170 LIBMEMCACHED_MEMCACHED_ADD_END();
174 memcached_return
memcached_replace(memcached_st
*ptr
,
175 char *key
, size_t key_length
,
176 char *value
, size_t value_length
,
181 LIBMEMCACHED_MEMCACHED_REPLACE_START();
182 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
183 expiration
, flags
, 0, REPLACE_OP
);
184 LIBMEMCACHED_MEMCACHED_REPLACE_END();
188 memcached_return
memcached_prepend(memcached_st
*ptr
,
189 char *key
, size_t key_length
,
190 char *value
, size_t value_length
,
195 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
196 expiration
, flags
, 0, PREPEND_OP
);
200 memcached_return
memcached_append(memcached_st
*ptr
,
201 char *key
, size_t key_length
,
202 char *value
, size_t value_length
,
207 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
208 expiration
, flags
, 0, APPEND_OP
);
212 memcached_return
memcached_cas(memcached_st
*ptr
,
213 char *key
, size_t key_length
,
214 char *value
, size_t value_length
,
220 rc
= memcached_send(ptr
, key
, key_length
, value
, value_length
,
221 expiration
, flags
, cas
, APPEND_OP
);