diff --git a/scripts/centralize_database b/scripts/centralize_database new file mode 100755 index 0000000000..3ff958117a --- /dev/null +++ b/scripts/centralize_database @@ -0,0 +1,62 @@ +#!/usr/bin/python +from __future__ import print_function +import sys +import swsssdk +import redis +import argparse + +def centralize_to_target_db(target_dbname): + target_dbport = swsssdk.SonicDBConfig.get_port(target_dbname) + target_dbhost = swsssdk.SonicDBConfig.get_hostname(target_dbname) + + dblists = swsssdk.SonicDBConfig.get_dblist() + for dbname in dblists: + dbport = swsssdk.SonicDBConfig.get_port(dbname) + dbhost = swsssdk.SonicDBConfig.get_hostname(dbname) + # if the db is on the same instance, no need to move + if dbport == target_dbport and dbhost == target_dbhost: + continue + + dbsocket = swsssdk.SonicDBConfig.get_socket(dbname) + dbid = swsssdk.SonicDBConfig.get_dbid(dbname) + + r = redis.Redis(host=dbhost, unix_socket_path=dbsocket, db=dbid) + + script = """ + local cursor = 0; + repeat + local dat = redis.call('SCAN', cursor, 'COUNT', 7000); + cursor = dat[1]; + redis.call('MIGRATE', KEYS[1], KEYS[2], '', KEYS[3], 5000, 'COPY', 'REPLACE', 'KEYS', unpack(dat[2])); + until cursor == '0'; + """ + r.eval(script, 3, target_dbhost, target_dbport, dbid) + + #SAVE rdb file + r = redis.Redis(host=target_dbhost, port=target_dbport) + r.save() + +def main(): + parser = argparse.ArgumentParser(description='centralize all db data into one db instances', + formatter_class=argparse.RawTextHelpFormatter, + epilog= +""" +Example : centralize_database APPL_DB +""") + parser.add_argument('target_db', type=str, help='move all db data into the instance where target db locates') + args = parser.parse_args() + + if args.target_db: + try: + centralize_to_target_db(args.target_db) + print(swsssdk.SonicDBConfig.get_instancename(args.target_db)) + except Exception as ex: + template = "An exception of type {0} occurred. Arguments:\n{1!r}" + message = template.format(type(ex).__name__, ex.args) + print(message, file=sys.stderr) + sys.exit(1) + else: + parser.print_help() + +if __name__ == "__main__": + main() diff --git a/scripts/fast-reboot b/scripts/fast-reboot index 48c643ae6a..e0b48f912c 100755 --- a/scripts/fast-reboot +++ b/scripts/fast-reboot @@ -247,10 +247,13 @@ function backup_database() end end " 0 > /dev/null - sonic-db-cli SAVE > /dev/null - #TODO : need a script to copy all rdb files if there is multiple db instances config - docker cp database:/var/lib/redis/$REDIS_FILE $WARM_DIR - docker exec -i database rm /var/lib/redis/$REDIS_FILE + + # move all db data into the instance where APPL_DB locates + target_db_inst=`centralize_database APPL_DB` + + # Dump redis content to a file 'dump.rdb' in warmboot directory + docker cp database:/var/lib/$target_db_inst/$REDIS_FILE $WARM_DIR + docker exec -i database rm /var/lib/$target_db_inst/$REDIS_FILE } function setup_control_plane_assistant() diff --git a/setup.py b/setup.py index 058f1c4ea1..e4101f67d7 100644 --- a/setup.py +++ b/setup.py @@ -110,7 +110,8 @@ 'scripts/warm-reboot', 'scripts/watermarkstat', 'scripts/watermarkcfg', - 'scripts/sonic-kdump-config' + 'scripts/sonic-kdump-config', + 'scripts/centralize_database' ], entry_points={ 'console_scripts': [