Rel #10828 - Update container index and fix airflow ingestion (#10873)

* Update container index and fix airflow ingestion

* Fix lint

---------

Co-authored-by: Ayush Shah <ayush@getcollate.io>
This commit is contained in:
Pere Miquel Brull 2023-04-06 07:09:38 +02:00 committed by GitHub
parent 38d4afb6cb
commit c53c7b680d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 11 additions and 3 deletions

View File

@ -217,7 +217,7 @@ class ContainerESDocument(BaseModel):
children: Optional[List[dict]] = None
prefix: Optional[str] = None
numberOfObjects: Optional[int] = None
size: Optional[int] = None
size: Optional[float] = None
fileFormats: Optional[List[str]] = None

View File

@ -239,7 +239,7 @@ CONTAINER_ELASTICSEARCH_INDEX_MAPPING = textwrap.dedent(
"type": "integer"
},
"size": {
"type": "integer"
"type": "long"
},
"fileFormats": {
"type": "keyword"

View File

@ -28,6 +28,9 @@ from metadata.generated.schema.entity.services.databaseService import DatabaseSe
from metadata.generated.schema.entity.services.messagingService import MessagingService
from metadata.generated.schema.entity.services.metadataService import MetadataService
from metadata.generated.schema.entity.services.mlmodelService import MlModelService
from metadata.generated.schema.entity.services.objectstoreService import (
ObjectStoreService,
)
from metadata.generated.schema.entity.services.pipelineService import PipelineService
from metadata.generated.schema.tests.testSuite import TestSuite
from metadata.ingestion.models.encoders import show_secrets_encoder
@ -169,6 +172,11 @@ def build_source(ingestion_pipeline: IngestionPipeline) -> WorkflowSource:
service: MetadataService = metadata.get_by_name(
entity=entity_class, fqn=ingestion_pipeline.service.name
)
elif service_type == "objectStoreService":
entity_class = ObjectStoreService
service: ObjectStoreService = metadata.get_by_name(
entity=entity_class, fqn=ingestion_pipeline.service.name
)
else:
raise InvalidServiceException(f"Invalid Service Type: {service_type}")
except ValidationError as original_error:

View File

@ -222,7 +222,7 @@
"type": "integer"
},
"size": {
"type": "integer"
"type": "long"
},
"fileFormats": {
"type": "keyword"