vendor: update all dependencies to latest versions

vendor/cloud.google.com/go/bigquery/job.go

@@ -16,12 +16,17 @@ package bigquery
 
 import (
 	"errors"
+	"fmt"
+	"math/rand"
+	"os"
+	"sync"
 	"time"
 
 	"cloud.google.com/go/internal"
 	gax "github.com/googleapis/gax-go"
 	"golang.org/x/net/context"
 	bq "google.golang.org/api/bigquery/v2"
+	"google.golang.org/api/iterator"
 )
 
 // A Job represents an operation which has been submitted to BigQuery for processing.
@@ -46,6 +51,7 @@ func (c *Client) JobFromID(ctx context.Context, id string) (*Job, error) {
 	return job, nil
 }
 
+// ID returns the job's ID.
 func (j *Job) ID() string {
 	return j.jobID
 }
@@ -54,7 +60,8 @@ func (j *Job) ID() string {
 type State int
 
 const (
-	Pending State = iota
+	StateUnspecified State = iota // used only as a default in JobIterator
+	Pending
 	Running
 	Done
 )
@@ -73,21 +80,43 @@ type JobStatus struct {
 	Statistics *JobStatistics
 }
 
-// setJobRef initializes job's JobReference if given a non-empty jobID.
+// createJobRef creates a JobReference.
 // projectID must be non-empty.
-func setJobRef(job *bq.Job, jobID, projectID string) {
+func createJobRef(jobID string, addJobIDSuffix bool, projectID string) *bq.JobReference {
 	if jobID == "" {
-		return
+		jobID = randomJobIDFn()
+	} else if addJobIDSuffix {
+		jobID += "-" + randomJobIDFn()
 	}
 	// We don't check whether projectID is empty; the server will return an
 	// error when it encounters the resulting JobReference.
-
-	job.JobReference = &bq.JobReference{
+	return &bq.JobReference{
 		JobId:     jobID,
 		ProjectId: projectID,
 	}
 }
 
+const alphanum = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
+
+var (
+	rngMu sync.Mutex
+	rng   = rand.New(rand.NewSource(time.Now().UnixNano() ^ int64(os.Getpid())))
+)
+
+// For testing.
+var randomJobIDFn = randomJobID
+
+func randomJobID() string {
+	// As of August 2017, the BigQuery service uses 27 alphanumeric characters for suffixes.
+	var b [27]byte
+	rngMu.Lock()
+	for i := 0; i < len(b); i++ {
+		b[i] = alphanum[rng.Intn(len(alphanum))]
+	}
+	rngMu.Unlock()
+	return string(b[:])
+}
+
 // Done reports whether the job has completed.
 // After Done returns true, the Err method will return an error if the job completed unsuccesfully.
 func (s *JobStatus) Done() bool {
@@ -99,20 +128,25 @@ func (s *JobStatus) Err() error {
 	return s.err
 }
 
+// Fill in the client field of Tables in the statistics.
+func (s *JobStatus) setClient(c *Client) {
+	if s.Statistics == nil {
+		return
+	}
+	if qs, ok := s.Statistics.Details.(*QueryStatistics); ok {
+		for _, t := range qs.ReferencedTables {
+			t.c = c
+		}
+	}
+}
+
 // Status returns the current status of the job. It fails if the Status could not be determined.
 func (j *Job) Status(ctx context.Context) (*JobStatus, error) {
 	js, err := j.c.service.jobStatus(ctx, j.projectID, j.jobID)
 	if err != nil {
 		return nil, err
 	}
-	// Fill in the client field of Tables in the statistics.
-	if js.Statistics != nil {
-		if qs, ok := js.Statistics.Details.(*QueryStatistics); ok {
-			for _, t := range qs.ReferencedTables {
-				t.c = j.c
-			}
-		}
-	}
+	js.setClient(j.c)
 	return js, nil
 }
 
@@ -324,3 +358,73 @@ type ExplainQueryStep struct {
 func (*ExtractStatistics) implementsStatistics() {}
 func (*LoadStatistics) implementsStatistics()    {}
 func (*QueryStatistics) implementsStatistics()   {}
+
+// Jobs lists jobs within a project.
+func (c *Client) Jobs(ctx context.Context) *JobIterator {
+	it := &JobIterator{
+		ctx:       ctx,
+		c:         c,
+		ProjectID: c.projectID,
+	}
+	it.pageInfo, it.nextFunc = iterator.NewPageInfo(
+		it.fetch,
+		func() int { return len(it.items) },
+		func() interface{} { b := it.items; it.items = nil; return b })
+	return it
+}
+
+// A JobInfo consists of a Job and a JobStatus.
+type JobInfo struct {
+	Job    *Job
+	Status *JobStatus
+}
+
+// JobIterator iterates over jobs in a project.
+type JobIterator struct {
+	ProjectID string // Project ID of the jobs to list. Default is the client's project.
+	AllUsers  bool   // Whether to list jobs owned by all users in the project, or just the current caller.
+	State     State  // List only jobs in the given state. Defaults to all states.
+
+	ctx      context.Context
+	c        *Client
+	pageInfo *iterator.PageInfo
+	nextFunc func() error
+	items    []JobInfo
+}
+
+func (it *JobIterator) PageInfo() *iterator.PageInfo { return it.pageInfo }
+
+func (it *JobIterator) Next() (JobInfo, error) {
+	if err := it.nextFunc(); err != nil {
+		return JobInfo{}, err
+	}
+	item := it.items[0]
+	it.items = it.items[1:]
+	return item, nil
+}
+
+func (it *JobIterator) fetch(pageSize int, pageToken string) (string, error) {
+	var st string
+	switch it.State {
+	case StateUnspecified:
+		st = ""
+	case Pending:
+		st = "pending"
+	case Running:
+		st = "running"
+	case Done:
+		st = "done"
+	default:
+		return "", fmt.Errorf("bigquery: invalid value for JobIterator.State: %d", it.State)
+	}
+	jobInfos, nextPageToken, err := it.c.service.listJobs(it.ctx, it.ProjectID, pageSize, pageToken, it.AllUsers, st)
+	if err != nil {
+		return "", err
+	}
+	for _, ji := range jobInfos {
+		ji.Job.c = it.c
+		ji.Status.setClient(it.c)
+		it.items = append(it.items, ji)
+	}
+	return nextPageToken, nil
+}