diff --git a/csst_dfs_api_cluster/common/catalog.py b/csst_dfs_api_cluster/common/catalog.py index f8a4ad66627712ee2e86d62363cfa529e3126d4b..936d9c671ce71abc87250da8f034a7144b04fb07 100644 --- a/csst_dfs_api_cluster/common/catalog.py +++ b/csst_dfs_api_cluster/common/catalog.py @@ -28,15 +28,18 @@ class CatalogApi(object): ''' try: t_start = time.time() - resp, _ = self.stub.Gaia3Search.with_call(ephem_pb2.EphemSearchRequest( - ra = ra, - dec = dec, - radius = radius, - minMag = min_mag, - maxMag = max_mag, - obstime = obstime, - limit = limit - ),metadata = get_auth_headers()) + resp, _ = self.stub.Gaia3Search(ephem_pb2.EphemSearchRequest( + ra = ra, + dec = dec, + radius = radius, + minMag = min_mag, + maxMag = max_mag, + obstime = obstime, + limit = limit + ), + wait_for_ready = True, + metadata = get_auth_headers() + ) t_end = time.time() log.info("gaia3_query used: %.6f's" %(t_end - t_start,)) if resp.success: diff --git a/csst_dfs_api_cluster/common/service.py b/csst_dfs_api_cluster/common/service.py index 2b05da3aacc590045640d36f85bd98b3373fa43d..2202e0f96379207ffea91ba62b6165a98ea597dd 100644 --- a/csst_dfs_api_cluster/common/service.py +++ b/csst_dfs_api_cluster/common/service.py @@ -8,7 +8,7 @@ class ServiceProxy: def channel(self): options = (('grpc.max_send_message_length', 1000 * 1024 * 1024), ('grpc.max_receive_message_length', 1000 * 1024 * 1024)) - channel = grpc.insecure_channel(self.gateway, options = options, compression = grpc.Compression.Gzip) + channel = grpc.insecure_channel(self.gateway, options = options) try: grpc.channel_ready_future(channel).result(timeout=10) except grpc.FutureTimeoutError: