QueryService: Use types from sdk (#84029)
This commit is contained in:
Ryan McKinley
@@ -0,0 +1,179 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
||||
data "github.com/grafana/grafana-plugin-sdk-go/experimental/apis/data/v0alpha1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
|
||||
query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
|
||||
"github.com/grafana/grafana/pkg/plugins"
|
||||
"github.com/grafana/grafana/pkg/services/datasources"
|
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration/plugincontext"
|
||||
"github.com/grafana/grafana/pkg/services/pluginsintegration/pluginstore"
|
||||
"github.com/grafana/grafana/pkg/setting"
|
||||
"github.com/grafana/grafana/pkg/tsdb/legacydata"
|
||||
)
|
||||
|
||||
type pluginClient struct {
|
||||
pluginClient plugins.Client
|
||||
pCtxProvider *plugincontext.Provider
|
||||
}
|
||||
|
||||
type pluginRegistry struct {
|
||||
pluginsMu sync.Mutex
|
||||
plugins *query.DataSourceApiServerList
|
||||
apis map[string]schema.GroupVersion
|
||||
groupToPlugin map[string]string
|
||||
pluginStore pluginstore.Store
|
||||
|
||||
// called on demand
|
||||
dataSourcesService datasources.DataSourceService
|
||||
}
|
||||
|
||||
var _ data.QueryDataClient = (*pluginClient)(nil)
|
||||
var _ query.DataSourceApiServerRegistry = (*pluginRegistry)(nil)
|
||||
|
||||
// NewDummyTestRunner creates a runner that only works with testdata
|
||||
func NewQueryClientForPluginClient(p plugins.Client, ctx *plugincontext.Provider) data.QueryDataClient {
|
||||
return &pluginClient{
|
||||
pluginClient: p,
|
||||
pCtxProvider: ctx,
|
||||
}
|
||||
}
|
||||
|
||||
func NewDataSourceRegistryFromStore(pluginStore pluginstore.Store,
|
||||
dataSourcesService datasources.DataSourceService,
|
||||
) query.DataSourceApiServerRegistry {
|
||||
return &pluginRegistry{
|
||||
pluginStore: pluginStore,
|
||||
dataSourcesService: dataSourcesService,
|
||||
}
|
||||
}
|
||||
|
||||
// ExecuteQueryData implements QueryHelper.
|
||||
func (d *pluginClient) QueryData(ctx context.Context, req data.QueryDataRequest) (int, *backend.QueryDataResponse, error) {
|
||||
queries, dsRef, err := legacydata.ToDataSourceQueries(req)
|
||||
if err != nil {
|
||||
return http.StatusBadRequest, nil, err
|
||||
}
|
||||
if dsRef == nil {
|
||||
return http.StatusBadRequest, nil, fmt.Errorf("expected single datasource request")
|
||||
}
|
||||
|
||||
// NOTE: this depends on uid unique across datasources
|
||||
settings, err := d.pCtxProvider.GetDataSourceInstanceSettings(ctx, dsRef.UID)
|
||||
if err != nil {
|
||||
return http.StatusBadRequest, nil, err
|
||||
}
|
||||
|
||||
qdr := &backend.QueryDataRequest{
|
||||
Queries: queries,
|
||||
}
|
||||
qdr.PluginContext, err = d.pCtxProvider.PluginContextForDataSource(ctx, settings)
|
||||
if err != nil {
|
||||
return http.StatusBadRequest, nil, err
|
||||
}
|
||||
|
||||
code := http.StatusOK
|
||||
rsp, err := d.pluginClient.QueryData(ctx, qdr)
|
||||
if err == nil {
|
||||
for _, v := range rsp.Responses {
|
||||
if v.Error != nil {
|
||||
code = http.StatusMultiStatus
|
||||
break
|
||||
}
|
||||
}
|
||||
} else {
|
||||
code = http.StatusInternalServerError
|
||||
}
|
||||
return code, rsp, err
|
||||
}
|
||||
|
||||
// GetDatasourceAPI implements DataSourceRegistry.
|
||||
func (d *pluginRegistry) GetDatasourceGroupVersion(pluginId string) (schema.GroupVersion, error) {
|
||||
d.pluginsMu.Lock()
|
||||
defer d.pluginsMu.Unlock()
|
||||
|
||||
if d.plugins == nil {
|
||||
err := d.updatePlugins()
|
||||
if err != nil {
|
||||
return schema.GroupVersion{}, err
|
||||
}
|
||||
}
|
||||
|
||||
var err error
|
||||
gv, ok := d.apis[pluginId]
|
||||
if !ok {
|
||||
err = fmt.Errorf("no API found for id: " + pluginId)
|
||||
}
|
||||
return gv, err
|
||||
}
|
||||
|
||||
// GetDatasourcePlugins no namespace? everything that is available
|
||||
func (d *pluginRegistry) GetDatasourceApiServers(ctx context.Context) (*query.DataSourceApiServerList, error) {
|
||||
d.pluginsMu.Lock()
|
||||
defer d.pluginsMu.Unlock()
|
||||
|
||||
if d.plugins == nil {
|
||||
err := d.updatePlugins()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return d.plugins, nil
|
||||
}
|
||||
|
||||
// This should be called when plugins change
|
||||
func (d *pluginRegistry) updatePlugins() error {
|
||||
groupToPlugin := map[string]string{}
|
||||
apis := map[string]schema.GroupVersion{}
|
||||
result := &query.DataSourceApiServerList{
|
||||
ListMeta: metav1.ListMeta{
|
||||
ResourceVersion: fmt.Sprintf("%d", time.Now().UnixMilli()),
|
||||
},
|
||||
}
|
||||
|
||||
// TODO? only backend plugins
|
||||
for _, dsp := range d.pluginStore.Plugins(context.Background(), plugins.TypeDataSource) {
|
||||
ts := setting.BuildStamp * 1000
|
||||
if dsp.Info.Build.Time > 0 {
|
||||
ts = dsp.Info.Build.Time
|
||||
}
|
||||
|
||||
group, err := plugins.GetDatasourceGroupNameFromPluginID(dsp.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
gv := schema.GroupVersion{Group: group, Version: "v0alpha1"} // default version
|
||||
apis[dsp.ID] = gv
|
||||
for _, alias := range dsp.AliasIDs {
|
||||
apis[alias] = gv
|
||||
}
|
||||
groupToPlugin[group] = dsp.ID
|
||||
|
||||
ds := query.DataSourceApiServer{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: dsp.ID,
|
||||
CreationTimestamp: metav1.NewTime(time.UnixMilli(ts)),
|
||||
},
|
||||
Title: dsp.Name,
|
||||
AliasIDs: dsp.AliasIDs,
|
||||
GroupVersion: gv.String(),
|
||||
Description: dsp.Info.Description,
|
||||
}
|
||||
result.Items = append(result.Items, ds)
|
||||
}
|
||||
|
||||
d.plugins = result
|
||||
d.apis = apis
|
||||
d.groupToPlugin = groupToPlugin
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,75 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
||||
data "github.com/grafana/grafana-plugin-sdk-go/experimental/apis/data/v0alpha1"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
|
||||
query "github.com/grafana/grafana/pkg/apis/query/v0alpha1"
|
||||
testdata "github.com/grafana/grafana/pkg/tsdb/grafana-testdata-datasource"
|
||||
"github.com/grafana/grafana/pkg/tsdb/legacydata"
|
||||
)
|
||||
|
||||
type testdataDummy struct{}
|
||||
|
||||
var _ data.QueryDataClient = (*testdataDummy)(nil)
|
||||
var _ query.DataSourceApiServerRegistry = (*testdataDummy)(nil)
|
||||
|
||||
// NewTestDataClient creates a runner that only works with testdata
|
||||
func NewTestDataClient() data.QueryDataClient {
|
||||
return &testdataDummy{}
|
||||
}
|
||||
|
||||
// NewTestDataRegistry returns a registry that only knows about testdata
|
||||
func NewTestDataRegistry() query.DataSourceApiServerRegistry {
|
||||
return &testdataDummy{}
|
||||
}
|
||||
|
||||
// ExecuteQueryData implements QueryHelper.
|
||||
func (d *testdataDummy) QueryData(ctx context.Context, req data.QueryDataRequest) (int, *backend.QueryDataResponse, error) {
|
||||
queries, _, err := legacydata.ToDataSourceQueries(req)
|
||||
if err != nil {
|
||||
return http.StatusBadRequest, nil, err
|
||||
}
|
||||
|
||||
qdr := &backend.QueryDataRequest{Queries: queries}
|
||||
rsp, err := testdata.ProvideService().QueryData(ctx, qdr)
|
||||
return query.GetResponseCode(rsp), rsp, err
|
||||
}
|
||||
|
||||
// GetDatasourceAPI implements DataSourceRegistry.
|
||||
func (*testdataDummy) GetDatasourceGroupVersion(pluginId string) (schema.GroupVersion, error) {
|
||||
if pluginId == "testdata" || pluginId == "grafana-testdata-datasource" {
|
||||
return schema.GroupVersion{
|
||||
Group: "testdata.datasource.grafana.app",
|
||||
Version: "v0alpha1",
|
||||
}, nil
|
||||
}
|
||||
return schema.GroupVersion{}, fmt.Errorf("unsupported plugin (only testdata for now)")
|
||||
}
|
||||
|
||||
// GetDatasourcePlugins implements QueryHelper.
|
||||
func (d *testdataDummy) GetDatasourceApiServers(ctx context.Context) (*query.DataSourceApiServerList, error) {
|
||||
return &query.DataSourceApiServerList{
|
||||
ListMeta: metav1.ListMeta{
|
||||
ResourceVersion: fmt.Sprintf("%d", time.Now().UnixMilli()),
|
||||
},
|
||||
Items: []query.DataSourceApiServer{
|
||||
{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Name: "grafana-testdata-datasource",
|
||||
CreationTimestamp: metav1.Now(),
|
||||
},
|
||||
Title: "Test Data",
|
||||
GroupVersion: "testdata.datasource.grafana.app/v0alpha1",
|
||||
AliasIDs: []string{"testdata"},
|
||||
},
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user