完善坛子工具库之--python版memcached遍历脚本

  1. 云栖社区>
  2. 博客>
  3. 正文

完善坛子工具库之--python版memcached遍历脚本

荣华 2016-03-28 10:52:27 浏览930
展开阅读全文
今天看到robbin大哥用ruby写了一个遍历memcached的脚本,由于自己不用ruby就鳴发了用python也写一个的想法,说做就做,虽然功能上没有ruby版的强憾,不过也可以凑合用一下。

这个python版本还可以继续优化。代码如下:
import socket, StringIO
'''''
@author: ahuaxuan
@date: 2008-10-22
'''
class mcstats(object):
def __init__(self, address, port):
self.address = address
self.port = port
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.s.connect((self.address, self.port))
def __del__(self):
self.s.close()
def connect(self, command):
num = self.s.send(command)
totalData = ''
while True:
data = self.s.recv(1024)
if len(data) <= 0:
break
totalData = totalData + data
if totalData.find('END') >= 0:
break
return totalData
def calcSlabsCount(self, data):
f = StringIO.StringIO(data)
tmp = None
for a in f:
if a.find('END') >= 0:
break
else:
tmp = a
f.close()
if tmp != None:
return int(tmp.split(":")[1])
else:
return 0
def showKVpairs(self, slabCounts, command):
for a in range(0, slabCounts):
tmpc = command + str(a + 1) + ' 0 \r\n'
data = self.connect(tmpc)
f = StringIO.StringIO(data)
for b in f:
if b.find('ITEM') >= 0:
arr = b.split(' ')
print 'key : ' + arr[1] + ' ------ value size : ' + arr[2][1:len(arr[2])]
f.close()
if __name__ == "__main__":
mcs = mcstats("localhost", 11211)
mcs.showKVpairs(mcs.calcSlabsCount(mcs.connect('stats items \r\n')), 'stats cachedump ')

网友评论

登录后评论
0/500
评论
荣华
+ 关注