subhamkrai
bf7daccf60
making necessary code changes to support controller runtime version. Signed-off-by: subhamkrai <srai@redhat.com>
97 lines
3.6 KiB
Go
97 lines
3.6 KiB
Go
/*
|
|
Copyright 2020 The Rook Authors. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// Package config allows a ceph config file to be stored in Kubernetes and mounted as volumes into
|
|
// Ceph daemon containers.
|
|
package config
|
|
|
|
import (
|
|
cephv1 "github.com/rook/rook/pkg/apis/ceph.rook.io/v1"
|
|
v1 "k8s.io/api/core/v1"
|
|
)
|
|
|
|
// ConfigureLivenessProbe returns the desired liveness probe for a given daemon
|
|
func ConfigureLivenessProbe(container v1.Container, livenessProbe *cephv1.ProbeSpec) v1.Container {
|
|
if livenessProbe == nil {
|
|
return container
|
|
}
|
|
if livenessProbe.Disabled {
|
|
container.LivenessProbe = nil
|
|
} else {
|
|
probe := livenessProbe.Probe
|
|
// If the spec value is not empty, let's apply it along with default when some fields are not specified
|
|
if probe != nil {
|
|
// Set the liveness probe on the container to overwrite the default probe created by Rook
|
|
container.LivenessProbe = GetProbeWithDefaults(probe, container.LivenessProbe)
|
|
}
|
|
}
|
|
|
|
return container
|
|
}
|
|
|
|
// ConfigureStartupProbe returns the desired startup probe for a given daemon
|
|
func ConfigureStartupProbe(container v1.Container, startupProbe *cephv1.ProbeSpec) v1.Container {
|
|
if startupProbe == nil {
|
|
return container
|
|
}
|
|
if startupProbe.Disabled {
|
|
container.StartupProbe = nil
|
|
} else {
|
|
probe := startupProbe.Probe
|
|
// If the spec value is not empty, let's apply it along with default when some fields are not specified
|
|
if probe != nil {
|
|
// Set the startup probe on the container to overwrite the default probe created by Rook
|
|
container.StartupProbe = GetProbeWithDefaults(probe, container.StartupProbe)
|
|
}
|
|
}
|
|
|
|
return container
|
|
}
|
|
|
|
func GetProbeWithDefaults(desiredProbe, currentProbe *v1.Probe) *v1.Probe {
|
|
newProbe := *desiredProbe
|
|
|
|
// Do not replace the handler with the previous one!
|
|
// On the first iteration, the handler appears empty and is then replaced by whatever first daemon value comes in
|
|
// e.g: [env -i sh -c ceph --admin-daemon /run/ceph/ceph-mon.b.asok mon_status] - meaning mon b was the first picked in the list of mons
|
|
// On the second iteration the value of mon b remains, since the pointer has been allocated
|
|
// This means the handler is not empty anymore and not replaced by the current one which it should
|
|
//
|
|
// Let's always force the default handler, there is no reason to change it anyway since the underlying content is generated based on the daemon's name
|
|
// so we can not make it generic via the spec
|
|
newProbe.ProbeHandler = currentProbe.ProbeHandler
|
|
|
|
// If the user has not specified thresholds and timeouts, set them to the same values as
|
|
// in the default liveness probe created by Rook.
|
|
if newProbe.FailureThreshold == 0 {
|
|
newProbe.FailureThreshold = currentProbe.FailureThreshold
|
|
}
|
|
if newProbe.PeriodSeconds == 0 {
|
|
newProbe.PeriodSeconds = currentProbe.PeriodSeconds
|
|
}
|
|
if newProbe.SuccessThreshold == 0 {
|
|
newProbe.SuccessThreshold = currentProbe.SuccessThreshold
|
|
}
|
|
if newProbe.TimeoutSeconds == 0 {
|
|
newProbe.TimeoutSeconds = currentProbe.TimeoutSeconds
|
|
}
|
|
if newProbe.InitialDelaySeconds == 0 {
|
|
newProbe.InitialDelaySeconds = currentProbe.InitialDelaySeconds
|
|
}
|
|
|
|
return &newProbe
|
|
}
|