|
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