Fix #5379 - Fix service type handling on REST APIs (#5380)

This commit is contained in:
Pere Miquel Brull 2022-06-08 20:36:10 +02:00 committed by GitHub
parent e851242a1a
commit 3cbd0bc1d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -71,13 +71,13 @@ def build_source(ingestion_pipeline: IngestionPipeline) -> WorkflowSource:
entity=PipelineService, fqn=ingestion_pipeline.service.name
)
elif service_type == "dashboardService":
service: MessagingService = metadata.get_by_name(
entity=MessagingService, fqn=ingestion_pipeline.service.name
)
elif service_type == "messagingService":
service: DashboardService = metadata.get_by_name(
entity=DashboardService, fqn=ingestion_pipeline.service.name
)
elif service_type == "messagingService":
service: MessagingService = metadata.get_by_name(
entity=MessagingService, fqn=ingestion_pipeline.service.name
)
if not service:
raise ValueError(f"Could not get service from type {service_type}")