mirror of
https://github.com/kubevirt/containerized-data-importer.git
synced 2025-06-03 06:30:22 +00:00
3bb70209d0
* refactor monitoring Signed-off-by: avlitman <alitman@redhat.com> * Upgrade pointer to pnt Signed-off-by: avlitman <alitman@redhat.com> * fix controller base and ready gague Signed-off-by: avlitman <alitman@redhat.com> --------- Signed-off-by: avlitman <alitman@redhat.com>
83 lines
2.9 KiB
Go
83 lines
2.9 KiB
Go
/*
|
|
Copyright 2021 The CDI Authors.
|
|
|
|
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.
|
|
limitations under the License.
|
|
See the License for the specific language governing permissions and
|
|
*/
|
|
|
|
package controller
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
|
|
cdiv1 "kubevirt.io/containerized-data-importer-api/pkg/apis/core/v1beta1"
|
|
"kubevirt.io/containerized-data-importer/pkg/monitoring/metrics/cdi-controller"
|
|
)
|
|
|
|
const (
|
|
noDigest = "NoDigest"
|
|
noImport = "NoImport"
|
|
outdated = "Outdated"
|
|
scheduled = "ImportScheduled"
|
|
inProgress = "ImportProgressing"
|
|
upToDate = "UpToDate"
|
|
)
|
|
|
|
func updateDataImportCronCondition(cron *cdiv1.DataImportCron, conditionType cdiv1.DataImportCronConditionType, status corev1.ConditionStatus, message, reason string) {
|
|
if conditionType == cdiv1.DataImportCronUpToDate {
|
|
metrics.SetDataImportCronOutdated(getPrometheusCronLabels(types.NamespacedName{Namespace: cron.Namespace, Name: cron.Name}), status != corev1.ConditionTrue)
|
|
}
|
|
if condition := FindDataImportCronConditionByType(cron, conditionType); condition != nil {
|
|
updateConditionState(&condition.ConditionState, status, message, reason)
|
|
} else {
|
|
condition = &cdiv1.DataImportCronCondition{Type: conditionType}
|
|
updateConditionState(&condition.ConditionState, status, message, reason)
|
|
cron.Status.Conditions = append(cron.Status.Conditions, *condition)
|
|
}
|
|
}
|
|
|
|
// FindDataImportCronConditionByType finds DataImportCronCondition by condition type
|
|
func FindDataImportCronConditionByType(cron *cdiv1.DataImportCron, conditionType cdiv1.DataImportCronConditionType) *cdiv1.DataImportCronCondition {
|
|
for i, condition := range cron.Status.Conditions {
|
|
if condition.Type == conditionType {
|
|
return &cron.Status.Conditions[i]
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func updateConditionState(condition *cdiv1.ConditionState, status corev1.ConditionStatus, message, reason string) {
|
|
conditionStatusUpdated := condition.Status != status
|
|
conditionUpdated := conditionStatusUpdated || condition.Message != message || condition.Reason != reason
|
|
if conditionUpdated {
|
|
now := metav1.Now()
|
|
if conditionStatusUpdated {
|
|
condition.Status = status
|
|
condition.LastTransitionTime = now
|
|
}
|
|
condition.LastHeartbeatTime = now
|
|
condition.Reason = reason
|
|
condition.Message = message
|
|
}
|
|
}
|
|
|
|
func getPrometheusCronLabels(cron types.NamespacedName) prometheus.Labels {
|
|
return prometheus.Labels{
|
|
prometheusNsLabel: cron.Namespace,
|
|
prometheusCronNameLabel: cron.Name,
|
|
}
|
|
}
|