Skip to content

Commit

Permalink
feat(bigquery): support config for region qualifiers to fetch jobs (d…
Browse files Browse the repository at this point in the history
  • Loading branch information
mayurinehate authored Oct 28, 2024
1 parent 25a4eef commit e90d744
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -282,6 +282,7 @@ def get_workunits_internal(self) -> Iterable[MetadataWorkUnit]:
include_usage_statistics=self.config.include_usage_statistics,
include_operations=self.config.usage.include_operational_stats,
top_n_queries=self.config.usage.top_n_queries,
region_qualifiers=self.config.region_qualifiers,
),
structured_report=self.report,
filters=self.filters,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -527,6 +527,12 @@ def have_table_data_read_permission(self) -> bool:
" Set to 1 to disable.",
)

region_qualifiers: List[str] = Field(
default=["region-us", "region-eu"],
description="BigQuery regions to be scanned for bigquery jobs when using `use_queries_v2`. "
"See [this](https://cloud.google.com/bigquery/docs/information-schema-jobs#scope_and_syntax) for details.",
)

# include_view_lineage and include_view_column_lineage are inherited from SQLCommonConfig
# but not used in bigquery so we hide them from docs.
include_view_lineage: bool = Field(default=True, hidden_from_docs=True)
Expand Down

0 comments on commit e90d744

Please sign in to comment.