Implements a basic SelectSeries API for querying metrics on profiles

grafana/fire-datasource/pkg/plugin/query.go

@@ -58,6 +58,23 @@ func (d *FireDatasource) query(ctx context.Context, pCtx backend.PluginContext,
 	// add the frames to the response.
 	response.Frames = append(response.Frames, frame)
 
+	log.DefaultLogger.Debug("Querying SelectSeries()", "queryModel", qm)
+
+	seriesResp, err := d.client.SelectSeries(ctx, connect.NewRequest(&querierv1.SelectSeriesRequest{
+		ProfileTypeID: qm.ProfileTypeID,
+		LabelSelector: qm.LabelSelector,
+		Start:         query.TimeRange.From.UnixMilli(),
+		End:           query.TimeRange.To.UnixMilli(),
+		Step:          query.Interval.Seconds(),
+		// todo add one or more group bys
+		GroupBy: []string{},
+	}))
+	if err != nil {
+		response.Error = err
+		return response
+	}
+	// todo remove me and add the series to the frame.
+	log.DefaultLogger.Debug("Series", seriesResp.Msg.Series)
 	return response
 }
 

grafana/fire-datasource/pkg/plugin/query_test.go

@@ -6,10 +6,11 @@ import (
 	"time"
 
 	"github.com/bufbuild/connect-go"
-	querierv1 "github.com/grafana/fire/pkg/gen/querier/v1"
 	"github.com/grafana/grafana-plugin-sdk-go/backend"
 	"github.com/grafana/grafana-plugin-sdk-go/data"
 	"github.com/stretchr/testify/require"
+
+	querierv1 "github.com/grafana/fire/pkg/gen/querier/v1"
 )
 
 // This is where the tests for the datasource backend live.
@@ -93,3 +94,7 @@ func (f FakeClient) SelectMergeStacktraces(ctx context.Context, c *connect.Reque
 		},
 	}, nil
 }
+
+func (f FakeClient) SelectSeries(ctx context.Context, req *connect.Request[querierv1.SelectSeriesRequest]) (*connect.Response[querierv1.SelectSeriesResponse], error) {
+	panic("implement me")
+}

pkg/distributor/distributor_test.go

@@ -26,13 +26,13 @@ import (
 	pushv1 "github.com/grafana/fire/pkg/gen/push/v1"
 	"github.com/grafana/fire/pkg/gen/push/v1/pushv1connect"
 	"github.com/grafana/fire/pkg/ingester/clientpool"
-	"github.com/grafana/fire/pkg/testutil"
+	"github.com/grafana/fire/pkg/testhelper"
 )
 
 func Test_ConnectPush(t *testing.T) {
 	mux := http.NewServeMux()
 	ing := newFakeIngester(t, false)
-	d, err := New(Config{}, testutil.NewMockRing([]ring.InstanceDesc{
+	d, err := New(Config{}, testhelper.NewMockRing([]ring.InstanceDesc{
 		{Addr: "foo"},
 	}, 3), func(addr string) (client.PoolClient, error) {
 		return ing, nil
@@ -84,7 +84,7 @@ func Test_Replication(t *testing.T) {
 			},
 		},
 	})
-	d, err := New(Config{}, testutil.NewMockRing([]ring.InstanceDesc{
+	d, err := New(Config{}, testhelper.NewMockRing([]ring.InstanceDesc{
 		{Addr: "1"},
 		{Addr: "2"},
 		{Addr: "3"},
@@ -107,7 +107,7 @@ func Test_Subservices(t *testing.T) {
 	ing := newFakeIngester(t, false)
 	d, err := New(Config{
 		PoolConfig: clientpool.PoolConfig{ClientCleanupPeriod: 1 * time.Second},
-	}, testutil.NewMockRing([]ring.InstanceDesc{
+	}, testhelper.NewMockRing([]ring.InstanceDesc{
 		{Addr: "foo"},
 	}, 1), func(addr string) (client.PoolClient, error) {
 		return ing, nil
@@ -138,7 +138,7 @@ type fakeIngester struct {
 	t        testing.TB
 	requests []*pushv1.PushRequest
 	fail     bool
-	testutil.FakePoolClient
+	testhelper.FakePoolClient
 
 	mtx sync.Mutex
 }