-
Notifications
You must be signed in to change notification settings - Fork 18
/
grafana_controller.go
125 lines (105 loc) · 3.47 KB
/
grafana_controller.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
// Copyright (c) 2022 Aiven, Helsinki, Finland. https://aiven.io/
package controllers
import (
"context"
"fmt"
"strings"
"github.com/aiven/aiven-go-client"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"github.com/aiven/aiven-operator/api/v1alpha1"
)
// GrafanaReconciler reconciles a Grafana object
type GrafanaReconciler struct {
Controller
}
// +kubebuilder:rbac:groups=aiven.io,resources=grafanas,verbs=get;list;watch;create;update;patch;delete
// +kubebuilder:rbac:groups=aiven.io,resources=grafanas/status,verbs=get;update;patch
// +kubebuilder:rbac:groups=aiven.io,resources=grafanas/finalizers,verbs=update
func (r *GrafanaReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
return r.reconcileInstance(ctx, req, newGenericServiceHandler(newGrafanaAdapter), &v1alpha1.Grafana{})
}
// SetupWithManager sets up the controller with the Manager.
func (r *GrafanaReconciler) SetupWithManager(mgr ctrl.Manager) error {
return ctrl.NewControllerManagedBy(mgr).
For(&v1alpha1.Grafana{}).
Owns(&corev1.Secret{}).
Complete(r)
}
func newGrafanaAdapter(object client.Object) (serviceAdapter, error) {
grafana, ok := object.(*v1alpha1.Grafana)
if !ok {
return nil, fmt.Errorf("object is not of type v1alpha1.Grafana")
}
return &grafanaAdapter{grafana}, nil
}
// grafanaAdapter handles an Aiven Grafana service
type grafanaAdapter struct {
*v1alpha1.Grafana
}
func (a *grafanaAdapter) getObjectMeta() *metav1.ObjectMeta {
return &a.ObjectMeta
}
func (a *grafanaAdapter) getServiceStatus() *v1alpha1.ServiceStatus {
return &a.Status
}
func (a *grafanaAdapter) getServiceCommonSpec() *v1alpha1.ServiceCommonSpec {
return &a.Spec.ServiceCommonSpec
}
func (a *grafanaAdapter) getUserConfig() any {
return &a.Spec.UserConfig
}
func (a *grafanaAdapter) newSecret(s *aiven.Service) *corev1.Secret {
name := a.Spec.ConnInfoSecretTarget.Name
if name == "" {
name = a.Name
}
stringData := map[string]string{
"GRAFANA_HOST": s.URIParams["host"],
"GRAFANA_PORT": s.URIParams["port"],
"GRAFANA_USER": s.URIParams["user"],
"GRAFANA_PASSWORD": s.URIParams["password"],
"GRAFANA_URI": s.URI,
"GRAFANA_HOSTS": strings.Join(s.ConnectionInfo.GrafanaURIs, ","),
}
// Removes empties
for k, v := range stringData {
if v == "" {
delete(stringData, k)
}
}
return &corev1.Secret{
ObjectMeta: metav1.ObjectMeta{Name: name, Namespace: a.Namespace},
StringData: stringData,
}
}
func (a *grafanaAdapter) newCreateRequest() aiven.CreateServiceRequest {
return aiven.CreateServiceRequest{
Cloud: a.Spec.CloudName,
MaintenanceWindow: getMaintenanceWindow(
a.Spec.MaintenanceWindowDow,
a.Spec.MaintenanceWindowTime,
),
Plan: a.Spec.Plan,
TerminationProtection: a.Spec.TerminationProtection,
ServiceName: a.Name,
ServiceType: "grafana",
ServiceIntegrations: nil,
DiskSpaceMB: v1alpha1.ConvertDiscSpace(a.Spec.DiskSpace),
}
}
func (a *grafanaAdapter) newUpdateRequest() aiven.UpdateServiceRequest {
return aiven.UpdateServiceRequest{
Cloud: a.Spec.CloudName,
MaintenanceWindow: getMaintenanceWindow(
a.Spec.MaintenanceWindowDow,
a.Spec.MaintenanceWindowTime,
),
Plan: a.Spec.Plan,
TerminationProtection: a.Spec.TerminationProtection,
Powered: true,
DiskSpaceMB: v1alpha1.ConvertDiscSpace(a.Spec.DiskSpace),
}
}