53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
#!/usr/bin/python
|
||
# -*- coding:utf-8 -*-
|
||
|
||
'''
|
||
原始http请求信息以Hash形式保存在redis，hash格式不利于随机取一个http请求
|
||
用此脚本将db0中http请求转移至db1，并以set形式保存
|
||
'''
|
||
|
||
import json
|
||
import redis
|
||
import sys
|
||
import param
|
||
|
||
def queryKey(key):
|
||
r = redis.StrictRedis(host=param.host, port=param.port, password=param.password, db=param.dbOld)
|
||
return r.keys(key)
|
||
|
||
def queryHashFields(key):
|
||
r = redis.StrictRedis(host=param.host, port=param.port, password=param.password, db=param.dbOld)
|
||
return r.hkeys(key)
|
||
|
||
def queryHashValue(key, filed):
|
||
r = redis.StrictRedis(host=param.host, port=param.port, password=param.password, db=param.dbOld)
|
||
return r.hget(key, filed)
|
||
|
||
def setAdd(key, value):
|
||
r = redis.StrictRedis(host=param.host, port=param.port, password=param.password, db=param.dbNew)
|
||
r.sadd(key, value)
|
||
|
||
def main():
|
||
if len(sys.argv) < 2:
|
||
print "缺少参数： python %s hostname" % sys.argv[0]
|
||
exit()
|
||
hostname = sys.argv[1]
|
||
print hostname
|
||
keys = queryKey(hostname)
|
||
for key in keys:
|
||
print key
|
||
fileds = queryHashFields(key)
|
||
total = len(fileds)
|
||
times = 0
|
||
print "共计%d个请求记录" % total
|
||
for filed in fileds:
|
||
times = times + 1
|
||
req = queryHashValue(key, filed)
|
||
setAdd(param.joblist, req)
|
||
if times*10%total == 0:
|
||
print str(times*100.0/total) + "%"
|
||
print "finish..."
|
||
|
||
if __name__ == '__main__':
|
||
main()
|