fix: add step to install extension when add node (#683)

Signed-off-by: lixd <xueduan.li@gmail.com> Co-authored-by: lixd <xueduan.li@gmail.com>
kubeclipper · Oct 28, 2023 · ec091b0 · ec091b0
1 parent 4276329
commit ec091b0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/pkg/clusteroperation/node.go b/pkg/clusteroperation/node.go
@@ -170,6 +170,14 @@ func (p *PatchNodes) makeWorkerOperation(extra component.ExtraMetadata, cluster
 		}
 		op.Steps = append(op.Steps, steps...)
 
+		// k8s-extension
+		ext := k8s.Extension{}
+		steps, err = ext.InitStepper(cluster).InstallSteps(stepNodes)
+		if err != nil {
+			return nil, err
+		}
+		op.Steps = append(op.Steps, steps...)
+
 		// kubernetes
 		steps, err = p.getPackageSteps(cluster, action, stepNodes)
 		if err != nil {
@@ -203,6 +211,14 @@ func (p *PatchNodes) makeWorkerOperation(extra component.ExtraMetadata, cluster
 		}
 		op.Steps = append(op.Steps, steps...)
 
+		// k8s-extension
+		ext := k8s.Extension{}
+		steps, err = ext.InitStepper(cluster).UninstallSteps(stepNodes)
+		if err != nil {
+			return nil, err
+		}
+		op.Steps = append(op.Steps, steps...)
+
 		// container runtime
 		steps, err = getCriStep(ctx, cluster, action, stepNodes)
 		if err != nil {