added a wait for mysql via haproxy after external restore script

Author: ashishkeshan

share/github-backup-utils/ghe-restore-mysql

@@ -32,6 +32,11 @@ snapshot_dir="$GHE_DATA_DIR/$GHE_RESTORE_SNAPSHOT"
 if is_external_database_snapshot; then
   if [ -n "$EXTERNAL_DATABASE_RESTORE_SCRIPT" ]; then
     $EXTERNAL_DATABASE_RESTORE_SCRIPT
+    # ensure that haproxy and mysql are accepting connections before continuining
+    if ! ghe-ssh "$GHE_HOSTNAME" -- /usr/local/share/enterprise/ghe-service-wait-mysql; then
+      error_message "Failed to connect to MySQL service!"
+      exit 2
+    fi
     bm_end "$(basename $0)"
     exit 0
   else