-
-
Save ridwandwisiswanto/18db7569fbce0d03f24510315ffd04c8 to your computer and use it in GitHub Desktop.
A simple script to migrate all keys from one Redis to another
This file contains hidden or 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 | |
| # use with this command | |
| # python redis_migrate.py host:port:password/db(int) host:port:password/db(int) | |
| 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 | |
| # untuk yang pakai auth | |
| def conn_string_type(string): | |
| format = '<host>:<port>:<password>/<db>' | |
| try: | |
| host, port,passworddb = string.split(':') | |
| password, db = passworddb.split('/') | |
| db = int(db) | |
| except ValueError: | |
| raise argparse.ArgumentTypeError('incorrect format, should be: %s' % format) | |
| return {'host': host, | |
| 'port': port, | |
| 'password': password, | |
| '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, replace=True) | |
| except redis.exceptions.ResponseError: | |
| print "Failed to restore key: %s" % key | |
| 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