def AcquirePluginData(self, request, context):
"""Trigger a data acquisition and store results in the data acquisition store service.
Args:
request (data_acquisition_pb2.AcquirePluginDataRequest): The data acquisition request.
context (GRPC ClientContext): tracks internal grpc statuses and information.
Returns:
An AcquirePluginDataResponse containing a request_id to use with GetStatus.
"""
response = data_acquisition_pb2.AcquirePluginDataResponse()
if self.acquire_response_fn is not None:
try:
if not self.acquire_response_fn(request, response):
return response
except Exception as e:
self.logger.exception(
'Failed during call to user acquire response function')
populate_response_header(
response,
request,
error_code=header_pb2.CommonError.CODE_INTERNAL_ERROR,
error_msg=str(e))
return response
self.request_manager.cleanup_requests()
response.request_id, state = self.request_manager.add_request()
self.logger.info('Beginning request %d for %s', response.request_id, [
capture.name
for capture in request.acquisition_requests.data_captures
])
self.executor.submit(self._data_collection_wrapper,
response.request_id, request, state)
response.status = data_acquisition_pb2.AcquireDataResponse.STATUS_OK
populate_response_header(response, request)
return response
def AcquirePluginData(self, request, context):
"""Trigger a data acquisition and store results in the data acquisition store service.
Args:
request (data_acquisition_pb2.AcquirePluginDataRequest): The data acquisition request.
context (GRPC ClientContext): tracks internal grpc statuses and information.
Returns:
An AcquirePluginDataResponse containing a request_id to use with GetStatus.
"""
response = data_acquisition_pb2.AcquirePluginDataResponse()
with ResponseContext(response, request, self.data_buffer_client):
self._start_plugin_acquire(request, response)
return response