feat: Enable ES Search for Databases (#19041)

* 🎉 Init

* linter

* bring back removed code for api collection

* remove comment

* fix type hint

---------

Co-authored-by: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
This commit is contained in:
mgorsk1 2025-08-06 20:21:09 +02:00 committed by GitHub
parent d31e2d8ba0
commit 956a13f3f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -230,15 +230,31 @@ def _(
@fqn_build_registry.add(Database) @fqn_build_registry.add(Database)
def _( def _(
_: Optional[OpenMetadata], # ES Search not enabled for Databases metadata: Optional[OpenMetadata],
*, *,
service_name: str, service_name: str,
database_name: str, database_name: str,
) -> str: skip_es_search: bool = True,
fetch_multiple_entities: bool = False,
) -> Union[Optional[str], Optional[List[str]]]:
if not skip_es_search:
entity = search_database_from_es(
metadata,
database_name,
service_name,
fetch_multiple_entities=fetch_multiple_entities,
)
if entity and fetch_multiple_entities:
return [str(database.fullyQualifiedName.root) for database in entity]
if entity:
return str(entity.fullyQualifiedName.root)
if not service_name or not database_name: if not service_name or not database_name:
raise FQNBuildingException( raise FQNBuildingException(
f"Args should be informed, but got service=`{service_name}`, db=`{database_name}``" f"Args should be informed, but got service=`{service_name}`, db=`{database_name}``"
) )
return _build(service_name, database_name) return _build(service_name, database_name)