feat: add job_timeout_ms to job configuration classes

google/cloud/bigquery/job/base.py

@@ -171,6 +171,33 @@ def __setattr__(self, name, value):
  )
  super(_JobConfig, self).__setattr__(name, value)
 
+ @property
+ def job_timeout_ms(self):
+ """Optional parameter. Job timeout in milliseconds. If this time limit is exceeded, BigQuery might attempt to stop the job.
+ https://cloud.google.com/bigquery/docs/reference/rest/v2/Job#JobConfiguration.FIELDS.job_timeout_ms
+ e.g.
+
+ job_config = bigquery.QueryJobConfig( jobtimeout = 5000 )
+ or
+ job_config.jobtimeout = 5000
+
+ Raises:
+ ValueError: If ``value`` type is invalid.
+ """
+
+ # None as this is an optional parameter.
+ if self._properties.get("jobTimeoutMs"):
+ return self._properties["jobTimeoutMs"]
+ return None
+
+ @job_timeout_ms.setter
+ def job_timeout_ms(self, value):
+ if not isinstance(value, int):
+ raise ValueError("Pass an int for jobTimeoutMs, e.g. 5000")
+
+ """ Docs indicate a string is expected by the API """
+ self._properties["jobTimeoutMs"] = str(value)
+
  @property
  def labels(self):
  """Dict[str, str]: Labels for the job.

tests/system/test_query.py

@@ -97,6 +97,22 @@ def test_query_w_timeout(bigquery_client, query_api_method):
  assert bigquery_client.cancel_job(query_job) is not None
 
 
+def test_job_timeout_ms():
+ # Confirm that default status is None.
+ job_config = bigquery.QueryJobConfig()
+ assert job_config.job_timeout_ms is None
+
+ # Confirm that integers get converted to strings.
+ job_config.job_timeout_ms = 5000
+ assert job_config.job_timeout_ms == "5000" # int is converted to string
+
+
+def test_job_timeout_ms_raises_valueerror():
+ # Confirm that attempting to set a non-integer values will raise an Error.
+ with pytest.raises(ValueError):
+ job_config.job_timeout_ms = "WillRaiseError"
+
+
 def test_query_statistics(bigquery_client, query_api_method):
  """
  A system test to exercise some of the extended query statistics.