Update MySQL replication tasks

alexmuller · alexmuller · commit f7d7dfd38a33 · 2015-11-06T10:16:35.000Z
Change the purpose of the task `setup_slave_from_master` so that it
does actually do the setup.

Rename the existing task to `replicate_slave_from_master` because that's
what it's actually doing.
diff --git a/mysql.py b/mysql.py
@@ -1,4 +1,5 @@
 from fabric.api import *
+from fabric.operations import prompt
 
 
 def run_mysql_command(cmd):
@@ -40,14 +41,37 @@ def fix_replication_from_slow_query_log_after_upgrade():
 @task
 def setup_slave_from_master(master):
     """
-    Sets up a slave from a master by taking a dump from the master,
-    copying it to the slave and then restoring the dump.
+    Sets up a slave from a master by:
+      - configuring MySQL replication config
+      - using the replicate_slave_from_master task to do an initial dump to the slave
 
     Usage: fab environment -H mysql-slave-1.backend mysql.setup_slave_from_master:'mysql-master-1.backend'
     """
     if len(env.hosts) > 1:
         exit('This job is currently only setup to run against one slave at a time')
 
+    mysql_master = prompt("Master host (eg 'master.mysql' or 'whitehall-master.mysql'):")
+    replication_username = 'replica_user'
+    replication_password = prompt("Password for MySQL user {0}:".format(replication_username))
+
+    run_mysql_command("STOP SLAVE;")
+    run_mysql_command("CHANGE MASTER TO MASTER_HOST='{0}', MASTER_USER='{1}', MASTER_PASSWORD='{2}';".format(
+        mysql_master, replication_username, replication_password))
+
+    replicate_slave_from_master(master)
+
+
+@task
+def replicate_slave_from_master(master):
+    """
+    Updates a slave from a master by taking a dump from the master,
+    copying it to the slave and then restoring the dump.
+
+    Usage: fab environment -H mysql-slave-1.backend mysql.replicate_slave_from_master:'mysql-master-1.backend'
+    """
+    if len(env.hosts) > 1:
+        exit('This job is currently only setup to run against one slave at a time')
+
     with settings(host_string=master):
         # The use of `--master-data` here implies `--lock-all-tables` per the
         # MySQL reference manual: http://dev.mysql.com/doc/refman/5.1/en/mysqldump.html#option_mysqldump_master-data
