yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-06 16:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Postgres usage not terminating with bad connection (#19707)

Browse Source 
* postgres usage not terminating with bad connection

* Added Operational error

---------

Co-authored-by: Akash Verma <akashverma@Akashs-MacBook-Pro-2.local>
This commit is contained in:
Akash Verma 2025-02-11 12:25:51 +05:30 committed by GitHub
parent 372953fc88
commit fcbe9241bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ingestion/src/metadata/ingestion/source/database/postgres/usage.py
View File

@ -15,6 +15,11 @@ import traceback
from datetime import datetime from datetime import datetime
from typing import Iterable from typing import Iterable
from sqlalchemy.exc import OperationalError
from metadata.generated.schema.entity.services.ingestionPipelines.status import (
StackTraceError,
)
from metadata.generated.schema.type.basic import DateTime from metadata.generated.schema.type.basic import DateTime
from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery
from metadata.ingestion.source.connections import get_connection from metadata.ingestion.source.connections import get_connection
@ -67,6 +72,16 @@ class PostgresUsageSource(PostgresQueryParserSource, UsageSource):
logger.error(str(err)) logger.error(str(err))
if queries: if queries:
yield TableQueries(queries=queries) yield TableQueries(queries=queries)
except OperationalError as err:
self.status.failed(
StackTraceError(
name="Usage",
error=f"Source Usage failed due to - {err}",
stackTrace=traceback.format_exc(),
)
)
except Exception as err: except Exception as err:
if query: if query:
logger.debug( logger.debug(