@@ -54,13 +54,15 @@ func OpenTopic(logger log.Logger, cfg *service.KafkaConfig) (*pubsub.Topic, erro
54
54
55
55
switch cfg .SASLMechanism {
56
56
case "AWS_MSK_IAM" :
57
- config .Net .SASL .Mechanism = sarama .SASLTypeOAuth
58
- config .Net .SASL .TokenProvider = & MSKAccessTokenProvider {
59
- Region : cfg .AWSRegion ,
60
- Profile : cfg .AWSProfile ,
61
- RoleARN : cfg .AWSRoleARN ,
62
- SessionName : cfg .AWSSessionName ,
57
+ if cfg .AWS != nil {
58
+ config .Net .SASL .TokenProvider = & MSKAccessTokenProvider {
59
+ Region : cfg .AWS .Region ,
60
+ Profile : cfg .AWS .Profile ,
61
+ RoleARN : cfg .AWS .RoleARN ,
62
+ SessionName : cfg .AWS .SessionName ,
63
+ }
63
64
}
65
+ config .Net .SASL .Mechanism = sarama .SASLTypeOAuth
64
66
config .Net .TLS .Enable = true
65
67
config .Net .TLS .Config = & tls.Config {}
66
68
@@ -82,7 +84,6 @@ func OpenTopic(logger log.Logger, cfg *service.KafkaConfig) (*pubsub.Topic, erro
82
84
Set ("sasl.mechanism" , log .String (string (config .Net .SASL .Mechanism ))).
83
85
Set ("sasl.user" , log .String (cfg .Key )).
84
86
Set ("topic" , log .String (cfg .Topic )).
85
- Set ("aws.region" , log .String (cfg .AWSRegion )).
86
87
Log ("opening kafka topic" )
87
88
88
89
return kafkapubsub .OpenTopic (cfg .Brokers , config , cfg .Topic , nil )
@@ -97,13 +98,15 @@ func OpenSubscription(logger log.Logger, cfg *service.KafkaConfig) (*pubsub.Subs
97
98
// Default to PLAIN if no SASL mechanism is specified
98
99
switch cfg .SASLMechanism {
99
100
case "AWS_MSK_IAM" :
100
- config .Net .SASL .Mechanism = sarama .SASLTypeOAuth
101
- config .Net .SASL .TokenProvider = & MSKAccessTokenProvider {
102
- Region : cfg .AWSRegion ,
103
- Profile : cfg .AWSProfile ,
104
- RoleARN : cfg .AWSRoleARN ,
105
- SessionName : cfg .AWSSessionName ,
101
+ if cfg .AWS != nil {
102
+ config .Net .SASL .TokenProvider = & MSKAccessTokenProvider {
103
+ Region : cfg .AWS .Region ,
104
+ Profile : cfg .AWS .Profile ,
105
+ RoleARN : cfg .AWS .RoleARN ,
106
+ SessionName : cfg .AWS .SessionName ,
107
+ }
106
108
}
109
+ config .Net .SASL .Mechanism = sarama .SASLTypeOAuth
107
110
config .Net .TLS .Enable = true
108
111
config .Net .TLS .Config = & tls.Config {}
109
112
@@ -129,7 +132,6 @@ func OpenSubscription(logger log.Logger, cfg *service.KafkaConfig) (*pubsub.Subs
129
132
Set ("sasl.mechanism" , log .String (string (config .Net .SASL .Mechanism ))).
130
133
Set ("sasl.user" , log .String (cfg .Key )).
131
134
Set ("topic" , log .String (cfg .Topic )).
132
- Set ("aws.region" , log .String (cfg .AWSRegion )).
133
135
Log ("setting up kafka subscription" )
134
136
135
137
return kafkapubsub .OpenSubscription (cfg .Brokers , config , cfg .Group , []string {cfg .Topic }, & kafkapubsub.SubscriptionOptions {
0 commit comments