mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 12:14:08 -06:00
336bc559a3
This pull request migrates testdata to coreplugin streaming capabilities, this is mostly a working concept of streaming plugins at the moment, the work will continue in the following pull requests.
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package plugins
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/grpcplugin"
|
|
"github.com/grafana/grafana-plugin-sdk-go/genproto/pluginv2"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/plugins/adapters"
|
|
"github.com/grafana/grafana/pkg/services/oauthtoken"
|
|
)
|
|
|
|
func newDataSourcePluginWrapperV2(log log.Logger, pluginId, pluginType string, dataClient grpcplugin.DataClient) *DatasourcePluginWrapperV2 {
|
|
return &DatasourcePluginWrapperV2{DataClient: dataClient, logger: log, pluginId: pluginId, pluginType: pluginType}
|
|
}
|
|
|
|
type DatasourcePluginWrapperV2 struct {
|
|
grpcplugin.DataClient
|
|
logger log.Logger
|
|
pluginId string
|
|
pluginType string
|
|
}
|
|
|
|
func (tw *DatasourcePluginWrapperV2) Query(ctx context.Context, ds *models.DataSource, query DataQuery) (DataResponse, error) {
|
|
instanceSettings, err := adapters.ModelToInstanceSettings(ds)
|
|
if err != nil {
|
|
return DataResponse{}, err
|
|
}
|
|
|
|
if query.Headers == nil {
|
|
query.Headers = make(map[string]string)
|
|
}
|
|
|
|
if oauthtoken.IsOAuthPassThruEnabled(ds) {
|
|
if token := oauthtoken.GetCurrentOAuthToken(ctx, query.User); token != nil {
|
|
delete(query.Headers, "Authorization")
|
|
query.Headers["Authorization"] = fmt.Sprintf("%s %s", token.Type(), token.AccessToken)
|
|
}
|
|
}
|
|
|
|
pbQuery := &pluginv2.QueryDataRequest{
|
|
PluginContext: &pluginv2.PluginContext{
|
|
OrgId: ds.OrgId,
|
|
PluginId: tw.pluginId,
|
|
User: backend.ToProto().User(adapters.BackendUserFromSignedInUser(query.User)),
|
|
DataSourceInstanceSettings: backend.ToProto().DataSourceInstanceSettings(instanceSettings),
|
|
},
|
|
Queries: []*pluginv2.DataQuery{},
|
|
Headers: query.Headers,
|
|
}
|
|
|
|
for _, q := range query.Queries {
|
|
modelJSON, err := q.Model.MarshalJSON()
|
|
if err != nil {
|
|
return DataResponse{}, err
|
|
}
|
|
pbQuery.Queries = append(pbQuery.Queries, &pluginv2.DataQuery{
|
|
Json: modelJSON,
|
|
IntervalMS: q.IntervalMS,
|
|
RefId: q.RefID,
|
|
MaxDataPoints: q.MaxDataPoints,
|
|
TimeRange: &pluginv2.TimeRange{
|
|
ToEpochMS: query.TimeRange.GetToAsMsEpoch(),
|
|
FromEpochMS: query.TimeRange.GetFromAsMsEpoch(),
|
|
},
|
|
QueryType: q.QueryType,
|
|
})
|
|
}
|
|
|
|
pbRes, err := tw.DataClient.QueryData(ctx, pbQuery)
|
|
if err != nil {
|
|
return DataResponse{}, err
|
|
}
|
|
|
|
tR := DataResponse{
|
|
Results: make(map[string]DataQueryResult, len(pbRes.Responses)),
|
|
}
|
|
|
|
for refID, pRes := range pbRes.Responses {
|
|
qr := DataQueryResult{
|
|
RefID: refID,
|
|
Dataframes: NewEncodedDataFrames(pRes.Frames),
|
|
}
|
|
if len(pRes.JsonMeta) != 0 {
|
|
qr.Meta = simplejson.NewFromAny(pRes.JsonMeta)
|
|
}
|
|
if pRes.Error != "" {
|
|
qr.Error = fmt.Errorf(pRes.Error)
|
|
qr.ErrorString = pRes.Error
|
|
}
|
|
tR.Results[refID] = qr
|
|
}
|
|
|
|
return tR, nil
|
|
}
|