Add version config to kafka utils (#188)

carousell · Sep 19, 2022 · 053997a · 053997a
1 parent dafc9bf
commit 053997a
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/utils/kafka/config.go b/utils/kafka/config.go
@@ -67,3 +67,18 @@ func WithErrorHandler(errorHandler func(error)) Option {
 		errorHandler: errorHandler,
 	}
 }
+
+type versionOption struct {
+	version sarama.KafkaVersion
+}
+
+func (opt *versionOption) apply(cfg *config) {
+	cfg.saramaConfig.Version = opt.version
+}
+
+// WithVersion specified the kafka cluster version to connect to
+func WithVersion(version sarama.KafkaVersion) Option {
+	return &versionOption{
+		version: version,
+	}
+}