bq

Documentation

Index

• type Client
  • func New(ctx context.Context, projectID types.GoogleProjectID) (*Client, error)
  • func (x *Client) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, ...) error
  • func (x *Client) GetMetadata(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID) (*bigquery.TableMetadata, error)
  • func (x *Client) Insert(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) error
  • func (x *Client) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) (interfaces.BigQueryStream, error)
  • func (x *Client) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)
  • func (x *Client) UpdateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, ...) error
• type GeneralMock
  • func NewGeneralMock() *GeneralMock
  • func (x *GeneralMock) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, ...) error
  • func (x *GeneralMock) GetMetadata(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID) (*bigquery.TableMetadata, error)
  • func (x *GeneralMock) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) (interfaces.BigQueryStream, error)
  • func (x *GeneralMock) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)
  • func (x *GeneralMock) UpdateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, ...) error
• type Mock
  • func NewMock() *Mock
  • func (x *Mock) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, ...) error
  • func (x *Mock) GetMetadata(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID) (*bigquery.TableMetadata, error)
  • func (x *Mock) Insert(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) error
  • func (x *Mock) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) (interfaces.BigQueryStream, error)
  • func (x *Mock) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)
  • func (x *Mock) UpdateTable(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, ...) error
• type MockInsertedData
• type MockStream
  • func (x *MockStream) Close() error
  • func (x *MockStream) Insert(ctx context.Context, data []any) error
• type Stream
  • func (x *Stream) Close() error
  • func (x *Stream) Insert(ctx context.Context, data []any) error

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

func New

func New(ctx context.Context, projectID types.GoogleProjectID) (*Client, error)

func (*Client) CreateTable

func (x *Client) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md *bigquery.TableMetadata) error

CreateTable implements interfaces.BigQuery.

func (*Client) GetMetadata

func (x *Client) GetMetadata(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID) (*bigquery.TableMetadata, error)

GetMetadata implements interfaces.BigQuery. If the table does not exist, it returns nil.

func (*Client) Insert

func (x *Client) Insert(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, schema bigquery.Schema, data []any) error

func (*Client) NewStream

func (x *Client) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, schema bigquery.Schema) (interfaces.BigQueryStream, error)

func (*Client) Query

func (x *Client) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)

Query implements interfaces.BigQuery.

func (*Client) UpdateTable

func (x *Client) UpdateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md bigquery.TableMetadataToUpdate, eTag string) error

UpdateSchema implements interfaces.BigQuery.

type GeneralMock

type GeneralMock struct {
	Metadata []*bigquery.TableMetadata

	OpenedStream []struct {
		Dataset types.BQDatasetID
		Table   types.BQTableID
		Schema  bigquery.Schema
	}
	Streams []*MockStream

	CreatedTable []struct {
		Dataset types.BQDatasetID
		Table   types.BQTableID
		MD      *bigquery.TableMetadata
	}
	UpdatedTable []struct {
		Dataset types.BQDatasetID
		Table   types.BQTableID
		MD      bigquery.TableMetadataToUpdate
		ETag    string
	}

	Queries []string
	// contains filtered or unexported fields
}

func NewGeneralMock

func NewGeneralMock() *GeneralMock

func (*GeneralMock) CreateTable

func (x *GeneralMock) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md *bigquery.TableMetadata) error

CreateTable implements interfaces.BigQuery.

func (*GeneralMock) GetMetadata

func (x *GeneralMock) GetMetadata(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID) (*bigquery.TableMetadata, error)

GetMetadata implements interfaces.BigQuery.

func (*GeneralMock) NewStream

func (x *GeneralMock) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, schema bigquery.Schema) (interfaces.BigQueryStream, error)

func (*GeneralMock) Query

func (x *GeneralMock) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)

Query implements interfaces.BigQuery.

func (*GeneralMock) UpdateTable

func (x *GeneralMock) UpdateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md bigquery.TableMetadataToUpdate, eTag string) error

UpdateTable implements interfaces.BigQuery.

type Mock

type Mock struct {
	MockQuery       func(ctx context.Context, query string) (interfaces.BigQueryIterator, error)
	MockInsert      (func(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, data []any) error)
	MockGetMetadata (func(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID) (*bigquery.TableMetadata, error))
	MockUpdateTable (func(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, md bigquery.TableMetadataToUpdate, eTag string) error)
	MockCreateTable (func(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md *bigquery.TableMetadata) error)
}

func NewMock

func NewMock() *Mock

func (*Mock) CreateTable

func (x *Mock) CreateTable(ctx context.Context, dataset types.BQDatasetID, table types.BQTableID, md *bigquery.TableMetadata) error

CreateTable implements interfaces.BigQuery.

func (*Mock) GetMetadata

func (x *Mock) GetMetadata(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID) (*bigquery.TableMetadata, error)

func (*Mock) Insert

func (x *Mock) Insert(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, schema bigquery.Schema, data []any) error

func (*Mock) NewStream

func (x *Mock) NewStream(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, schema bigquery.Schema) (interfaces.BigQueryStream, error)

func (*Mock) Query

func (x *Mock) Query(ctx context.Context, query string) (interfaces.BigQueryIterator, error)

func (*Mock) UpdateTable

func (x *Mock) UpdateTable(ctx context.Context, datasetID types.BQDatasetID, tableID types.BQTableID, md bigquery.TableMetadataToUpdate, eTag string) error

type MockInsertedData

type MockInsertedData struct {
	DatasetID types.BQDatasetID
	TableID   types.BQTableID
	Schema    bigquery.Schema
	Data      []any
}

type MockStream

type MockStream struct {
	Inserted [][]any
	// contains filtered or unexported fields
}

func (*MockStream) Close

func (x *MockStream) Close() error

func (*MockStream) Insert

func (x *MockStream) Insert(ctx context.Context, data []any) error

type Stream

type Stream struct {
	// contains filtered or unexported fields
}

func (*Stream) Close

func (x *Stream) Close() error

func (*Stream) Insert

func (x *Stream) Insert(ctx context.Context, data []any) error