context.WithTimeout 1s on ping

ozontech · Jul 17, 2024 · 1820cc9 · 1820cc9
1 parent c47f1e1
commit 1820cc9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/plugin/input/kafka/client.go b/plugin/input/kafka/client.go
@@ -2,6 +2,7 @@ package kafka
 
 import (
 	"context"
+	"time"
 
 	"github.com/ozontech/file.d/cfg"
 	"github.com/twmb/franz-go/pkg/kgo"
@@ -47,7 +48,11 @@ func NewClient(c *Config, l *zap.Logger) *kgo.Client {
 	if err != nil {
 		l.Fatal("can't create kafka client", zap.Error(err))
 	}
-	err = client.Ping(context.TODO())
+
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+	defer cancel()
+
+	err = client.Ping(ctx)
 	if err != nil {
 		l.Fatal("can't connect to kafka", zap.Error(err))
 	}

diff --git a/plugin/output/kafka/client.go b/plugin/output/kafka/client.go
@@ -51,7 +51,10 @@ func NewClient(c *Config, l *zap.Logger) *kgo.Client {
 		l.Fatal("can't create kafka client", zap.Error(err))
 	}
 
-	err = client.Ping(context.TODO())
+	ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+	defer cancel()
+
+	err = client.Ping(ctx)
 	if err != nil {
 		l.Fatal("can't connect to kafka", zap.Error(err))
 	}