Commit de5cf475 authored by Wei Shoulin's avatar Wei Shoulin
Browse files

ephem to stream resp

parent d244700e
import grpc import grpc
import time from collections import deque
import logging import logging
from csst_dfs_commons.models import Result from csst_dfs_commons.models import Result
from csst_dfs_commons.models.common import from_proto_model_list, Gaia3Record from csst_dfs_commons.models.common import from_proto_model_list, Gaia3Record
...@@ -27,7 +27,7 @@ class CatalogApi(object): ...@@ -27,7 +27,7 @@ class CatalogApi(object):
return: csst_dfs_common.models.Result return: csst_dfs_common.models.Result
''' '''
try: try:
records = [] records = deque([])
totalCount = 0 totalCount = 0
for resp in self.stub.Gaia3Search(ephem_pb2.EphemSearchRequest( for resp in self.stub.Gaia3Search(ephem_pb2.EphemSearchRequest(
ra = ra, ra = ra,
...@@ -44,7 +44,6 @@ class CatalogApi(object): ...@@ -44,7 +44,6 @@ class CatalogApi(object):
totalCount = resp.totalCount totalCount = resp.totalCount
else: else:
return Result.error(message = str(resp.error.detail)) return Result.error(message = str(resp.error.detail))
return Result.ok_data(data = records).append("totalCount", totalCount) return Result.ok_data(data = list(records)).append("totalCount", totalCount)
except grpc.RpcError as e: except grpc.RpcError as e:
return Result.error(message="%s:%s" % (e.code().value, e.details())) return Result.error(message="%s:%s" % (e.code().value, e.details()))
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment