-
Notifications
You must be signed in to change notification settings - Fork 14
/
reconciler.go
109 lines (86 loc) · 3.92 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
/*
Copyright (c) 2021 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 awss3source
import (
"context"
"fmt"
corev1 "k8s.io/api/core/v1"
"knative.dev/pkg/reconciler"
"github.com/triggermesh/aws-event-sources/pkg/apis/sources/v1alpha1"
reconcilerv1alpha1 "github.com/triggermesh/aws-event-sources/pkg/client/generated/injection/reconciler/sources/v1alpha1/awss3source"
listersv1alpha1 "github.com/triggermesh/aws-event-sources/pkg/client/generated/listers/sources/v1alpha1"
s3client "github.com/triggermesh/aws-event-sources/pkg/client/s3"
"github.com/triggermesh/aws-event-sources/pkg/reconciler/common"
"github.com/triggermesh/aws-event-sources/pkg/reconciler/common/event"
)
// Reconciler implements controller.Reconciler for the event source type.
type Reconciler struct {
// Getter than can obtain clients for interacting with the S3 and SQS APIs
s3Cg s3client.ClientGetter
srcLister func(namespace string) listersv1alpha1.AWSS3SourceNamespaceLister
// SQS adapter
base common.GenericDeploymentReconciler
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.AWSS3Source) reconciler.Event {
// inject source into context for usage in reconciliation logic
ctx = v1alpha1.WithSource(ctx, src)
s3Client, sqsClient, err := r.s3Cg.Get(src)
if err != nil {
src.Status.MarkNotSubscribed(v1alpha1.AWSS3ReasonNoClient, "Cannot obtain AWS API clients")
return fmt.Errorf("%w", reconciler.NewEvent(corev1.EventTypeWarning, ReasonFailedSubscribe,
"Error creating AWS API clients: %s", err))
}
queueARN, err := ensureQueue(ctx, sqsClient)
if err != nil {
return fmt.Errorf("failed to reconcile SQS queue: %w", err)
}
if err := r.base.ReconcileSource(ctx, r); err != nil {
return fmt.Errorf("failed to reconcile SQS event source adapter: %w", err)
}
return r.ensureNotificationsEnabled(ctx, s3Client, queueARN)
}
// FinalizeKind is called when the resource is deleted.
func (r *Reconciler) FinalizeKind(ctx context.Context, src *v1alpha1.AWSS3Source) reconciler.Event {
// inject source into context for usage in finalization logic
ctx = v1alpha1.WithSource(ctx, src)
s3Client, sqsClient, err := r.s3Cg.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)
}
if err := ensureNoQueue(ctx, sqsClient); err != nil {
return fmt.Errorf("failed to finalize SQS queue: %w", err)
}
// The finalizer blocks the deletion of the source object until
// ensureNotificationsDisabled succeeds to ensure that we don't leave
// any dangling event notification configurations behind us.
return r.ensureNotificationsDisabled(ctx, s3Client)
}
// sourceID returns an ID that identifies the given source instance in AWS
// resources or resources tags.
func sourceID(src v1alpha1.EventSource) string {
return "io.triggermesh.awss3sources." + src.GetNamespace() + "." + src.GetName()
}