Skip to content

Add DatasetProvider interface for BigQuery Table API access #4

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
djmantri wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Add DatasetProvider interface for BigQuery Table API access #4

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
31 changes: 30 additions & 1 deletion driver/connection.go
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,12 +1,38 @@
package driver

import (
"cloud.google.com/go/bigquery"
"context"
"database/sql/driver"
"fmt"

"cloud.google.com/go/bigquery"
)

// DatasetProvider is an interface that provides access to the underlying
// BigQuery dataset. This interface can be used to access the BigQuery client
// for operations that are not supported by the standard database/sql interface,
// such as updating table clustering.
//
// Usage with database/sql:
//
// conn, err := db.Conn(ctx)
// if err != nil {
// return err
// }
// defer conn.Close()
//
// err = conn.Raw(func(driverConn interface{}) error {
// if provider, ok := driverConn.(driver.DatasetProvider); ok {
// dataset := provider.GetDataset()
// // do operations on the dataset
// }
// return errors.New("connection does not implement DatasetProvider")
// })
type DatasetProvider interface {
// GetDataset returns the BigQuery dataset for the current connection.
GetDataset() *bigquery.Dataset
}

type bigQueryConnection struct {
ctx context.Context
client *bigquery.Client
Expand All @@ -16,6 +42,9 @@ type bigQueryConnection struct {
dataset *bigquery.Dataset
}

// Ensure bigQueryConnection implements DatasetProvider
var _ DatasetProvider = (*bigQueryConnection)(nil)

func (connection *bigQueryConnection) GetDataset() *bigquery.Dataset {
if connection.dataset != nil {
return connection.dataset
Expand Down