feat: enable affinity for Sensor (#1176)

Signed-off-by: Derek Wang <[email protected]>
argoproj · Apr 12, 2021 · 8fd9ae7 · 8fd9ae7
1 parent f752bc0
commit 8fd9ae7
Show file tree

Hide file tree

Showing 9 changed files with 343 additions and 238 deletions.
diff --git a/api/openapi-spec/swagger.json b/api/openapi-spec/swagger.json
diff --git a/api/sensor.html b/api/sensor.html
diff --git a/api/sensor.md b/api/sensor.md
diff --git a/controllers/sensor/resource.go b/controllers/sensor/resource.go
@@ -295,6 +295,7 @@ func buildDeploymentSpec(args *AdaptorArgs) (*appv1.DeploymentSpec, error) {
 		spec.Template.Spec.SecurityContext = args.Sensor.Spec.Template.SecurityContext
 		spec.Template.Spec.NodeSelector = args.Sensor.Spec.Template.NodeSelector
 		spec.Template.Spec.Tolerations = args.Sensor.Spec.Template.Tolerations
+		spec.Template.Spec.Affinity = args.Sensor.Spec.Template.Affinity
 		spec.Template.Spec.ImagePullSecrets = args.Sensor.Spec.Template.ImagePullSecrets
 		spec.Template.Spec.PriorityClassName = args.Sensor.Spec.Template.PriorityClassName
 		spec.Template.Spec.Priority = args.Sensor.Spec.Template.Priority

diff --git a/pkg/apis/sensor/v1alpha1/generated.pb.go b/pkg/apis/sensor/v1alpha1/generated.pb.go
diff --git a/pkg/apis/sensor/v1alpha1/generated.proto b/pkg/apis/sensor/v1alpha1/generated.proto
diff --git a/pkg/apis/sensor/v1alpha1/openapi_generated.go b/pkg/apis/sensor/v1alpha1/openapi_generated.go
diff --git a/pkg/apis/sensor/v1alpha1/types.go b/pkg/apis/sensor/v1alpha1/types.go
@@ -179,6 +179,9 @@ type Template struct {
 	// More info: https://kubernetes.io/docs/concepts/configuration/pod-priority-preemption/
 	// +optional
 	Priority *int32 `json:"priority,omitempty" protobuf:"bytes,10,opt,name=priority"`
+	// If specified, the pod's scheduling constraints
+	// +optional
+	Affinity *corev1.Affinity `json:"affinity,omitempty" protobuf:"bytes,11,opt,name=affinity"`
 }
 
 // EventDependency describes a dependency

diff --git a/pkg/apis/sensor/v1alpha1/zz_generated.deepcopy.go b/pkg/apis/sensor/v1alpha1/zz_generated.deepcopy.go