Add kafka SASL version control to kafka_consumer (#6350)

This commit is contained in:
Daniel Nelson
2020-01-02 16:27:26 -08:00
committed by GitHub
parent 1edb73916f
commit 2486006495
7 changed files with 137 additions and 7 deletions

View File

@@ -0,0 +1,25 @@
package kafka
import (
"errors"
"github.com/Shopify/sarama"
)
func SASLVersion(kafkaVersion sarama.KafkaVersion, saslVersion *int) (int16, error) {
if saslVersion == nil {
if kafkaVersion.IsAtLeast(sarama.V1_0_0_0) {
return sarama.SASLHandshakeV1, nil
}
return sarama.SASLHandshakeV0, nil
}
switch *saslVersion {
case 0:
return sarama.SASLHandshakeV0, nil
case 1:
return sarama.SASLHandshakeV1, nil
default:
return 0, errors.New("invalid SASL version")
}
}