From 5459b5d9370ed9dfdf90bcdd42deb5f0e2337237 Mon Sep 17 00:00:00 2001 From: Jonathan Amsterdam Date: Mon, 21 Sep 2020 18:13:24 -0400 Subject: 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 Run-TryBot: Jonathan Amsterdam TryBot-Result: kokoro Reviewed-by: Julie Qiu --- internal/queue/queue.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'internal/queue/queue.go') 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 } -- cgit v1.3-5-g9baa