feat: add ContextKeyDialFunc (#1534)

ory · Jun 24, 2024 · 567ceb9 · 567ceb9
1 parent e452ae2
commit 567ceb9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmd/client/grpc_client.go b/cmd/client/grpc_client.go
@@ -39,7 +39,8 @@ const (
 	EnvAuthToken   = "KETO_BEARER_TOKEN" //nolint:gosec // just the key, not the value
 	EnvAuthority   = "KETO_AUTHORITY"
 
-	ContextKeyTimeout contextKeys = "timeout"
+	ContextKeyTimeout  contextKeys = "timeout"
+	ContextKeyDialFunc contextKeys = "dial"
 )
 
 type connectionDetails struct {
@@ -137,6 +138,10 @@ func Conn(ctx context.Context, remote string, details connectionDetails) (*grpc.
 		defer cancel()
 	}
 
+	if dial, ok := ctx.Value(ContextKeyDialFunc).(func(context.Context, string) (*grpc.ClientConn, error)); ok {
+		return dial(ctx, remote)
+	}
+
 	return grpc.DialContext(
 		ctx,
 		remote,