yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-07-29 20:30:19 +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 ( from metadata.generated.schema.entity.services.connections.metadata.openMetadataConnection import (
OpenMetadataConnection, OpenMetadataConnection,
) )
from metadata.generated.schema.type.basic import DateTime
from metadata.generated.schema.type.queryParserData import ParsedData, QueryParserData from metadata.generated.schema.type.queryParserData import ParsedData, QueryParserData
from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery from metadata.generated.schema.type.tableQuery import TableQueries, TableQuery
from metadata.ingestion.api.models import Either, StackTraceError from metadata.ingestion.api.models import Either, StackTraceError
@ -41,11 +42,10 @@ def parse_sql_statement(record: TableQuery, dialect: Dialect) -> Optional[Parsed
:return: QueryParserData :return: QueryParserData
""" """
start_date = record.analysisDate start_time = record.analysisDate
if isinstance(record.analysisDate, str): if isinstance(start_time, DateTime):
start_date = datetime.datetime.strptime( start_date = start_time.__root__.date()
str(record.analysisDate), "%Y-%m-%d %H:%M:%S" start_time = datetime.datetime.strptime(str(start_date.isoformat()), "%Y-%m-%d")
).date()
lineage_parser = LineageParser(record.query, dialect=dialect) 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, query_type=record.query_type,
exclude_usage=record.exclude_usage, exclude_usage=record.exclude_usage,
userName=record.userName, userName=record.userName,
date=int(start_date.__root__.timestamp()), date=int(start_time.timestamp()),
serviceName=record.serviceName, serviceName=record.serviceName,
duration=record.duration, duration=record.duration,
) )