-
Notifications
You must be signed in to change notification settings - Fork 69
/
reconciler.go
127 lines (100 loc) · 4.55 KB
/
reconciler.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
/*
Copyright 2022 TriggerMesh 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 awseventbridgesource
import (
"context"
"fmt"
corev1 "k8s.io/api/core/v1"
"knative.dev/pkg/reconciler"
commonv1alpha1 "github.com/triggermesh/triggermesh/pkg/apis/common/v1alpha1"
"github.com/triggermesh/triggermesh/pkg/apis/sources/v1alpha1"
reconcilerv1alpha1 "github.com/triggermesh/triggermesh/pkg/client/generated/injection/reconciler/sources/v1alpha1/awseventbridgesource"
listersv1alpha1 "github.com/triggermesh/triggermesh/pkg/client/generated/listers/sources/v1alpha1"
common "github.com/triggermesh/triggermesh/pkg/reconciler"
"github.com/triggermesh/triggermesh/pkg/reconciler/event"
eventbridgeclient "github.com/triggermesh/triggermesh/pkg/sources/client/eventbridge"
)
// Reconciler implements controller.Reconciler for the event source type.
type Reconciler struct {
// Getter than can obtain clients for interacting with the EventBridge and SQS APIs
ebCg eventbridgeclient.ClientGetter
// SQS adapter
base common.GenericDeploymentReconciler[*v1alpha1.AWSEventBridgeSource, listersv1alpha1.AWSEventBridgeSourceNamespaceLister]
adapterCfg *adapterConfig
}
// Check that our Reconciler implements Interface
var _ reconcilerv1alpha1.Interface = (*Reconciler)(nil)
// Check that our Reconciler implements Finalizer
var _ reconcilerv1alpha1.Finalizer = (*Reconciler)(nil)
// ReconcileKind implements Interface.ReconcileKind.
func (r *Reconciler) ReconcileKind(ctx context.Context, src *v1alpha1.AWSEventBridgeSource) reconciler.Event {
// inject source into context for usage in reconciliation logic
ctx = commonv1alpha1.WithReconcilable(ctx, src)
ebClient, sqsClient, err := r.ebCg.Get(src)
if err != nil {
src.Status.MarkNotSubscribed(v1alpha1.AWSEventBridgeReasonNoClient, "Cannot obtain AWS API clients")
return fmt.Errorf("%w", reconciler.NewEvent(corev1.EventTypeWarning, ReasonFailedSubscribe,
"Error creating AWS API clients: %s", err))
}
queue, err := EnsureQueue(ctx, sqsClient)
if err != nil {
return fmt.Errorf("failed to reconcile SQS queue: %w", err)
}
ruleARN, err := EnsureRule(ctx, ebClient, queue)
if err != nil {
return fmt.Errorf("failed to reconcile event rule: %w", err)
}
if err := EnsureQueuePolicy(ctx, sqsClient, queue, ruleARN); err != nil {
return fmt.Errorf("failed to apply access policy to SQS queue: %w", err)
}
if err := SetRuleTarget(ctx, ebClient, ruleARN, queue.ARN); err != nil {
return fmt.Errorf("failed to set target on event rule: %w", err)
}
if err := r.base.ReconcileAdapter(ctx, r); err != nil {
return fmt.Errorf("failed to reconcile SQS event source adapter: %w", err)
}
return nil
}
// FinalizeKind is called when the resource is deleted.
func (r *Reconciler) FinalizeKind(ctx context.Context, src *v1alpha1.AWSEventBridgeSource) reconciler.Event {
// inject source into context for usage in finalization logic
ctx = commonv1alpha1.WithReconcilable(ctx, src)
ebClient, sqsClient, err := r.ebCg.Get(src)
switch {
case isNotFound(err):
// the finalizer is unlikely to recover from a missing Secret,
// so we simply record a warning event and return
event.Warn(ctx, ReasonFailedUnsubscribe,
"Secret missing while finalizing event source. Ignoring: %s", err)
return nil
case err != nil:
return reconciler.NewEvent(corev1.EventTypeWarning, ReasonFailedUnsubscribe,
"Error creating AWS API clients: %s", err)
}
// The finalizer blocks the deletion of the source object until
// ensureNoRule and ensureQueue succeed to ensure that we don't leave
// any dangling EventBridge event rule behind us.
queueName, err := EnsureNoRule(ctx, ebClient, sqsClient)
if err != nil {
return fmt.Errorf("failed to finalize event rule: %w", err)
}
if err := EnsureNoQueue(ctx, sqsClient, queueName); err != nil {
return fmt.Errorf("failed to finalize SQS queue: %w", err)
}
return nil
}
// sourceID returns an ID that identifies the given source instance in AWS
// resources or resources tags.
func sourceID(src commonv1alpha1.Reconcilable) string {
return "io.triggermesh.awseventbridgesources." + src.GetNamespace() + "." + src.GetName()
}