Skip to content

Commit

Permalink
fix(ingest/bigquery): show report in output (#8867)
Browse files Browse the repository at this point in the history
Co-authored-by: Mayuri Nehate <[email protected]>
Co-authored-by: Andrew Sikowitz <[email protected]>
  • Loading branch information
3 people authored Sep 22, 2023
1 parent aff1e7a commit c946c01
Showing 1 changed file with 13 additions and 10 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@

import pydantic

from datahub.ingestion.api.report import Report
from datahub.ingestion.source.sql.sql_generic_profiler import ProfilingSqlReport
from datahub.ingestion.source_report.ingestion_stage import IngestionStageReport
from datahub.ingestion.source_report.time_window import BaseTimeWindowReport
Expand All @@ -16,18 +17,20 @@
logger: logging.Logger = logging.getLogger(__name__)


class BigQuerySchemaApiPerfReport:
list_projects = PerfTimer()
list_datasets = PerfTimer()
get_columns_for_dataset = PerfTimer()
get_tables_for_dataset = PerfTimer()
list_tables = PerfTimer()
get_views_for_dataset = PerfTimer()
@dataclass
class BigQuerySchemaApiPerfReport(Report):
list_projects: PerfTimer = field(default_factory=PerfTimer)
list_datasets: PerfTimer = field(default_factory=PerfTimer)
get_columns_for_dataset: PerfTimer = field(default_factory=PerfTimer)
get_tables_for_dataset: PerfTimer = field(default_factory=PerfTimer)
list_tables: PerfTimer = field(default_factory=PerfTimer)
get_views_for_dataset: PerfTimer = field(default_factory=PerfTimer)


class BigQueryAuditLogApiPerfReport:
get_exported_log_entries = PerfTimer()
list_log_entries = PerfTimer()
@dataclass
class BigQueryAuditLogApiPerfReport(Report):
get_exported_log_entries: PerfTimer = field(default_factory=PerfTimer)
list_log_entries: PerfTimer = field(default_factory=PerfTimer)


@dataclass
Expand Down

0 comments on commit c946c01

Please sign in to comment.