Add BigQuery reservation option support

go/adbc/driver/bigquery/connection_test.go

@@ -265,3 +265,45 @@ func TestBuildField(t *testing.T) {
 		})
 	}
 }
+
+func TestStatementSetGetOptionReservation(t *testing.T) {
+	cnxn := &connectionImpl{}
+	st := &statement{cnxn: cnxn}
+
+	const reservation = "projects/my-project/locations/US/reservations/my-reservation"
+
+	// Initially empty
+	got, err := st.GetOption(OptionStringQueryReservation)
+	if err != nil {
+		t.Fatalf("GetOption returned error: %v", err)
+	}
+	if got != "" {
+		t.Fatalf("expected empty reservation by default, got %q", got)
+	}
+
+	// Set the reservation
+	if err := st.SetOption(OptionStringQueryReservation, reservation); err != nil {
+		t.Fatalf("SetOption returned error: %v", err)
+	}
+
+	// Round-trip via GetOption
+	got, err = st.GetOption(OptionStringQueryReservation)
+	if err != nil {
+		t.Fatalf("GetOption returned error: %v", err)
+	}
+	if got != reservation {
+		t.Fatalf("expected %q, got %q", reservation, got)
+	}
+
+	// Clear the reservation
+	if err := st.SetOption(OptionStringQueryReservation, ""); err != nil {
+		t.Fatalf("SetOption(empty) returned error: %v", err)
+	}
+	got, err = st.GetOption(OptionStringQueryReservation)
+	if err != nil {
+		t.Fatalf("GetOption returned error: %v", err)
+	}
+	if got != "" {
+		t.Fatalf("expected empty after clear, got %q", got)
+	}
+}

go/adbc/driver/bigquery/driver.go

@@ -30,12 +30,12 @@ import (
 )
 
 const (
-	OptionStringAuthType  = "adbc.bigquery.sql.auth_type"
+	OptionStringAuthType    = "adbc.bigquery.sql.auth_type"
 	OptionStringAPIEndpoint = "adbc.bigquery.sql.api_endpoint"
-	OptionStringLocation  = "adbc.bigquery.sql.location"
-	OptionStringProjectID = "adbc.bigquery.sql.project_id"
-	OptionStringDatasetID = "adbc.bigquery.sql.dataset_id"
-	OptionStringTableID   = "adbc.bigquery.sql.table_id"
+	OptionStringLocation    = "adbc.bigquery.sql.location"
+	OptionStringProjectID   = "adbc.bigquery.sql.project_id"
+	OptionStringDatasetID   = "adbc.bigquery.sql.dataset_id"
+	OptionStringTableID     = "adbc.bigquery.sql.table_id"
 
 	OptionValueAuthTypeDefault = "adbc.bigquery.sql.auth_type.auth_bigquery"
 
@@ -77,6 +77,11 @@ const (
 	OptionBoolQueryDryRun              = "adbc.bigquery.sql.query.dry_run"
 	OptionBoolQueryCreateSession       = "adbc.bigquery.sql.query.create_session"
 	OptionIntQueryJobTimeout           = "adbc.bigquery.sql.query.job_timeout"
+	// OptionStringQueryReservation specifies the BigQuery reservation to use for query
+	// execution. The value must be a full reservation resource path of the form
+	// "projects/<project>/locations/<location>/reservations/<reservation>".
+	// When set, the query job is submitted under the specified reservation.
+	OptionStringQueryReservation = "adbc.bigquery.sql.query.reservation"
 
 	OptionIntQueryResultBufferSize    = "adbc.bigquery.sql.query.result_buffer_size"
 	OptionIntQueryPrefetchConcurrency = "adbc.bigquery.sql.query.prefetch_concurrency"

go/adbc/driver/bigquery/statement.go

@@ -215,6 +215,8 @@ func (st *statement) GetOption(key string) (string, error) {
 		return strconv.FormatBool(st.queryConfig.DryRun), nil
 	case OptionBoolQueryCreateSession:
 		return strconv.FormatBool(st.queryConfig.CreateSession), nil
+	case OptionStringQueryReservation:
+		return st.queryConfig.Reservation, nil
 	case OptionBoolQueryLinkFailedJob:
 		return strconv.FormatBool(st.linkFailedJob), nil
 	case OptionBoolUseStorageApiDisabledClient:
@@ -404,6 +406,8 @@ func (st *statement) SetOption(key string, v string) error {
 		} else {
 			return err
 		}
+	case OptionStringQueryReservation:
+		st.queryConfig.Reservation = v
 	case OptionStringIngestPath:
 		st.ingestPath = v
 	case OptionStringIngestFileDelimiter: