Skip to content

Commit 0149cf7

Browse files
authored
fix: respect timeout setting in grpc method calls (open-feature#60)
Signed-off-by: Federico Bond <[email protected]>
1 parent 6508234 commit 0149cf7

File tree

1 file changed

+6
-5
lines changed
  • providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd

1 file changed

+6
-5
lines changed

providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/provider.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -133,32 +133,33 @@ def _resolve(
133133
evaluation_context: EvaluationContext,
134134
):
135135
context = self._convert_context(evaluation_context)
136+
call_args = {"timeout": self.timeout}
136137
try:
137138
if flag_type == FlagType.BOOLEAN:
138139
request = schema_pb2.ResolveBooleanRequest(
139140
flag_key=flag_key, context=context
140141
)
141-
response = self.stub.ResolveBoolean(request)
142+
response = self.stub.ResolveBoolean(request, **call_args)
142143
elif flag_type == FlagType.STRING:
143144
request = schema_pb2.ResolveStringRequest(
144145
flag_key=flag_key, context=context
145146
)
146-
response = self.stub.ResolveString(request)
147+
response = self.stub.ResolveString(request, **call_args)
147148
elif flag_type == FlagType.OBJECT:
148149
request = schema_pb2.ResolveObjectRequest(
149150
flag_key=flag_key, context=context
150151
)
151-
response = self.stub.ResolveObject(request)
152+
response = self.stub.ResolveObject(request, **call_args)
152153
elif flag_type == FlagType.FLOAT:
153154
request = schema_pb2.ResolveFloatRequest(
154155
flag_key=flag_key, context=context
155156
)
156-
response = self.stub.ResolveFloat(request)
157+
response = self.stub.ResolveFloat(request, **call_args)
157158
elif flag_type == FlagType.INTEGER:
158159
request = schema_pb2.ResolveIntRequest(
159160
flag_key=flag_key, context=context
160161
)
161-
response = self.stub.ResolveInt(request)
162+
response = self.stub.ResolveInt(request, **call_args)
162163
else:
163164
raise ValueError(f"Unknown flag type: {flag_type}")
164165

0 commit comments

Comments
 (0)