feat: set explicit dialer for custom kafka transport

getsentry · Dec 15, 2024 · ab23c00 · ab23c00
1 parent 3b7843e
commit ab23c00
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/vroom/utils.go b/cmd/vroom/utils.go
@@ -4,8 +4,10 @@ import (
 	"crypto/tls"
 	"crypto/x509"
 	"log"
+	"net"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/segmentio/kafka-go"
 	"github.com/segmentio/kafka-go/sasl"
@@ -81,5 +83,9 @@ func createKafkaRoundTripper(e ServiceConfig) kafka.RoundTripper {
 	return &kafka.Transport{
 		SASL: saslMechanism,
 		TLS:  tlsConfig,
+		Dial: (&net.Dialer{
+			Timeout:   3 * time.Second,
+			DualStack: true,
+		}).DialContext,
 	}
-}
+}