diff options
| author | Jonathan Amsterdam <jba@google.com> | 2020-09-21 18:13:24 -0400 |
|---|---|---|
| committer | Jonathan Amsterdam <jba@google.com> | 2020-09-22 15:16:06 +0000 |
| commit | 5459b5d9370ed9dfdf90bcdd42deb5f0e2337237 (patch) | |
| tree | 29427f6f14fae862257a490dde49c013604ad0cb /internal/queue/queue.go | |
| parent | d1c56be7ea7b8f8007c4dcfb14799fd9cde7ec01 (diff) | |
| download | go-x-pkgsite-5459b5d9370ed9dfdf90bcdd42deb5f0e2337237.tar.xz | |
internal/queue: use an experiment getter instead of the DB
To get the list of experiments, pass in an ExperimentGetter instead of
a DB client.
Change-Id: I89ccd402a178d195c85efcdaf363e59c45358b2f
Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/256381
Trust: Jonathan Amsterdam <jba@google.com>
Run-TryBot: Jonathan Amsterdam <jba@google.com>
TryBot-Result: kokoro <noreply+kokoro@google.com>
Reviewed-by: Julie Qiu <julie@golang.org>
Diffstat (limited to 'internal/queue/queue.go')
| -rw-r--r-- | internal/queue/queue.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/internal/queue/queue.go b/internal/queue/queue.go index ac89fcc2..110c72a3 100644 --- a/internal/queue/queue.go +++ b/internal/queue/queue.go @@ -18,7 +18,7 @@ import ( "golang.org/x/pkgsite/internal/derrors" "golang.org/x/pkgsite/internal/experiment" "golang.org/x/pkgsite/internal/log" - "golang.org/x/pkgsite/internal/postgres" + "golang.org/x/pkgsite/internal/middleware" taskspb "google.golang.org/genproto/googleapis/cloud/tasks/v2" "google.golang.org/grpc/codes" "google.golang.org/grpc/status" @@ -31,9 +31,9 @@ type Queue interface { // New creates a new Queue with name queueName based on the configuration // in cfg. When running locally, Queue uses numWorkers concurrent workers. -func New(ctx context.Context, cfg *config.Config, queueName string, numWorkers int, db *postgres.DB, processFunc inMemoryProcessFunc) (Queue, error) { +func New(ctx context.Context, cfg *config.Config, queueName string, numWorkers int, expGetter middleware.ExperimentGetter, processFunc inMemoryProcessFunc) (Queue, error) { if !cfg.OnGCP() { - experiments, err := db.GetExperiments(ctx) + experiments, err := expGetter(ctx) if err != nil { return nil, err } |