solve the label issue of singlenode through affinity

Signed-off-by: hancheng wang <wanghanchengchn@gmail.com> solve the label issue of singlenode through affinity Signed-off-by: hancheng wang <wanghanchengchn@gmail.com>
vhive-serverless · Apr 18, 2024 · f0cbf36 · f0cbf36
1 parent 5143a83
commit f0cbf36
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/scripts/cloudlab/start_onenode_vhive_cluster.go b/scripts/cloudlab/start_onenode_vhive_cluster.go
@@ -122,6 +122,10 @@ func StartOnenodeVhiveCluster(sandbox string) error {
 		return err
 	}
 
+	if err = cluster.SetupMasterNode(sandbox); err != nil {
+		return err
+	}
+
 	utils.InfoPrintf("All logs are stored in %s", ctrdLogDir)
 	return nil
 }
diff --git a/scripts/cluster/create_one_node_cluster.go b/scripts/cluster/create_one_node_cluster.go
@@ -70,11 +70,6 @@ func CreateOneNodeCluster(stockContainerd string) error {
 		return err
 	}
 
-	err = SetupMasterNode(stockContainerd)
-	if err != nil {
-		return err
-	}
-
 	return nil
 }
 

diff --git a/scripts/setup.go b/scripts/setup.go
@@ -79,6 +79,7 @@ func main() {
 	availableCmds := []string{
 		"create_multinode_cluster",
 		"create_one_node_cluster",
+		"prepare_one_node_cluster",
 		"setup_master_node",
 		"setup_worker_kubelet",
 		"setup_node",
@@ -157,6 +158,18 @@ func main() {
 		}
 		utils.InfoPrintf("Create one-node Cluster\n")
 		err = cluster.CreateOneNodeCluster(setupFlags.Args()[1])
+		if err == nil {
+			utils.InfoPrintf("Set up master node\n")
+			err = cluster.SetupMasterNode(setupFlags.Args()[1])
+		}
+	case "prepare_one_node_cluster":
+		if setupFlags.NArg() < 2 {
+			utils.FatalPrintf("Missing parameters: %s <stock-containerd>\n", subCmd)
+			utils.CleanEnvironment()
+			os.Exit(1)
+		}
+		utils.InfoPrintf("Create one-node Cluster\n")
+		err = cluster.CreateOneNodeCluster(setupFlags.Args()[1])
 	case "setup_master_node":
 		if setupFlags.NArg() < 2 {
 			utils.FatalPrintf("Missing parameters: %s <stock-containerd>\n", subCmd)