65 lines
2 KiB
Go
65 lines
2 KiB
Go
|
// Copyright 2015 Google Inc. All Rights Reserved.
|
||
|
//
|
||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
// you may not use this file except in compliance with the License.
|
||
|
// You may obtain a copy of the License at
|
||
|
//
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
//
|
||
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
// See the License for the specific language governing permissions and
|
||
|
// limitations under the License.
|
||
|
|
||
|
package bigquery
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
|
||
|
"golang.org/x/net/context"
|
||
|
)
|
||
|
|
||
|
func (conf *readTableConf) fetch(ctx context.Context, s service, token string) (*readDataResult, error) {
|
||
|
return s.readTabledata(ctx, conf, token)
|
||
|
}
|
||
|
|
||
|
func (conf *readTableConf) setPaging(pc *pagingConf) { conf.paging = *pc }
|
||
|
|
||
|
// Read fetches the contents of the table.
|
||
|
func (t *Table) Read(ctx context.Context) *RowIterator {
|
||
|
return newRowIterator(ctx, t.c.service, &readTableConf{
|
||
|
projectID: t.ProjectID,
|
||
|
datasetID: t.DatasetID,
|
||
|
tableID: t.TableID,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (conf *readQueryConf) fetch(ctx context.Context, s service, token string) (*readDataResult, error) {
|
||
|
return s.readQuery(ctx, conf, token)
|
||
|
}
|
||
|
|
||
|
func (conf *readQueryConf) setPaging(pc *pagingConf) { conf.paging = *pc }
|
||
|
|
||
|
// Read fetches the results of a query job.
|
||
|
// If j is not a query job, Read returns an error.
|
||
|
func (j *Job) Read(ctx context.Context) (*RowIterator, error) {
|
||
|
if !j.isQuery {
|
||
|
return nil, errors.New("Cannot read from a non-query job")
|
||
|
}
|
||
|
return newRowIterator(ctx, j.service, &readQueryConf{
|
||
|
projectID: j.projectID,
|
||
|
jobID: j.jobID,
|
||
|
}), nil
|
||
|
}
|
||
|
|
||
|
// Read submits a query for execution and returns the results via a RowIterator.
|
||
|
// It is a shorthand for Query.Run followed by Job.Read.
|
||
|
func (q *Query) Read(ctx context.Context) (*RowIterator, error) {
|
||
|
job, err := q.Run(ctx)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return job.Read(ctx)
|
||
|
}
|