Add node placement support to the example operator

Signed-off-by: Jakub Dzon <[email protected]>
kubevirt · Oct 15, 2020 · 853184e · 853184e
1 parent 0ec883c
commit 853184e
Show file tree

Hide file tree

Showing 6 changed files with 1,560 additions and 7 deletions.
diff --git a/examples/sample-operator/manifests/sample-operator/v0.0.4/operator.yaml b/examples/sample-operator/manifests/sample-operator/v0.0.4/operator.yaml
diff --git a/examples/sample-operator/manifests/sample-operator/v0.0.4/sampleconfig_cr.yaml b/examples/sample-operator/manifests/sample-operator/v0.0.4/sampleconfig_cr.yaml
@@ -0,0 +1,5 @@
+apiVersion: sample.lifecycle.kubevirt.io/v1alpha1
+kind: SampleConfig
+metadata:
+  name: example-sampleconfig
+spec: {}
diff --git a/examples/sample-operator/pkg/apis/sample/v1alpha1/sampleconfig_types.go b/examples/sample-operator/pkg/apis/sample/v1alpha1/sampleconfig_types.go
@@ -7,6 +7,9 @@ import (
 
 // SampleConfigSpec defines the desired state of SampleConfig
 type SampleConfigSpec struct {
+	// Rules on which nodes controller pod(s) will be scheduled
+	// +optional
+	Infra sdkapi.NodePlacement `json:"infra,omitempty"`
 }
 
 // SampleConfigStatus defines the observed state of SampleConfig

diff --git a/examples/sample-operator/pkg/controller/sampleconfig/cr-manager.go b/examples/sample-operator/pkg/controller/sampleconfig/cr-manager.go
@@ -54,7 +54,7 @@ func (m *CrManager) Status(cr runtime.Object) *sdkapi.Status {
 }
 
 // GetAllResources provides all resources managed by the cr
-func (m *CrManager) GetAllResources(_ runtime.Object) ([]runtime.Object, error) {
+func (m *CrManager) GetAllResources(obj runtime.Object) ([]runtime.Object, error) {
 	namespace := m.operatorArgs.Namespace
 
 	serviceAccount := resourceBuilder.CreateServiceAccount(HTTPServerName)
@@ -66,7 +66,8 @@ func (m *CrManager) GetAllResources(_ runtime.Object) ([]runtime.Object, error)
 	roleBinding := resourceBuilder.CreateRoleBinding(HTTPServerName, HTTPServerName, HTTPServerName, namespace)
 	roleBinding.Namespace = namespace
 
-	httpDeployment := m.createHTTPServerDeployment()
+	cr := obj.(*v1alpha1.SampleConfig)
+	httpDeployment := m.createHTTPServerDeployment(&cr.Spec.Infra)
 	httpService := m.createHTTPServerService()
 
 	return []runtime.Object{
@@ -88,7 +89,7 @@ func (m *CrManager) createHTTPServerService() *v1.Service {
 	return httpService
 }
 
-func (m *CrManager) createHTTPServerDeployment() *appsv1.Deployment {
+func (m *CrManager) createHTTPServerDeployment(placement *sdkapi.NodePlacement) *appsv1.Deployment {
 	ports := []v1.ContainerPort{
 		{
 			Name:          "http",
@@ -102,7 +103,7 @@ func (m *CrManager) createHTTPServerDeployment() *appsv1.Deployment {
 			*container,
 		},
 	}
-	return resourceBuilder.CreateOperatorDeployment(HTTPServerDeploymentName, m.operatorArgs.Namespace, deploymentMatchKey, HTTPServerName, HTTPServerName, 1, podSpec)
+	return resourceBuilder.CreateDeployment(HTTPServerDeploymentName, m.operatorArgs.Namespace, deploymentMatchKey, HTTPServerName, HTTPServerName, 1, podSpec, placement)
 }
 
 // GetDependantResourcesListObjects returns resource list objects of dependant resources