Set read/write/metadata timeouts to the provider configured timeout

Mongey · Apr 8, 2024 · 40dacea · 40dacea
1 parent a6f2ce6
commit 40dacea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kafka/config.go b/kafka/config.go
@@ -115,6 +115,10 @@ func (c *Config) newKafkaConfig() (*sarama.Config, error) {
 	kafkaConfig.Net.Proxy.Enable = true
 	kafkaConfig.Net.Proxy.Dialer = proxy.FromEnvironment()
 
+	kafkaConfig.Net.ReadTimeout = time.Duration(c.Timeout) * time.Second
+	kafkaConfig.Net.WriteTimeout = time.Duration(c.Timeout) * time.Second
+	kafkaConfig.Metadata.Timeout = time.Duration(c.Timeout) * time.Second
+
 	if c.saslEnabled() {
 		switch c.SASLMechanism {
 		case "scram-sha512":