Commit 560df5a2 authored by Wei Shoulin's avatar Wei Shoulin
Browse files

timeit

parent 01d770ad
import grpc import grpc
import time
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
...@@ -7,6 +9,7 @@ from .service import ServiceProxy ...@@ -7,6 +9,7 @@ from .service import ServiceProxy
from .constants import * from .constants import *
from .utils import get_auth_headers from .utils import get_auth_headers
log = logging.getLogger('csst')
class CatalogApi(object): class CatalogApi(object):
def __init__(self): def __init__(self):
self.stub = ephem_pb2_grpc.EphemSearchSrvStub(ServiceProxy().channel()) self.stub = ephem_pb2_grpc.EphemSearchSrvStub(ServiceProxy().channel())
...@@ -36,6 +39,10 @@ class CatalogApi(object): ...@@ -36,6 +39,10 @@ class CatalogApi(object):
),metadata = get_auth_headers()) ),metadata = get_auth_headers())
if resp.success: if resp.success:
t_start = time.time()
data = from_proto_model_list(Gaia3Record, resp.records)
t_end = time.time()
log.info("object deserialization used: %.6f's" %(t_end - t_start,))
return Result.ok_data(data=from_proto_model_list(Gaia3Record, resp.records)).append("totalCount", resp.totalCount) return Result.ok_data(data=from_proto_model_list(Gaia3Record, resp.records)).append("totalCount", resp.totalCount)
else: else:
return Result.error(message = str(resp.error.detail)) return Result.error(message = str(resp.error.detail))
......
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