fix bugs

Signed-off-by: Friedrich Gonzalez <1517449+friedrichg@users.noreply.github.com>

Author: friedrichg

pkg/bench/query_runner.go

@@ -245,7 +245,7 @@ func (q *queryRunner) resolveAddrs() error {
 	defer cancel()
 
 	// If some of the dns resolution fails, log the error.
-	if err := q.dnsProvider.Resolve(ctx, []string{q.cfg.Endpoint}); err != nil {
+	if err := q.dnsProvider.Resolve(ctx, []string{q.cfg.Endpoint}, true); err != nil {
 		level.Error(q.logger).Log("msg", "failed to resolve addresses", "err", err)
 	}
 

pkg/bench/write_runner.go

@@ -223,7 +223,7 @@ func (w *WriteBenchmarkRunner) resolveAddrs() error {
 	defer cancel()
 
 	// If some of the dns resolution fails, log the error.
-	if err := w.dnsProvider.Resolve(ctx, []string{w.cfg.Endpoint}); err != nil {
+	if err := w.dnsProvider.Resolve(ctx, []string{w.cfg.Endpoint}, true); err != nil {
 		level.Error(w.logger).Log("msg", "failed to resolve addresses", "err", err)
 	}
 

pkg/commands/bucket_validation.go

@@ -115,7 +115,7 @@ func (b *BucketValidationCommand) validate(_ *kingpin.ParseContext) error {
 	b.logger = log.NewLogfmtLogger(log.NewSyncWriter(os.Stderr))
 	ctx := context.Background()
 
-	bucketClient, err := bucket.NewClient(ctx, b.cfg, "testClient", b.logger, prometheus.DefaultRegisterer)
+	bucketClient, err := bucket.NewClient(ctx, b.cfg, nil, "testClient", b.logger, prometheus.DefaultRegisterer)
 	if err != nil {
 		return errors.Wrap(err, "failed to create the bucket client")
 	}

pkg/commands/remote_read.go

@@ -252,12 +252,17 @@ func (c *RemoteReadCommand) prepare() (query func(context.Context) ([]*prompb.Ti
 
 	return func(ctx context.Context) ([]*prompb.TimeSeries, error) {
 		log.Infof("Querying time from=%s to=%s with selector=%s", from.Format(time.RFC3339), to.Format(time.RFC3339), c.selector)
-		resp, err := readClient.Read(ctx, pbQuery)
+		ss, err := readClient.Read(ctx, pbQuery, false)
 		if err != nil {
 			return nil, err
 		}
 
-		return resp.Timeseries, nil
+		result, _, err := remote.ToQueryResult(ss, 0)
+		if err != nil {
+			return nil, err
+		}
+
+		return result.Timeseries, nil
 
 	}, from, to, nil
 }