def get_executions_by_context(
self, context_id: int) -> List[metadata_store_pb2.Execution]:
"""Gets all direct executions that a context associates with.
Args:
context_id: The id of the querying context
Returns:
Executions associating with the context.
"""
request = metadata_store_service_pb2.GetExecutionsByContextRequest()
request.context_id = context_id
request.options.max_result_size = 100
request.options.order_by_field.field = (
metadata_store_pb2.ListOperationOptions.OrderByField.Field.
CREATE_TIME)
request.options.order_by_field.is_asc = False
result = []
while True:
response = metadata_store_service_pb2.GetExecutionsByContextResponse(
)
self._call('GetExecutionsByContext', request, response)
for x in response.executions:
result.append(x)
if not response.next_page_token:
break
request.options.next_page_token = response.next_page_token
return result
def _get_executions_by_pipeline_name(
self, pipeline_name: Text) -> List[metadata_store_pb2.Execution]:
"""Helper function returns executions under a given pipeline name."""
# step 1: get context id by context name
request = metadata_store_service_pb2.GetContextByTypeAndNameRequest(
type_name='pipeline', context_name=pipeline_name)
context_id = self._stub.GetContextByTypeAndName(request).context.id
# step 2: get executions by context id
request = metadata_store_service_pb2.GetExecutionsByContextRequest(
context_id=context_id)
return self._stub.GetExecutionsByContext(request).executions
def get_executions_by_context(
self, context_id: int) -> List[metadata_store_pb2.Execution]:
"""Gets all direct executions that a context associates with.
Args:
context_id: The id of the querying context
Returns:
Executions associating with the context.
"""
request = metadata_store_service_pb2.GetExecutionsByContextRequest()
request.context_id = context_id
response = metadata_store_service_pb2.GetExecutionsByContextResponse()
self._call('GetExecutionsByContext', request, response)
result = []
for x in response.executions:
result.append(x)
return result
