def list_jobs(
self,
include_terminated: bool,
project: Optional[str] = None,
table_name: Optional[str] = None,
) -> List[SparkJob]:
"""
List ingestion jobs currently running in Feast.
Args:
include_terminated: Flag to include terminated jobs or not
project: Optionally specify the project to use as filter when retrieving jobs
table_name: Optionally specify name of feature table to use as filter when retrieving jobs
Returns:
List of SparkJob ingestion jobs.
"""
if not self._use_job_service:
return list_jobs(include_terminated, self, project, table_name)
else:
request = ListJobsRequest(
include_terminated=include_terminated,
project=project,
table_name=cast(str, table_name),
)
response = self._job_service.ListJobs(request)
return [
get_remote_job_from_proto(self._job_service,
self._feast._extra_grpc_params, job)
for job in response.jobs
]
def get_job_by_id(self, job_id: str) -> SparkJob:
if not self._use_job_service:
return get_job_by_id(job_id, self)
else:
request = GetJobRequest(job_id=job_id)
response = self._job_service.GetJob(request)
return get_remote_job_from_proto(self._job_service,
self._feast._extra_grpc_params,
response.job)
def list_jobs(self,
include_terminated: bool,
table_name: Optional[str] = None) -> List[SparkJob]:
if not self._use_job_service:
return list_jobs(include_terminated, self, table_name)
else:
request = ListJobsRequest(include_terminated=include_terminated,
table_name=table_name)
response = self._job_service.ListJobs(request)
return [
get_remote_job_from_proto(self._job_service,
self._feast._extra_grpc_params, job)
for job in response.jobs
]