相信不少朋友在迁移数据、扩容存储时需要将memcached中的数据导一份出来,然后在新的memcached中写入,操作步骤如下:
第一步:导出所有key
方法一:memdump工具
memdump --servers="xxx.xxx.xxx.xxx:port" > keys.txt
方法二:
echo "stats cachedump slabid 0" | nc xxx.xxx.xxx.xxx port > keys.txt
第二步:获取所有key对应的value
第三步:将所有的key-value写入新的memcached
问题出来了,当我们在执行第一步时,发现导出的key的数量并不和memcached的number数一样多,如下图:
看了一下memdump命令帮助,发现文档里面描述了不能保证获取到所有的keys,另外查看memdump工具的源码,发现同样是使用stats cachedump命令实现的
于是就看了一下memcached的源码,发现有缓冲区大小限制,最大2M,汗
修改方案:
1、加大缓冲区,但该方法有局限性,当cache中的keys数太大时又面临缓冲区不够用的情况
2、修改stats cachedump命令,使其支持范围导出, stats cachedump slabid start limit
下面修改的代码,仅供参考:
Index: items.c =================================================================== --- items.c (revision 793) +++ items.c (working copy) @@ -276,18 +276,23 @@ } /*@null@*/ -char *do_item_cachedump(const unsigned int slabs_clsid, const unsigned int limit, unsigned int *bytes) { +char *do_item_cachedump(const unsigned int slabs_clsid, const unsigned int start, const unsigned int limit, unsigned int *bytes) { unsigned int memlimit = 2 * 1024 * 1024; /* 2MB max response size */ char *buffer; unsigned int bufcurr; item *it; + int i; unsigned int len; unsigned int shown = 0; char temp[512]; if (slabs_clsid > LARGEST_ID) return NULL; it = heads[slabs_clsid]; - + i = 0; + while (it != NULL && i < start) { + it = it->next; + i++; + } buffer = malloc((size_t)memlimit); if (buffer == 0) return NULL; bufcurr = 0; Index: items.h =================================================================== --- items.h (revision 793) +++ items.h (working copy) @@ -12,7 +12,7 @@ int do_item_replace(item *it, item *new_it); /*@null@*/ -char *do_item_cachedump(const unsigned int slabs_clsid, const unsigned int limit, unsigned int *bytes); +char *do_item_cachedump(const unsigned int slabs_clsid, const unsigned int start, const unsigned int limit, unsigned int *bytes); char *do_item_stats(int *bytes); /*@null@*/ Index: memcached.c =================================================================== --- memcached.c (revision 793) +++ memcached.c (working copy) @@ -1115,7 +1115,7 @@ if (strcmp(subcommand, "cachedump") == 0) { char *buf; - unsigned int bytes, id, limit = 0; + unsigned int bytes, id, limit = 0, start = 0; if(ntokens < 5) { out_string(c, "CLIENT_ERROR bad command line"); @@ -1123,14 +1123,21 @@ } id = strtoul(tokens[2].value, NULL, 10); - limit = strtoul(tokens[3].value, NULL, 10); + if(ntokens == 5){ + start = 0; + limit = strtoul(tokens[3].value, NULL, 10); + } + else if(ntokens == 6){ + start = strtoul(tokens[3].value, NULL, 10); + limit = strtoul(tokens[4].value, NULL, 10); + } if(errno == ERANGE) { out_string(c, "CLIENT_ERROR bad command line format"); return; } - buf = item_cachedump(id, limit, &bytes); + buf = item_cachedump(id, start, limit, &bytes); write_and_free(c, buf, bytes); return; } Index: memcached.h =================================================================== --- memcached.h (revision 793) +++ memcached.h (working copy) @@ -280,7 +280,7 @@ char *mt_defer_delete(item *it, time_t exptime); int mt_is_listen_thread(void); item *mt_item_alloc(char *key, size_t nkey, int flags, rel_time_t exptime, int nbytes); -char *mt_item_cachedump(const unsigned int slabs_clsid, const unsigned int limit, unsigned int *bytes); +char *mt_item_cachedump(const unsigned int slabs_clsid, const unsigned int start, const unsigned int limit, unsigned int *bytes); void mt_item_flush_expired(void); item *mt_item_get_notedeleted(const char *key, const size_t nkey, bool *delete_locked); int mt_item_link(item *it); @@ -309,7 +309,7 @@ # define defer_delete(x,y) mt_defer_delete(x,y) # define is_listen_thread() mt_is_listen_thread() # define item_alloc(x,y,z,a,b) mt_item_alloc(x,y,z,a,b) -# define item_cachedump(x,y,z) mt_item_cachedump(x,y,z) +# define item_cachedump(x,y,z,a) mt_item_cachedump(x,y,z,a) # define item_flush_expired() mt_item_flush_expired() # define item_get_notedeleted(x,y,z) mt_item_get_notedeleted(x,y,z) # define item_link(x) mt_item_link(x) @@ -342,7 +342,7 @@ # define dispatch_event_add(t,c) event_add(&(c)->event, 0) # define is_listen_thread() 1 # define item_alloc(x,y,z,a,b) do_item_alloc(x,y,z,a,b) -# define item_cachedump(x,y,z) do_item_cachedump(x,y,z) +# define item_cachedump(x,y,z,a) do_item_cachedump(x,y,z,a) # define item_flush_expired() do_item_flush_expired() # define item_get_notedeleted(x,y,z) do_item_get_notedeleted(x,y,z) # define item_link(x) do_item_link(x) Index: thread.c =================================================================== --- thread.c (revision 793) +++ thread.c (working copy) @@ -528,11 +528,11 @@ /* * Dumps part of the cache */ -char *mt_item_cachedump(unsigned int slabs_clsid, unsigned int limit, unsigned int *bytes) { +char *mt_item_cachedump(unsigned int slabs_clsid, const unsigned int start, unsigned int limit, unsigned int *bytes) { char *ret; pthread_mutex_lock(&cache_lock); - ret = do_item_cachedump(slabs_clsid, limit, bytes); + ret = do_item_cachedump(slabs_clsid, start, limit, bytes); pthread_mutex_unlock(&cache_lock); return ret; }