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
This commit is contained in:
Pere Miquel Brull 2024-12-12 10:40:03 +01:00 committed by GitHub
parent c24bc870d0
commit 17c5ba8698
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 0 deletions

View File

@ -12,6 +12,8 @@
""" """
Define constants useful for the metadata ingestion 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.chart import Chart
from metadata.generated.schema.entity.data.container import Container from metadata.generated.schema.entity.data.container import Container
from metadata.generated.schema.entity.data.dashboard import Dashboard from metadata.generated.schema.entity.data.dashboard import Dashboard
@ -131,6 +133,8 @@ ENTITY_REFERENCE_CLASS_MAP = {
"metadataService": MetadataService, "metadataService": MetadataService,
"searchService": SearchService, "searchService": SearchService,
# Data Asset Entities # Data Asset Entities
"apiCollection": APICollection,
"apiEndpoint": APIEndpoint,
"table": Table, "table": Table,
"storedProcedure": StoredProcedure, "storedProcedure": StoredProcedure,
"database": Database, "database": Database,

View File

@ -18,6 +18,8 @@ from pydantic import BaseModel
from metadata.generated.schema.analytics.reportData import ReportData from metadata.generated.schema.analytics.reportData import ReportData
from metadata.generated.schema.entity.classification.tag import Tag 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.chart import Chart
from metadata.generated.schema.entity.data.container import Container from metadata.generated.schema.entity.data.container import Container
from metadata.generated.schema.entity.data.dashboard import Dashboard 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.storedProcedure import StoredProcedure
from metadata.generated.schema.entity.data.table import Table from metadata.generated.schema.entity.data.table import Table
from metadata.generated.schema.entity.data.topic import Topic 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.services.databaseService import DatabaseService
from metadata.generated.schema.entity.teams.team import Team from metadata.generated.schema.entity.teams.team import Team
from metadata.generated.schema.entity.teams.user import User from metadata.generated.schema.entity.teams.user import User
@ -42,6 +45,9 @@ logger = utils_logger()
T = TypeVar("T", bound=BaseModel) T = TypeVar("T", bound=BaseModel)
ES_INDEX_MAP = { 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", Table.__name__: "table_search_index",
Database.__name__: "database_search_index", Database.__name__: "database_search_index",
DatabaseSchema.__name__: "database_schema_search_index", DatabaseSchema.__name__: "database_schema_search_index",