update samples

Signed-off-by: Eliott Bouhana <[email protected]>
DataDog · Oct 23, 2024 · ba08808 · ba08808
1 parent 66f4f0a
commit ba08808
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 14 deletions.
diff --git a/internal/injector/builtin/testdata/client/redis.go.snap b/internal/injector/builtin/testdata/client/redis.go.snap
diff --git a/samples/client/redis.go b/samples/client/redis.go
@@ -8,6 +8,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"time"
 
 	redisv0 "github.com/go-redis/redis"
 	redisv7 "github.com/go-redis/redis/v7"
@@ -65,7 +66,11 @@ func redigoClient(ctx context.Context, net, addr string) {
 		use(conn)
 	}
 
-	if conn, err := redigo.DialURL(fmt.Sprintf("%s://%s", net, addr)); err != nil {
+	options := []redigo.DialOption{
+		redigo.DialConnectTimeout(5 * time.Second),
+	}
+
+	if conn, err := redigo.DialURL(fmt.Sprintf("%s://%s", net, addr), options...); err != nil {
 		panic(err)
 	} else {
 		use(conn)