Allow to disable probes.

Author: lamai93

docs/Manual/Deployment/Kubernetes/DeploymentResource.md

@@ -401,6 +401,14 @@ for each server of this group.
 This setting is not available for group `coordinators`, `syncmasters` & `syncworkers`
 because servers in these groups do not need persistent storage.
 
+### `spec.<group>.probes.livenessProbeDisabled: bool`
+
+If set to true, the operator does not generate a liveness probe for new pods belonging to this group.
+
+### `spec.<group>.probes.readinessProbeDisabled: bool`
+
+If set to true, the operator does not generate a readiness probe for new pods belonging to this group.
+
 ### `spec.<group>.tolerations: []Toleration`
 
 This setting specifies the `tolerations` for the `Pod`s created

pkg/apis/deployment/v1alpha/server_group_spec.go

@@ -56,6 +56,40 @@ type ServerGroupSpec struct {
 	ServiceAccountName *string `json:"serviceAccountName,omitempty"`
 	// NodeSelector speficies a set of selectors for nodes
 	NodeSelector map[string]string `json:"nodeSelector,omitempty"`
+	// Probes specifies additional behaviour for probes
+	Probes *ServerGroupProbesSpec `json:"probes,omitempty"`
+}
+
+// ServerGroupProbesSpec contains specification for probes for pods of the server group
+type ServerGroupProbesSpec struct {
+	// LivenessProbeDisabled if true livenessProbes are disabled
+	LivenessProbeDisabled *bool `json:"livenessProbeDisabled,omitempty"`
+	// LivenessProbeDisabled if specified the given probes is used as liveness probe
+	//LivenessProbeOverride *v1.Probe `json:"LivenessProbeOverride,omitempty"`
+	// LivenessProbeDisabled if true readinessProbes are disabled
+	ReadinessProbeDisabled *bool `json:"ReadinessProbeDisabled,omitempty"`
+	// ReadinessProbeOverride if specified the given probes is used as readiness probe
+	//ReadinessProbeOverride *v1.Probe `json:"ReadinessProbeOverride,omitempty"`
+}
+
+// // HasLivenessProbeOverride returns true if a livenessprobe override is set
+// func (s ServerGroupProbesSpec) HasLivenessProbeOverride() bool {
+// 	return s.LivenessProbeOverride != nil
+// }
+
+// // HasReadinessProbeOverride returns true if a readinessprobe override is set
+// func (s ServerGroupProbesSpec) HasReadinessProbeOverride() bool {
+// 	return s.ReadinessProbeOverride != nil
+// }
+
+// IsLivenessProbeDisabled returns true if liveness probes are disabled
+func (s ServerGroupProbesSpec) IsLivenessProbeDisabled() bool {
+	return util.BoolOrDefault(s.LivenessProbeDisabled)
+}
+
+// IsReadinessProbeDisabled returns true if readiness probes are disabled
+func (s ServerGroupProbesSpec) IsReadinessProbeDisabled() bool {
+	return util.BoolOrDefault(s.ReadinessProbeDisabled)
 }
 
 // GetCount returns the value of count.
@@ -98,6 +132,19 @@ func (s ServerGroupSpec) GetServiceAccountName() string {
 	return util.StringOrDefault(s.ServiceAccountName)
 }
 
+// HasProbesSpec returns true if Probes is non nil
+func (s ServerGroupSpec) HasProbesSpec() bool {
+	return s.Probes != nil
+}
+
+// GetProbesSpec returns the Probes spec or the nil value if not set
+func (s ServerGroupSpec) GetProbesSpec() ServerGroupProbesSpec {
+	if s.HasProbesSpec() {
+		return *s.Probes
+	}
+	return ServerGroupProbesSpec{}
+}
+
 // Validate the given group spec
 func (s ServerGroupSpec) Validate(group ServerGroup, used bool, mode DeploymentMode, env Environment) error {
 	if used {

pkg/apis/deployment/v1alpha/zz_generated.deepcopy.go

@@ -331,6 +331,15 @@ func createArangoSyncArgs(apiObject metav1.Object, spec api.DeploymentSpec, grou
 
 // createLivenessProbe creates configuration for a liveness probe of a server in the given group.
 func (r *Resources) createLivenessProbe(spec api.DeploymentSpec, group api.ServerGroup) (*k8sutil.HTTPProbeConfig, error) {
+	groupspec := spec.GetServerGroupSpec(group)
+
+	if groupspec.HasProbesSpec() {
+		probesspec := groupspec.GetProbesSpec()
+		if probesspec.IsLivenessProbeDisabled() {
+			return nil, nil
+		}
+	}
+
 	switch group {
 	case api.ServerGroupSingle, api.ServerGroupAgents, api.ServerGroupDBServers:
 		authorization := ""
@@ -394,9 +403,19 @@ func (r *Resources) createLivenessProbe(spec api.DeploymentSpec, group api.Serve
 
 // createReadinessProbe creates configuration for a readiness probe of a server in the given group.
 func (r *Resources) createReadinessProbe(spec api.DeploymentSpec, group api.ServerGroup, version driver.Version) (*k8sutil.HTTPProbeConfig, error) {
+	groupspec := spec.GetServerGroupSpec(group)
+
+	if groupspec.HasProbesSpec() {
+		probesspec := groupspec.GetProbesSpec()
+		if probesspec.IsLivenessProbeDisabled() {
+			return nil, nil
+		}
+	}
+
 	if group != api.ServerGroupSingle && group != api.ServerGroupCoordinators {
 		return nil, nil
 	}
+
 	authorization := ""
 	if spec.IsAuthenticated() {
 		secretData, err := r.getJWTSecret(spec)