Series Upgrade

Implement the series-upgrade feature allowing to move between Ubuntu series. 

3 files changed, 34 insertions, 0 deletions

diff --git a/hooks/hooks.py b/hooks/hooks.py
index ec4adfb..5e0d261 100755
--- a/hooks/hooks.py
+++ b/hooks/hooks.py
@@ -613,6 +613,20 @@ def remove_replset_from_upstart():
 return(retVal)
 
 
+def step_down_replset_primary():
+
+ retVal = mongo_client('localhost', 'rs.stepDown()')
+ c = MongoClient('localhost')
+
+ while True:
+ # If we are still primary wait a bit more for step down
+ if not am_i_primary():
+ break
+
+ juju_log("step_down_replset_primary returns: %s" % retVal, level=DEBUG)
+ return(retVal)
+
+
 def remove_rest_from_upstart():
 """Removes --rest from upstart script
 """
@@ -1831,6 +1845,24 @@ def write_logrotate_config(config_data,
 return True
 
 
+@hooks.hook('pre-series-upgrade')
+def pre_series_upgrade():
+ juju_log("Running prepare series upgrade hook", "INFO")
+ if am_i_primary():
+ step_down_replset_primary()
+ service('stop', 'mongodb')
+ status_set(
+ "blocked",
+ "Ready for do-release-upgrade and reboot. "
+ "Set complete when finished.")
+
+
+@hooks.hook('post-series-upgrade')
+def post_series_upgrade():
+ juju_log("Running complete series upgrade hook", "INFO")
+ update_status()
+
+
 ###############################################################################
 # Main section
 ###############################################################################
diff --git a/hooks/post-series-upgrade b/hooks/post-series-upgrade
new file mode 120000
index 0000000..9416ca6
--- /dev/null
+++ b/hooks/post-series-upgrade
@@ -0,0 +1 @@
+hooks.py \ No newline at end of file
diff --git a/hooks/pre-series-upgrade b/hooks/pre-series-upgrade
new file mode 120000
index 0000000..9416ca6
--- /dev/null
+++ b/hooks/pre-series-upgrade
@@ -0,0 +1 @@
+hooks.py \ No newline at end of file