feat: search statistics

google/cloud/bigquery/job/query.py

@@ -198,6 +198,39 @@ def from_api_repr(cls, stats: Dict[str, str]) -> "DmlStats":
         return cls(*args)
 
 
+class SearchReason(typing.NamedTuple):
+    """docstring"""
+
+    code: Optional[str] = ""
+    message: Optional[str] = ""
+    baseTable: Optional[TableReference] = None
+    indexName: Optional[str] = ""
+
+    @classmethod
+    def from_api_repr(cls, reason):
+        code = reason.get("code")
+        message = reason.get("message")
+        baseTable = reason.get("baseTable")
+        indexName = reason.get("indexName")
+
+        return cls(code, message, baseTable, indexName)
+
+
+class SearchStats(typing.NamedTuple):
+    """docstring"""
+
+    mode: str = ""
+    reason: List[SearchReason] = []
+
+    @classmethod
+    def from_api_repr(cls, stats: Dict[str, Any]):
+        mode = stats.get("indexUsageMode", "")
+        reason = [
+            SearchReason.from_api_repr(r) for r in stats.get("indexUnusedReasons", [])
+        ]
+        return cls(mode, reason)
+
+
 class ScriptOptions:
     """Options controlling the execution of scripts.
 
@@ -724,7 +757,6 @@ def to_api_repr(self) -> dict:
             Dict: A dictionary in the format used by the BigQuery API.
         """
         resource = copy.deepcopy(self._properties)
-
         # Query parameters have an addition property associated with them
         # to indicate if the query is using named or positional parameters.
         query_parameters = resource["query"].get("queryParameters")
@@ -858,6 +890,14 @@ def priority(self):
         """
         return self.configuration.priority
 
+    @property
+    def search_stats(self):
+        """docstring"""
+
+        stats = self._job_statistics().get("searchStatistics")
+        if stats is not None:
+            return SearchStats.from_api_repr(stats)
+
     @property
     def query(self):
         """str: The query text used in this query job.

tests/unit/job/test_query.py

@@ -911,6 +911,28 @@ def test_dml_stats(self):
         assert isinstance(job.dml_stats, DmlStats)
         assert job.dml_stats.inserted_row_count == 35
 
+    def test_search_stats(self):
+        from google.cloud.bigquery.job.query import SearchStats
+
+        client = _make_client(project=self.PROJECT)
+        job = self._make_one(self.JOB_ID, self.QUERY, client)
+        assert job.search_stats is None
+
+        statistics = job._properties["statistics"] = {}
+        assert job.search_stats is None
+
+        query_stats = statistics["query"] = {}
+        assert job.search_stats is None
+
+        query_stats["searchStatistics"] = {
+            "indexUsageMode": "INDEX_USAGE_MODE_UNSPECIFIED",
+            "indexUnusedReasons": [],
+        }
+        # job.search_stats is a daisy-chain of calls and gets:
+        # job.search_stats << job._job_statistics << job._properties
+        assert isinstance(job.search_stats, SearchStats)
+        assert job.search_stats.mode == "INDEX_USAGE_MODE_UNSPECIFIED"
+
     def test_result(self):
         from google.cloud.bigquery.table import RowIterator
 

tests/unit/job/test_query_stats.py

@@ -108,6 +108,75 @@ def test_from_api_repr_full_stats(self):
         assert result.updated_row_count == 4
 
 
+class TestSearchStatistics:
+    @staticmethod
+    def _get_target_class():
+        from google.cloud.bigquery.job.query import SearchStats
+
+        return SearchStats
+
+    def _make_one(self, *args, **kwargs):
+        return self._get_target_class()(*args, **kwargs)
+
+    def test_ctor_defaults(self):
+        search_stats = self._make_one()
+        assert search_stats.mode == "INDEX_USAGE_MODE_UNSPECIFIED"
+        assert search_stats.reason == []
+
+    def test_from_api_repr_unspecified(self):
+        klass = self._get_target_class()
+        result = klass.from_api_repr(
+            {"indexUsageMode": "INDEX_USAGE_MODE_UNSPECIFIED", "indexUnusedReasons": []}
+        )
+
+        assert isinstance(result, klass)
+        assert result.mode == "INDEX_USAGE_MODE_UNSPECIFIED"
+        assert result.reason == []
+
+
+class TestSearchReasons:
+    @staticmethod
+    def _get_target_class():
+        from google.cloud.bigquery.job.query import SearchReason
+
+        return SearchReason
+
+    def _make_one(self, *args, **kwargs):
+        return self._get_target_class()(*args, **kwargs)
+
+    def test_ctor_defaults(self):
+        search_reason = self._make_one()
+        assert search_reason.code == ""
+        assert search_reason.message == ""
+        assert search_reason.baseTable == None
+        assert search_reason.indexName == ""
+
+    def test_from_api_repr_unspecified(self):
+        klass = self._get_target_class()
+        result = klass.from_api_repr(
+            {
+                "code": "INDEX_CONFIG_NOT_AVAILABLE",
+                "message": "There is no search index...",
+                "baseTable": {
+                    "projectId": "bigquery-public-data",
+                    "datasetId": "usa_names",
+                    "tableId": "usa_1910_current",
+                },
+                "indexName": None,
+            }
+        )
+
+        assert isinstance(result, klass)
+        assert result.code == "INDEX_CONFIG_NOT_AVAILABLE"
+        assert result.message == "There is no search index..."
+        assert result.baseTable == {
+            "projectId": "bigquery-public-data",
+            "datasetId": "usa_names",
+            "tableId": "usa_1910_current",
+        }
+        assert result.indexName == None
+
+
 class TestQueryPlanEntryStep(_Base):
     KIND = "KIND"
     SUBSTEPS = ("SUB1", "SUB2")