Dev: bootstrap: Setup the stage dependency on init and join side

Author: liangxin1300

crmsh/bootstrap.py

@@ -2136,6 +2136,70 @@ def remove_node_from_cluster(node):
     invoke("corosync-cfgtool -R")
 
 
+def ssh_stage_finished():
+    """
+    Dectect if the ssh stage is finished
+    """
+    key_file_manager = ssh_key.KeyFileManager(sh.cluster_shell())
+    keys = key_file_manager.list_public_key_for_user(None, 'hacluster')
+    return bool(keys)
+
+
+def csync2_stage_finished():
+    """
+    Dectect if the csync2 stage is finished
+    """
+    return ServiceManager().service_is_active(CSYNC2_SERVICE)
+
+
+def corosync_stage_finished():
+    """
+    Dectect if the corosync stage is finished
+    """
+    try:
+        conf_parser.ConfParser.verify_config_file()
+    except ValueError as e:
+        logger.error(e)
+        return False
+    return True
+
+
+INIT_STAGES_DEPENDENCY_MAP = {
+        # stage: (function, is_internal)
+        "ssh": (ssh_stage_finished, False),
+        "csync2": (csync2_stage_finished, False),
+        "corosync": (corosync_stage_finished, False),
+        "remote_auth": (init_remote_auth, True),
+        "sbd": (lambda: True, False),
+        "upgradeutil": (init_upgradeutil, True),
+        "cluster": (is_online, False)
+}
+
+
+JOIN_STAGES_DEPENDENCY_MAP = {
+        # stage: (function, is_internal)
+        "ssh": (ssh_stage_finished, False),
+        "csync2": (csync2_stage_finished, False),
+        "ssh_merge": (lambda: True, False),
+        "cluster": (is_online, False)
+}
+
+
+def check_stage_dependency(stage):
+    """
+    """
+    dependency_map = INIT_STAGES_DEPENDENCY_MAP if _context.type == "init" else JOIN_STAGES_DEPENDENCY_MAP
+    if stage not in dependency_map:
+        return
+    for stage_name, (func, is_internal) in dependency_map.items():
+        if stage_name == stage:
+            break
+        if is_internal:
+            func()
+        elif not func():
+            utils.fatal(f"Please run '{stage_name}' stage first")
+
+
 def bootstrap_init(context):
     """
     Init cluster process
@@ -2163,6 +2227,7 @@ def bootstrap_init(context):
         _context.node_list_in_cluster = [utils.this_node()]
 
     if stage != "":
+        check_stage_dependency(stage)
         globals()["init_" + stage]()
     else:
         init_ssh()