112 lines
3.3 KiB
Go
Executable File
112 lines
3.3 KiB
Go
Executable File
// Copyright 2020 VEXXHOST, Inc.
|
|
//
|
|
// 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 builders
|
|
|
|
import (
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
monitoringv1 "opendev.org/vexxhost/openstack-operator/api/monitoring/v1"
|
|
|
|
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
|
|
)
|
|
|
|
// PodMonitorBuilder provides an interface to build podmonitors
|
|
type PodMonitorBuilder struct {
|
|
obj *monitoringv1.PodMonitor
|
|
podMetricsEndpoints []*PodMetricsEndpointBuilder
|
|
owner metav1.Object
|
|
scheme *runtime.Scheme
|
|
}
|
|
|
|
// PodMonitor returns a new podmonitor builder
|
|
func PodMonitor(existing *monitoringv1.PodMonitor, owner metav1.Object, scheme *runtime.Scheme) *PodMonitorBuilder {
|
|
return &PodMonitorBuilder{
|
|
obj: existing,
|
|
owner: owner,
|
|
scheme: scheme,
|
|
}
|
|
}
|
|
|
|
func (pm *PodMonitorBuilder) Labels(labels map[string]string) *PodMonitorBuilder {
|
|
pm.obj.Labels = labels
|
|
return pm
|
|
}
|
|
|
|
func (pm *PodMonitorBuilder) Selector(matchLabels map[string]string) *PodMonitorBuilder {
|
|
pm.obj.Spec.Selector = metav1.LabelSelector{
|
|
MatchLabels: matchLabels,
|
|
}
|
|
return pm
|
|
}
|
|
|
|
func (pm *PodMonitorBuilder) PodTargetLabels(podTargetLabels []string) *PodMonitorBuilder {
|
|
pm.obj.Spec.PodTargetLabels = podTargetLabels
|
|
return pm
|
|
}
|
|
func (pm *PodMonitorBuilder) JobLabel(jobLabel string) *PodMonitorBuilder {
|
|
pm.obj.Spec.JobLabel = jobLabel
|
|
return pm
|
|
}
|
|
func (pm *PodMonitorBuilder) NamespaceSelector(any bool, matchNames []string) *PodMonitorBuilder {
|
|
pm.obj.Spec.NamespaceSelector = monitoringv1.NamespaceSelector{
|
|
Any: any,
|
|
MatchNames: matchNames,
|
|
}
|
|
|
|
return pm
|
|
}
|
|
func (pm *PodMonitorBuilder) SampleLimit(sampleLimit uint64) *PodMonitorBuilder {
|
|
pm.obj.Spec.SampleLimit = sampleLimit
|
|
return pm
|
|
}
|
|
|
|
func (pm *PodMonitorBuilder) PodMetricsEndpoints(pme ...*PodMetricsEndpointBuilder) *PodMonitorBuilder {
|
|
pm.podMetricsEndpoints = pme
|
|
return pm
|
|
}
|
|
|
|
// Build returns the object after making certain assertions
|
|
func (pm *PodMonitorBuilder) Build() error {
|
|
pm.obj.Spec.PodMetricsEndpoints = []monitoringv1.PodMetricsEndpoint{}
|
|
for _, pmeBuilder := range pm.podMetricsEndpoints {
|
|
podMetricsEndpoint, err := pmeBuilder.Build()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pm.obj.Spec.PodMetricsEndpoints = append(pm.obj.Spec.PodMetricsEndpoints, podMetricsEndpoint)
|
|
}
|
|
|
|
if !pm.isOwnedByOthers() {
|
|
return controllerutil.SetControllerReference(pm.owner, pm.obj, pm.scheme)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// isOwnedByOthers checks if this podMonitor has been possessed by an another object already.
|
|
func (pm *PodMonitorBuilder) isOwnedByOthers() bool {
|
|
ownerName := pm.owner.GetName()
|
|
|
|
existingRefs := pm.obj.GetOwnerReferences()
|
|
for _, r := range existingRefs {
|
|
if r.Name == ownerName {
|
|
return false
|
|
} else if r.Controller != nil && *r.Controller {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|