-
Notifications
You must be signed in to change notification settings - Fork 112
/
eventsender.go
149 lines (126 loc) · 5.03 KB
/
eventsender.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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
package eventsender
import (
"context"
"fmt"
"strings"
ce "github.com/cloudevents/sdk-go/v2"
"github.com/go-logr/logr"
apicommon "github.com/keptn/lifecycle-toolkit/lifecycle-operator/apis/lifecycle/v1/common"
"github.com/keptn/lifecycle-toolkit/lifecycle-operator/controllers/common/config"
"github.com/keptn/lifecycle-toolkit/lifecycle-operator/controllers/lifecycle/interfaces"
"golang.org/x/exp/maps"
"k8s.io/client-go/tools/record"
"sigs.k8s.io/controller-runtime/pkg/client"
)
//go:generate moq -pkg fake -skip-ensure -out ./fake/event_mock.go . IEvent:MockEvent
type IEvent interface {
Emit(phase apicommon.KeptnPhaseType, eventType string, reconcileObject client.Object, status string, message string, version string)
}
// ===== Main =====
type EventMultiplexer struct {
logger logr.Logger
emitters []IEvent
}
func NewEventMultiplexer(logger logr.Logger, recorder record.EventRecorder, client ce.Client) *EventMultiplexer {
multiplexer := &EventMultiplexer{
logger: logger,
}
multiplexer.register(newCloudEventSender(logger, client))
multiplexer.register(NewK8sSender(recorder))
return multiplexer
}
func (e *EventMultiplexer) register(emitter IEvent) {
if emitter != nil {
e.emitters = append(e.emitters, emitter)
}
}
func (e *EventMultiplexer) Emit(phase apicommon.KeptnPhaseType, eventType string, reconcileObject client.Object, status string, message string, version string) {
for _, emitter := range e.emitters {
e.logger.Info(fmt.Sprintf("Emitting event using %T", emitter))
go emitter.Emit(phase, eventType, reconcileObject, status, message, version)
}
}
// ===== Cloud Event Sender =====
type cloudEvent struct {
client ce.Client
logger logr.Logger
}
func newCloudEventSender(logger logr.Logger, client ce.Client) *cloudEvent {
return &cloudEvent{
client: client,
logger: logger,
}
}
// Emit creates a Cloud Event and send it to the endpoint
func (e *cloudEvent) Emit(phase apicommon.KeptnPhaseType, eventType string, reconcileObject client.Object, status string, message string, version string) {
endpoint := config.Instance().GetCloudEventsEndpoint()
if endpoint == "" {
// if no endpoint is configured we don't emit any event
if !strings.HasPrefix(endpoint, "http") {
e.logger.V(5).Info(fmt.Sprintf("CloudEvent endpoint configured but it does not start with http: %s", endpoint))
}
return
}
event := ce.NewEvent()
event.SetSource("keptn.sh")
event.SetType(fmt.Sprintf("%s.%s", phase.LongName, status))
msg := setEventMessage(phase, reconcileObject, message, version)
err := event.SetData(ce.ApplicationJSON, map[string]interface{}{
"message": msg,
"type": eventType,
"version": version,
"resource": map[string]string{
"group": reconcileObject.GetObjectKind().GroupVersionKind().Group,
"kind": reconcileObject.GetObjectKind().GroupVersionKind().Kind,
"version": reconcileObject.GetObjectKind().GroupVersionKind().Version,
"name": reconcileObject.GetName(),
"namespace": reconcileObject.GetNamespace(),
},
})
if err != nil {
e.logger.V(5).Info(fmt.Sprintf("Failed to set data for CloudEvent: %v", err))
return
}
ctx := ce.ContextWithTarget(context.TODO(), endpoint)
if result := e.client.Send(ctx, event); ce.IsUndelivered(result) {
e.logger.V(5).Info(fmt.Sprintf("Failed to send CloudEvent: %v", event))
}
}
// ===== K8s Event Sender =====
type k8sEvent struct {
recorder record.EventRecorder
}
func NewK8sSender(recorder record.EventRecorder) IEvent {
return &k8sEvent{
recorder: recorder,
}
}
// Emit creates k8s Event and adds it to Eventqueue
func (e *k8sEvent) Emit(phase apicommon.KeptnPhaseType, eventType string, reconcileObject client.Object, status string, message string, version string) {
msg := setEventMessage(phase, reconcileObject, message, version)
annotations := setAnnotations(reconcileObject, phase)
e.recorder.AnnotatedEventf(reconcileObject, annotations, eventType, fmt.Sprintf("%s%s", phase.ShortName, status), msg)
}
func setEventMessage(phase apicommon.KeptnPhaseType, reconcileObject client.Object, longReason string, version string) string {
if version == "" {
return fmt.Sprintf("%s: %s / Namespace: %s, Name: %s", phase.LongName, longReason, reconcileObject.GetNamespace(), reconcileObject.GetName())
}
return fmt.Sprintf("%s: %s / Namespace: %s, Name: %s, Version: %s", phase.LongName, longReason, reconcileObject.GetNamespace(), reconcileObject.GetName(), version)
}
func setAnnotations(reconcileObject client.Object, phase apicommon.KeptnPhaseType) map[string]string {
if reconcileObject == nil || reconcileObject.GetName() == "" || reconcileObject.GetNamespace() == "" {
return nil
}
annotations := map[string]string{
"namespace": reconcileObject.GetNamespace(),
"name": reconcileObject.GetName(),
"phase": phase.ShortName,
}
piWrapper, err := interfaces.NewEventObjectWrapperFromClientObject(reconcileObject)
if err == nil {
maps.Copy(annotations, piWrapper.GetEventAnnotations())
}
annotationsObject := reconcileObject.GetAnnotations()
annotations["traceparent"] = annotationsObject["traceparent"]
return annotations
}