Created
October 16, 2018 10:07
-
-
Save DmitriyLyalyuev/486ff4bdc847f5dd442bea8c1724edc8 to your computer and use it in GitHub Desktop.
Redis keys migrating tool
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
import argparse | |
import redis | |
def connect_redis(conn_dict): | |
conn = redis.StrictRedis(host=conn_dict['host'], | |
port=conn_dict['port'], | |
db=conn_dict['db']) | |
return conn | |
def conn_string_type(string): | |
format = '<host>:<port>/<db>' | |
try: | |
host, portdb = string.split(':') | |
port, db = portdb.split('/') | |
db = int(db) | |
except ValueError: | |
raise argparse.ArgumentTypeError('incorrect format, should be: %s' % format) | |
return {'host': host, | |
'port': port, | |
'db': db} | |
def migrate_redis(source, destination): | |
src = connect_redis(source) | |
dst = connect_redis(destination) | |
for key in src.keys('*'): | |
ttl = src.ttl(key) | |
# we handle TTL command returning -1 (no expire) or -2 (no key) | |
if ttl < 0: | |
ttl = 0 | |
print "Dumping key: %s" % key | |
value = src.dump(key) | |
print "Restoring key: %s" % key | |
try: | |
dst.restore(key, ttl * 1000, value) | |
except: | |
pass | |
return | |
def run(): | |
parser = argparse.ArgumentParser() | |
parser.add_argument('source', type=conn_string_type) | |
parser.add_argument('destination', type=conn_string_type) | |
options = parser.parse_args() | |
migrate_redis(options.source, options.destination) | |
if __name__ == '__main__': | |
run() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment