98 lines
2.7 KiB
Go
Executable File
98 lines
2.7 KiB
Go
Executable File
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
|
|
}
|