From 17c5ba8698c2bcbb4d1f89d34030891731b21a1a Mon Sep 17 00:00:00 2001 From: Pere Miquel Brull Date: Thu, 12 Dec 2024 10:40:03 +0100 Subject: [PATCH] MINOR - Add APIs to Ref Map (#19019) * MINOR - Add APIs to Ref Map * MINOR - Add APIs to Ref Map * MINOR - Add APIs to Ref Map * format --- ingestion/src/metadata/utils/constants.py | 4 ++++ ingestion/src/metadata/utils/elasticsearch.py | 6 ++++++ 2 files changed, 10 insertions(+) diff --git a/ingestion/src/metadata/utils/constants.py b/ingestion/src/metadata/utils/constants.py index 49a0e7c5eef..e9a20bcce42 100644 --- a/ingestion/src/metadata/utils/constants.py +++ b/ingestion/src/metadata/utils/constants.py @@ -12,6 +12,8 @@ """ Define constants useful for the metadata ingestion """ +from metadata.generated.schema.entity.data.apiCollection import APICollection +from metadata.generated.schema.entity.data.apiEndpoint import APIEndpoint from metadata.generated.schema.entity.data.chart import Chart from metadata.generated.schema.entity.data.container import Container from metadata.generated.schema.entity.data.dashboard import Dashboard @@ -131,6 +133,8 @@ ENTITY_REFERENCE_CLASS_MAP = { "metadataService": MetadataService, "searchService": SearchService, # Data Asset Entities + "apiCollection": APICollection, + "apiEndpoint": APIEndpoint, "table": Table, "storedProcedure": StoredProcedure, "database": Database, diff --git a/ingestion/src/metadata/utils/elasticsearch.py b/ingestion/src/metadata/utils/elasticsearch.py index 0f0ab014b05..14bbe94ab02 100644 --- a/ingestion/src/metadata/utils/elasticsearch.py +++ b/ingestion/src/metadata/utils/elasticsearch.py @@ -18,6 +18,8 @@ from pydantic import BaseModel from metadata.generated.schema.analytics.reportData import ReportData from metadata.generated.schema.entity.classification.tag import Tag +from metadata.generated.schema.entity.data.apiCollection import APICollection +from metadata.generated.schema.entity.data.apiEndpoint import APIEndpoint from metadata.generated.schema.entity.data.chart import Chart from metadata.generated.schema.entity.data.container import Container from metadata.generated.schema.entity.data.dashboard import Dashboard @@ -33,6 +35,7 @@ from metadata.generated.schema.entity.data.searchIndex import SearchIndex from metadata.generated.schema.entity.data.storedProcedure import StoredProcedure from metadata.generated.schema.entity.data.table import Table from metadata.generated.schema.entity.data.topic import Topic +from metadata.generated.schema.entity.services.apiService import ApiService from metadata.generated.schema.entity.services.databaseService import DatabaseService from metadata.generated.schema.entity.teams.team import Team from metadata.generated.schema.entity.teams.user import User @@ -42,6 +45,9 @@ logger = utils_logger() T = TypeVar("T", bound=BaseModel) ES_INDEX_MAP = { + ApiService.__name__: "api_service_search_index", + APICollection.__name__: "api_collection_search_index", + APIEndpoint.__name__: "api_endpoint_search_index", Table.__name__: "table_search_index", Database.__name__: "database_search_index", DatabaseSchema.__name__: "database_schema_search_index",