X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fmemcached.hpp;h=17c306e9a2a467ddecf6bca4140bb8839b97d6ad;hb=56bb95c23210147d45b792e4e494e995d9de76dc;hp=d093192db138bc2176cdf128a408977e01b0636c;hpb=498a29bcc235aa3bd8999d73095df8b8cbb10413;p=awesomized%2Flibmemcached diff --git a/libmemcached/memcached.hpp b/libmemcached/memcached.hpp index d093192d..17c306e9 100644 --- a/libmemcached/memcached.hpp +++ b/libmemcached/memcached.hpp @@ -20,6 +20,7 @@ #include +#include #include #include #include @@ -35,17 +36,51 @@ class Memcache { public: - Memcache() - : + Memcache() + : + servers_list(), memc(), + servers(NULL), result() { memcached_create(&memc); } - Memcache(memcached_st *clone) - : + Memcache(const std::string &in_servers_list) + : + servers_list(in_servers_list), memc(), + servers(NULL), + result() + { + memcached_create(&memc); + servers= memcached_servers_parse(servers_list.c_str()); + memcached_server_push(&memc, servers); + } + + Memcache(const std::string &hostname, + in_port_t port) + : + servers_list(), + memc(), + servers(NULL), + result() + { + memcached_create(&memc); + servers_list.append(hostname); + servers_list.append(":"); + std::ostringstream strsmt; + strsmt << port; + servers_list.append(strsmt.str()); + servers= memcached_servers_parse(servers_list.c_str()); + memcached_server_push(&memc, servers); + } + + Memcache(memcached_st *clone) + : + servers_list(), + memc(), + servers(NULL), result() { memcached_clone(&memc, clone); @@ -53,15 +88,31 @@ public: Memcache(const Memcache &rhs) : + servers_list(rhs.servers_list), memc(), + servers(NULL), result() { memcached_clone(&memc, const_cast(&rhs.getImpl())); + servers= memcached_servers_parse(servers_list.c_str()); + memcached_server_push(&memc, servers); + } + + Memcache &operator=(const Memcache &rhs) + { + if (this != &rhs) + { + memcached_clone(&memc, const_cast(&rhs.getImpl())); + servers= memcached_servers_parse(servers_list.c_str()); + memcached_server_push(&memc, servers); + } + return *this; } ~Memcache() { memcached_free(&memc); + memcached_server_list_free(servers); } /** @@ -83,15 +134,99 @@ public: /** * Return an error string for the given return structure. * - * @param[in] rc a memcached_return structure + * @param[in] rc a memcached_return_t structure * @return error string corresponding to given return code in the library. */ - const std::string getError(memcached_return rc) const + const std::string getError(memcached_return_t rc) const { /* first parameter to strerror is unused */ return memcached_strerror(NULL, rc); } + + bool setBehavior(memcached_behavior_t flag, uint64_t data) + { + memcached_return_t rc; + rc= memcached_behavior_set(&memc, flag, data); + return (rc == MEMCACHED_SUCCESS); + } + + uint64_t getBehavior(memcached_behavior_t flag) { + return memcached_behavior_get(&memc, flag); + } + + /** + * Return the string which contains the list of memcached servers being + * used. + * + * @return a std::string containing the list of memcached servers + */ + const std::string getServersList() const + { + return servers_list; + } + + /** + * Set the list of memcached servers to use. + * + * @param[in] in_servers_list list of servers + * @return true on success; false otherwise + */ + bool setServers(const std::string &in_servers_list) + { + servers_list.assign(in_servers_list); + servers= memcached_servers_parse(in_servers_list.c_str()); + memcached_server_push(&memc, servers); + return (servers == NULL); + } + + /** + * Add a server to the list of memcached servers to use. + * + * @param[in] server_name name of the server to add + * @param[in] port port number of server to add + * @return true on success; false otherwise + */ + bool addServer(const std::string &server_name, in_port_t port) + { + memcached_return_t rc; + std::ostringstream strstm; + servers_list.append(","); + servers_list.append(server_name); + servers_list.append(":"); + strstm << port; + servers_list.append(strstm.str()); + servers= memcached_server_list_append(servers, + server_name.c_str(), + port, + &rc); + memcached_server_push(&memc, servers); + return (rc == MEMCACHED_SUCCESS); + } + + /** + * Remove a server from the list of memcached servers to use. + * + * @param[in] server_name name of the server to remove + * @param[in] port port number of server to remove + * @return true on success; false otherwise + */ + bool removeServer(const std::string &server_name, in_port_t port) + { + std::string tmp_str; + std::ostringstream strstm; + tmp_str.append(","); + tmp_str.append(server_name); + tmp_str.append(":"); + strstm << port; + tmp_str.append(strstm.str()); + memcached_server_st *server= memcached_servers_parse(tmp_str.c_str()); + + memcached_return_t rc= memcached_server_remove(server); + + return (rc == MEMCACHED_SUCCESS); + } + /** * Fetches an individual value from the server. mget() must always * be called before using this method. @@ -100,13 +235,13 @@ public: * @param[out] ret_val store returned object in this vector * @return a memcached return structure */ - memcached_return fetch(std::string &key, + memcached_return_t fetch(std::string &key, std::vector &ret_val) { char ret_key[MEMCACHED_MAX_KEY]; size_t value_length= 0; size_t key_length= 0; - memcached_return rc; + memcached_return_t rc; uint32_t flags= 0; char *value= memcached_fetch(&memc, ret_key, &key_length, &value_length, &flags, &rc); @@ -115,7 +250,13 @@ public: ret_val.reserve(value_length); ret_val.assign(value, value + value_length); key.assign(ret_key); + free(value); + } + else if (value) + { + free(value); } + return rc; } @@ -127,11 +268,11 @@ public: * this vector * @return true on success; false otherwise */ - bool get(const std::string &key, + bool get(const std::string &key, std::vector &ret_val) throw (Error) { uint32_t flags= 0; - memcached_return rc; + memcached_return_t rc; size_t value_length= 0; if (key.empty()) @@ -144,6 +285,7 @@ public: { ret_val.reserve(value_length); ret_val.assign(value, value + value_length); + free(value); return true; } return false; @@ -152,7 +294,7 @@ public: /** * Fetches an individual from a server which is specified by * the master_key parameter that is used for determining which - * server an object was stored in if key partitioning was + * server an object was stored in if key partitioning was * used for storage. * * @param[in] master_key key that specifies server object is stored on @@ -161,12 +303,12 @@ public: * this vector * @return true on success; false otherwise */ - bool getByKey(const std::string &master_key, - const std::string &key, + bool getByKey(const std::string &master_key, + const std::string &key, std::vector &ret_val) throw(Error) { uint32_t flags= 0; - memcached_return rc; + memcached_return_t rc; size_t value_length= 0; if (master_key.empty() || key.empty()) @@ -181,6 +323,7 @@ public: { ret_val.reserve(value_length); ret_val.assign(value, value + value_length); + free(value); return true; } return false; @@ -216,12 +359,12 @@ public: } /* - * If the std::vector of keys is empty then we cannot + * If the std::vector of keys is empty then we cannot * call memcached_mget as we will get undefined behavior. */ if (! real_keys.empty()) { - memcached_return rc= memcached_mget(&memc, &real_keys[0], &key_len[0], + memcached_return_t rc= memcached_mget(&memc, &real_keys[0], &key_len[0], real_keys.size()); return (rc == MEMCACHED_SUCCESS); } @@ -249,7 +392,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_set(&memc, + memcached_return_t rc= memcached_set(&memc, key.c_str(), key.length(), &value[0], value.size(), expiration, flags); @@ -267,8 +410,8 @@ public: * @param[in] flags flags to store with the object * @return true on succcess; false otherwise */ - bool setByKey(const std::string &master_key, - const std::string &key, + bool setByKey(const std::string &master_key, + const std::string &key, const std::vector &value, time_t expiration, uint32_t flags) throw(Error) @@ -279,7 +422,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_set_by_key(&memc, master_key.c_str(), + memcached_return_t rc= memcached_set_by_key(&memc, master_key.c_str(), master_key.length(), key.c_str(), key.length(), &value[0], value.size(), @@ -348,9 +491,8 @@ public: retval= set(it->first, it->second, expiration, flags); if (retval == false) { - char err_buff[64]; - sprintf(err_buff, "There was an error setting the key %s", - it->first.c_str()); + std::string err_buff("There was an error setting the key "); + err_buff.append(it->first); throw(Error(err_buff, false)); } ++it; @@ -374,7 +516,7 @@ public: { throw(Error("the key supplied is empty!", false)); } - memcached_return rc= memcached_increment(&memc, key.c_str(), key.length(), + memcached_return_t rc= memcached_increment(&memc, key.c_str(), key.length(), offset, value); return (rc == MEMCACHED_SUCCESS); } @@ -396,7 +538,7 @@ public: { throw(Error("the key supplied is empty!", false)); } - memcached_return rc= memcached_decrement(&memc, key.c_str(), + memcached_return_t rc= memcached_decrement(&memc, key.c_str(), key.length(), offset, value); return (rc == MEMCACHED_SUCCESS); @@ -418,7 +560,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_add(&memc, key.c_str(), key.length(), + memcached_return_t rc= memcached_add(&memc, key.c_str(), key.length(), &value[0], value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); } @@ -433,22 +575,22 @@ public: * @param[in] value of object to add * @return true on success; false otherwise */ - bool addByKey(const std::string &master_key, - const std::string &key, + bool addByKey(const std::string &master_key, + const std::string &key, const std::vector &value) throw(Error) { if (master_key.empty() || - key.empty() || + key.empty() || value.empty()) { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_add_by_key(&memc, + memcached_return_t rc= memcached_add_by_key(&memc, master_key.c_str(), master_key.length(), key.c_str(), key.length(), - &value[0], + &value[0], value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); @@ -469,7 +611,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_replace(&memc, key.c_str(), key.length(), + memcached_return_t rc= memcached_replace(&memc, key.c_str(), key.length(), &value[0], value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); @@ -485,8 +627,8 @@ public: * @param[in[ value value to replace object with * @return true on success; false otherwise */ - bool replaceByKey(const std::string &master_key, - const std::string &key, + bool replaceByKey(const std::string &master_key, + const std::string &key, const std::vector &value) { if (master_key.empty() || @@ -495,13 +637,13 @@ public: { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_replace_by_key(&memc, - master_key.c_str(), + memcached_return_t rc= memcached_replace_by_key(&memc, + master_key.c_str(), master_key.length(), - key.c_str(), + key.c_str(), key.length(), - &value[0], - value.size(), + &value[0], + value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); } @@ -520,7 +662,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_prepend(&memc, key.c_str(), key.length(), + memcached_return_t rc= memcached_prepend(&memc, key.c_str(), key.length(), &value[0], value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); } @@ -535,8 +677,8 @@ public: * @param[in] value data to prepend to object's value * @return true on success; false otherwise */ - bool prependByKey(const std::string &master_key, - const std::string &key, + bool prependByKey(const std::string &master_key, + const std::string &key, const std::vector &value) throw(Error) { @@ -546,7 +688,7 @@ public: { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_prepend_by_key(&memc, + memcached_return_t rc= memcached_prepend_by_key(&memc, master_key.c_str(), master_key.length(), key.c_str(), @@ -572,7 +714,7 @@ public: { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_append(&memc, + memcached_return_t rc= memcached_append(&memc, key.c_str(), key.length(), &value[0], @@ -602,13 +744,13 @@ public: { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_append_by_key(&memc, - master_key.c_str(), + memcached_return_t rc= memcached_append_by_key(&memc, + master_key.c_str(), master_key.length(), - key.c_str(), + key.c_str(), key.length(), - &value[0], - value.size(), + &value[0], + value.size(), 0, 0); return (rc == MEMCACHED_SUCCESS); } @@ -621,16 +763,16 @@ public: * @param[in] value value to store for object in server * @param[in] cas_arg "cas" value */ - bool cas(const std::string &key, - const std::vector &value, + bool cas(const std::string &key, + const std::vector &value, uint64_t cas_arg) throw(Error) { if (key.empty() || value.empty()) { throw(Error("the key or value supplied is empty!", false)); } - memcached_return rc= memcached_cas(&memc, key.c_str(), key.length(), - &value[0], value.size(), + memcached_return_t rc= memcached_cas(&memc, key.c_str(), key.length(), + &value[0], value.size(), 0, 0, cas_arg); return (rc == MEMCACHED_SUCCESS); } @@ -645,9 +787,9 @@ public: * @param[in] value value to store for object in server * @param[in] cas_arg "cas" value */ - bool casByKey(const std::string &master_key, - const std::string &key, - const std::vector &value, + bool casByKey(const std::string &master_key, + const std::string &key, + const std::vector &value, uint64_t cas_arg) throw(Error) { if (master_key.empty() || @@ -656,12 +798,12 @@ public: { throw(Error("the master key, key or value supplied is empty!", false)); } - memcached_return rc= memcached_cas_by_key(&memc, - master_key.c_str(), + memcached_return_t rc= memcached_cas_by_key(&memc, + master_key.c_str(), master_key.length(), - key.c_str(), + key.c_str(), key.length(), - &value[0], + &value[0], value.size(), 0, 0, cas_arg); return (rc == MEMCACHED_SUCCESS); @@ -679,7 +821,7 @@ public: { throw(Error("the key supplied is empty!", false)); } - memcached_return rc= memcached_delete(&memc, key.c_str(), key.length(), 0); + memcached_return_t rc= memcached_delete(&memc, key.c_str(), key.length(), 0); return (rc == MEMCACHED_SUCCESS); } @@ -697,7 +839,7 @@ public: { throw(Error("the key supplied is empty!", false)); } - memcached_return rc= memcached_delete(&memc, + memcached_return_t rc= memcached_delete(&memc, key.c_str(), key.length(), expiration); @@ -705,7 +847,7 @@ public: } /** - * Delete an object from the server specified by the key given. + * Delete an object from the server specified by the key given. * * @param[in] master_key specifies server to remove object from * @param[in] key key of object to delete @@ -718,7 +860,7 @@ public: { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_delete_by_key(&memc, + memcached_return_t rc= memcached_delete_by_key(&memc, master_key.c_str(), master_key.length(), key.c_str(), @@ -728,14 +870,14 @@ public: } /** - * Delete an object from the server specified by the key given. + * Delete an object from the server specified by the key given. * * @param[in] master_key specifies server to remove object from * @param[in] key key of object to delete * @param[in] expiration time to delete the object after * @return true on success; false otherwise */ - bool removeByKey(const std::string &master_key, + bool removeByKey(const std::string &master_key, const std::string &key, time_t expiration) throw(Error) { @@ -743,11 +885,11 @@ public: { throw(Error("the master key or key supplied is empty!", false)); } - memcached_return rc= memcached_delete_by_key(&memc, - master_key.c_str(), + memcached_return_t rc= memcached_delete_by_key(&memc, + master_key.c_str(), master_key.length(), - key.c_str(), - key.length(), + key.c_str(), + key.length(), expiration); return (rc == MEMCACHED_SUCCESS); } @@ -761,7 +903,7 @@ public: */ bool flush(time_t expiration) { - memcached_return rc= memcached_flush(&memc, expiration); + memcached_return_t rc= memcached_flush(&memc, expiration); return (rc == MEMCACHED_SUCCESS); } @@ -775,11 +917,11 @@ public: * @param[in] num_of_callbacks number of callback functions * @return true on success; false otherwise */ - bool fetchExecute(memcached_execute_function *callback, + bool fetchExecute(memcached_execute_fn *callback, void *context, - unsigned int num_of_callbacks) + uint32_t num_of_callbacks) { - memcached_return rc= memcached_fetch_execute(&memc, + memcached_return_t rc= memcached_fetch_execute(&memc, callback, context, num_of_callbacks); @@ -797,9 +939,65 @@ public: return version; } + /** + * Retrieve memcached statistics. Populate a std::map with the retrieved + * stats. Each server will map to another std::map of the key:value stats. + * + * @param[out] stats_map a std::map to be populated with the memcached + * stats + * @return true on success; false otherwise + */ + bool getStats(std::map< std::string, std::map > + &stats_map) + { + memcached_return_t rc; + memcached_stat_st *stats= memcached_stat(&memc, NULL, &rc); + + if (rc != MEMCACHED_SUCCESS && + rc != MEMCACHED_SOME_ERRORS) + { + return false; + } + + uint32_t server_count= memcached_server_count(&memc); + + /* + * For each memcached server, construct a std::map for its stats and add + * it to the std::map of overall stats. + */ + for (uint32_t x= 0; x < server_count; x++) + { + std::ostringstream strstm; + std::string server_name(memcached_server_name(&memc, servers[x])); + server_name.append(":"); + strstm << memcached_server_port(&memc, servers[x]); + server_name.append(strstm.str()); + + std::map server_stats; + char **list= NULL; + char **ptr= NULL; + + list= memcached_stat_get_keys(&memc, &stats[x], &rc); + for (ptr= list; *ptr; ptr++) + { + char *value= memcached_stat_get_value(&memc, &stats[x], *ptr, &rc); + server_stats[*ptr]= value; + free(value); + } + + stats_map[server_name]= server_stats; + free(list); + } + + memcached_stat_free(&memc, stats); + return true; + } + private: + std::string servers_list; memcached_st memc; + memcached_server_st *servers; memcached_result_st result; };