【Redis】用python操作redis叢集

小亮520cl發表於2018-05-05
RedisPython

https://blog.csdn.net/bitcarmanlee/article/details/51852126

 密碼不能寫到列表中去：

  1. 有密碼 就是這樣 redisconn = StrictRedisCluster(startup_nodes=redis_nodes，password=password)


結合上一篇文章，改進指令碼後操作redis-cluster
  1. [root@ip-172-31-47-226 tmp]# more /root/get_nottl.py
  2. # encoding: utf-8
  3. """
  4. author: yangyi@youzan.com
  5. time: 2018/4/26 下午4:34
  6. func: 獲取資料庫中沒有設定ttl的 key
  7. """
  8. import redis
  9. import argparse
  10. import time
  11. import sys
  12. import rediscluster

  14. class ShowProcess:
  15.     """
  16.     顯示處理進度的類
  17.     呼叫該類相關函式即可實現處理進度的顯示
  18.     """
  19.     i = 0 # 當前的處理進度
  20.     max_steps = 0 # 總共需要處理的次數
  21.     max_arrow = 50 # 進度條的長度

  23.     # 初始化函式，需要知道總共的處理次數
  24.     def __init__(self, max_steps):
  25.         self.max_steps = max_steps
  26.         self.i = 0

  28.     # 顯示函式，根據當前的處理進度i顯示進度
  29.     # 效果為[>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]100.00%
  30.     def show_process(self, i = None):
  31.         if i is not None:
  32.             self.i = i
  33.         else:
  34.             self.i += 1
  35.         num_arrow = int(int(self.i) * int(self.max_arrow) / int(self.max_steps)) # 計算顯示多少個'>'
  36.         num_line = self.max_arrow - num_arrow # 計算顯示多少個'-'
  37.         percent = self.i * 100.0 / self.max_steps # 計算完成進度，格式為xx.xx%
  38.         process_bar = '[' + '>' * num_arrow + ' ' * num_line + ']'+ '%.2f' % percent + '%' + '\r' # 帶輸出的字串，'\r'表示不換行回到最左邊
  39.         sys.stdout.write(process_bar) # 這兩句列印字元到終端
  40.         sys.stdout.flush()

  42.     def close(self, words='done'):
  43.         print ''
  44.         print words
  45.         self.i = 0


  48. def check_ttl(redis_conn, no_ttl_file, dbindex):
  49.     start_time = time.time()
  50.     no_ttl_num = 0
  51.     allkey = redis_conn.dbsize()
  52.     keys_num = sum(list(set(allkey.values())))
  53.     print "key分佈如下:",allkey
  54.     print "there are {num} keys in db {index} ".format(num=keys_num, index=dbindex)
  55.     process_bar = ShowProcess(keys_num)
  56.     with open(no_ttl_file, 'a') as f:

  58.         for key in redis_conn.scan_iter(count=10):
  59.             process_bar.show_process()
  60.             if redis_conn.ttl(key) == -1:
  61.                 no_ttl_num += 1
  62.                 if no_ttl_num < 10000:
  63.                     f.write(key+'\n')
  64.             else:
  65.                 continue

  67.     process_bar.close()
  68.     print "cost time(s):", time.time() - start_time
  69.     print "no ttl keys number:", no_ttl_num
  70.     print "we write keys with no ttl to the file: %s" % no_ttl_file


  73. def main():
  74.     parser = argparse.ArgumentParser()
  75.     parser.add_argument('-p', type=int, dest='port', action='store', help='port of redis ')
  76.     parser.add_argument('-a', type=str, dest='password', action='store', help='password of redis ')
  77.     parser.add_argument('-d', type=str, dest='db_list', action='store', default=0, help='ex : -d all / -d 1,2,3,4 ')
  78.     args = parser.parse_args()
  79.     port = args.port
  80.     pwd = args.password
  81.     if args.db_list == 'all':
  82.         db_list = [i for i in xrange(0, 16)]
  83.     else:
  84.         db_list = [int(i) for i in args.db_list.split(',')]

  86.     for index in db_list:
  87.         try:
  88.             startup_nodes = [{"host": "172.31.47.226", "port": port,"db":index}]
  89.             r = rediscluster.StrictRedisCluster(startup_nodes=startup_nodes, password=pwd)
  90.         except Exception, e:
  91.             print e
  92.         else:
  93.             no_ttl_keys_file = "/tmp/{port}_{db}_no_ttl_keys.txt".format(port=port, db=index)
  94.             check_ttl(r, no_ttl_keys_file, index)


  97. if __name__ == '__main__':
  98.     main()





來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/29096438/viewspace-2153849/，如需轉載，請註明出處，否則將追究法律責任。

相關文章