Set grpc transport request timeout (openconfig#2541)

nokia · Jan 12, 2024 · b526036 · b526036
1 parent b0c90a5
commit b526036
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/topologies/binding/binding.go b/topologies/binding/binding.go
@@ -252,8 +252,12 @@ func (a *staticATE) DialOTG(ctx context.Context, opts ...grpc.DialOption) (gosna
 	if err != nil {
 		return nil, err
 	}
+
 	api := gosnappi.NewApi()
-	api.NewGrpcTransport().SetClientConnection(conn)
+	transport := api.NewGrpcTransport().SetClientConnection(conn)
+	if timeout := a.r.dutGRPC(a.dev, dutSvcParams[introspect.OTG]).Timeout; timeout != 0 {
+		transport.SetRequestTimeout(time.Duration(timeout) * time.Second)
+	}
 	return api, nil
 }