|
2 | 2 | import types
|
3 | 3 | import grpc
|
4 | 4 | from google.protobuf.json_format import ParseDict
|
5 |
| -from google.protobuf.message_factory import MessageFactory #, GetMessageClass |
| 5 | +from google.protobuf.message_factory import MessageFactory # , GetMessageClass |
6 | 6 | from google.protobuf.descriptor_pool import DescriptorPool
|
7 | 7 | from google.protobuf.descriptor import ServiceDescriptor, MethodDescriptor
|
8 | 8 | from grpc_reflection.v1alpha.proto_reflection_descriptor_database import (
|
@@ -109,16 +109,24 @@ def _retry_call(
|
109 | 109 | ):
|
110 | 110 | retries = 0
|
111 | 111 |
|
| 112 | + _LOGGER.debug( |
| 113 | + f"client._retry_call.init :: client_call_details: {client_call_details}" |
| 114 | + ) |
| 115 | + |
112 | 116 | while True:
|
113 | 117 | try:
|
| 118 | + _LOGGER.debug(f"client._retry_call.start!!!") |
114 | 119 | response_or_iterator = continuation(
|
115 | 120 | client_call_details, request_or_iterator
|
116 | 121 | )
|
| 122 | + _LOGGER.debug(f"client._retry_call.response_or_iterator Success!!!") |
117 | 123 |
|
118 | 124 | if is_stream:
|
119 | 125 | response_or_iterator = self._generate_response(response_or_iterator)
|
120 | 126 | else:
|
| 127 | + _LOGGER.debug(f"client._retry_call.start _check_error!!!") |
121 | 128 | self._check_error(response_or_iterator)
|
| 129 | + _LOGGER.debug(f"client._retry_call.finished _check_error!!!") |
122 | 130 |
|
123 | 131 | return response_or_iterator
|
124 | 132 |
|
@@ -207,13 +215,17 @@ def _bind_grpc_method(
|
207 | 215 | request_desc = self._desc_pool.FindMessageTypeByName(
|
208 | 216 | method_desc.input_type.full_name
|
209 | 217 | )
|
210 |
| - request_message_desc = MessageFactory(self._desc_pool).GetPrototype(request_desc) |
| 218 | + request_message_desc = MessageFactory(self._desc_pool).GetPrototype( |
| 219 | + request_desc |
| 220 | + ) |
211 | 221 | # request_message_desc = GetMessageClass(request_desc)
|
212 | 222 |
|
213 | 223 | response_desc = self._desc_pool.FindMessageTypeByName(
|
214 | 224 | method_desc.output_type.full_name
|
215 | 225 | )
|
216 |
| - response_message_desc = MessageFactory(self._desc_pool).GetPrototype(response_desc) |
| 226 | + response_message_desc = MessageFactory(self._desc_pool).GetPrototype( |
| 227 | + response_desc |
| 228 | + ) |
217 | 229 | # response_message_desc = GetMessageClass(response_desc)
|
218 | 230 |
|
219 | 231 | if method_desc.client_streaming and method_desc.server_streaming:
|
@@ -286,7 +298,9 @@ def _init_grpc_reflection(self):
|
286 | 298 | request_desc = self._desc_pool.FindMessageTypeByName(
|
287 | 299 | method_desc.input_type.full_name
|
288 | 300 | )
|
289 |
| - self._request_map[method_key] = MessageFactory(self._desc_pool).GetPrototype(request_desc) |
| 301 | + self._request_map[method_key] = MessageFactory( |
| 302 | + self._desc_pool |
| 303 | + ).GetPrototype(request_desc) |
290 | 304 | # self._request_map[method_key] = GetMessageClass(request_desc)
|
291 | 305 |
|
292 | 306 | if service_desc.name not in self._api_resources:
|
@@ -324,6 +338,11 @@ def client(endpoint=None, ssl_enabled=False, max_message_length=None, **client_o
|
324 | 338 | if endpoint is None:
|
325 | 339 | raise Exception("Client's endpoint is undefined.")
|
326 | 340 |
|
| 341 | + _LOGGER.debug(f"pygrpc.client :: endpoint: {endpoint}") |
| 342 | + _LOGGER.debug( |
| 343 | + f"pygrpc.client :: _GRPC_CHANNEL: {_GRPC_CHANNEL} / is in? {endpoint in _GRPC_CHANNEL}" |
| 344 | + ) |
| 345 | + |
327 | 346 | if endpoint not in _GRPC_CHANNEL:
|
328 | 347 | options = []
|
329 | 348 |
|
|
0 commit comments