yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-28 20:00:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix usage count issue (#13109)

Browse Source
This commit is contained in:
Mayur Singal 2023-09-08 10:36:29 +05:30 committed by GitHub
parent 5d780dc8eb
commit ba09f874df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ingestion/src/metadata/ingestion/processor/query_parser.py
View File

@ -20,6 +20,7 @@ from metadata.config.common import ConfigModel
from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
OpenMetadataConnection,
)
from metadata.generated.schema.type.basic import DateTime
from metadata.generated.schema.type.queryParserData import ParsedData, QueryParserData
from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery
from metadata.ingestion.api.models import Either, StackTraceError
@ -41,11 +42,10 @@ def parse_sql_statement(record: TableQuery, dialect: Dialect) -> Optional[Parsed
:return: QueryParserData
"""
start_date = record.analysisDate
if isinstance(record.analysisDate, str):
start_date = datetime.datetime.strptime(
str(record.analysisDate), "%Y-%m-%d %H:%M:%S"
).date()
start_time = record.analysisDate
if isinstance(start_time, DateTime):
start_date = start_time.__root__.date()
start_time = datetime.datetime.strptime(str(start_date.isoformat()), "%Y-%m-%d")
lineage_parser = LineageParser(record.query, dialect=dialect)
@ -61,7 +61,7 @@ def parse_sql_statement(record: TableQuery, dialect: Dialect) -> Optional[Parsed
query_type=record.query_type,
exclude_usage=record.exclude_usage,
userName=record.userName,
date=int(start_date.__root__.timestamp()),
date=int(start_time.timestamp()),
serviceName=record.serviceName,
duration=record.duration,
)