Refactor schema owner resolution for inheritance

Co-authored-by: yourton.ma <yourton.ma@gmail.com>

Author: cursoragent

ingestion/src/metadata/ingestion/source/database/database_service.py

@@ -680,33 +680,31 @@ def get_owner_ref(self, table_name: str) -> Optional[EntityReferenceList]:
         try:
             parent_owner = None
 
-            # Get parent owner from schema's ownerConfig resolution
-            # We need to resolve the schema's owner first to enable proper inheritance
+            # Get parent owner from schema entity
+            # Since schema entity is not stored in context, we fetch it from the API
             if (
                 hasattr(self.source_config, "ownerConfig")
                 and self.source_config.ownerConfig
             ):
-                schema_fqn = f"{self.context.get().database}.{self.context.get().database_schema}"
-                
-                # First, get the database owner for schema inheritance
-                database_owner = None
-                database_entity = getattr(self.context.get(), "database_entity", None)
-                if database_entity:
-                    db_owners = database_entity.owners
-                    if db_owners and db_owners.root:
-                        database_owner = db_owners.root[0].name
-                
-                # Resolve schema owner (which may inherit from database)
-                schema_owner_ref = get_owner_from_config(
+                schema_fqn = fqn.build(
                     metadata=self.metadata,
-                    owner_config=self.source_config.ownerConfig,
-                    entity_type="databaseSchema",
-                    entity_name=schema_fqn,
-                    parent_owner=database_owner,
+                    entity_type=DatabaseSchema,
+                    service_name=self.context.get().database_service,
+                    database_name=self.context.get().database,
+                    schema_name=self.context.get().database_schema,
                 )
 
-                if schema_owner_ref and schema_owner_ref.root:
-                    parent_owner = schema_owner_ref.root[0].name
+                # Fetch schema entity to get its resolved owner
+                try:
+                    schema_entity = self.metadata.get_by_name(
+                        entity=DatabaseSchema,
+                        fqn=schema_fqn,
+                        fields=["owners"],
+                    )
+                    if schema_entity and schema_entity.owners and schema_entity.owners.root:
+                        parent_owner = schema_entity.owners.root[0].name
+                except Exception as exc:
+                    logger.debug(f"Could not fetch schema entity for owner inheritance: {exc}")
 
             table_fqn = f"{self.context.get().database}.{self.context.get().database_schema}.{table_name}"