/
adapter.go
102 lines (85 loc) · 3.65 KB
/
adapter.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
/*
Copyright 2023 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 azureservicebussource
import (
"strconv"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
"knative.dev/eventing/pkg/reconciler/source"
"knative.dev/pkg/apis"
commonv1alpha1 "github.com/triggermesh/triggermesh/pkg/apis/common/v1alpha1"
"github.com/triggermesh/triggermesh/pkg/apis/sources/v1alpha1"
common "github.com/triggermesh/triggermesh/pkg/reconciler"
"github.com/triggermesh/triggermesh/pkg/reconciler/resource"
)
// adapterConfig contains properties used to configure the source's adapter.
// These are automatically populated by envconfig.
type adapterConfig struct {
// Container image
// Uses a common adapter for both Azure Service Bus sources instead of a source-specific image.
Image string `envconfig:"AZURESERVICEBUSSOURCE_IMAGE" default:"gcr.io/triggermesh/azureservicebussource-adapter"`
// Configuration accessor for logging/metrics/tracing
configs source.ConfigAccessor
}
// Verify that Reconciler implements common.AdapterBuilder.
var _ common.AdapterBuilder[*appsv1.Deployment] = (*Reconciler)(nil)
// BuildAdapter implements common.AdapterBuilder.
func (r *Reconciler) BuildAdapter(src commonv1alpha1.Reconcilable, sinkURI *apis.URL) (*appsv1.Deployment, error) {
typedSrc := src.(*v1alpha1.AzureServiceBusSource)
return common.NewAdapterDeployment(src, sinkURI,
resource.Image(r.adapterCfg.Image),
resource.EnvVars(MakeAppEnv(typedSrc)...),
resource.EnvVars(r.adapterCfg.configs.ToEnvVars()...),
), nil
}
// MakeAppEnv extracts environment variables from the object.
// Exported to be used in external tools for local test environments.
func MakeAppEnv(o *v1alpha1.AzureServiceBusSource) []corev1.EnvVar {
var webSocketsEnable bool
if wss := o.Spec.WebSocketsEnable; wss != nil {
webSocketsEnable = *wss
}
var appEnvs []corev1.EnvVar
//TODO should this be here?
if sasAuth := o.Spec.Auth.SASToken; sasAuth != nil {
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvServiceBusKeyName, sasAuth.KeyName)
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvServiceBusKeyValue, sasAuth.KeyValue)
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvServiceBusConnStr, sasAuth.ConnectionString)
}
if spAuth := o.Spec.Auth.ServicePrincipal; spAuth != nil {
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvAADTenantID, spAuth.TenantID)
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvAADClientID, spAuth.ClientID)
appEnvs = common.MaybeAppendValueFromEnvVar(appEnvs, common.EnvAADClientSecret, spAuth.ClientSecret)
}
if o.Spec.MaxConcurrent != nil {
appEnvs = append(appEnvs, corev1.EnvVar{
Name: common.EnvServiceBusMaxConcurrent,
Value: strconv.Itoa(*o.Spec.MaxConcurrent),
})
}
var resourceID string
if o.Spec.TopicID != nil {
if sID := o.Status.SubscriptionID; sID != nil {
resourceID = sID.String()
}
} else if o.Spec.QueueID != nil {
resourceID = o.Spec.QueueID.String()
}
return append(appEnvs, []corev1.EnvVar{{
Name: common.EnvServiceBusEntityResourceID,
Value: resourceID,
}, {
Name: common.EnvServiceBusWebSocketsEnable,
Value: strconv.FormatBool(webSocketsEnable),
}}...)
}