mirror of
https://github.com/grafana/grafana.git
synced 2024-11-28 19:54:10 -06:00
86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package testdatasource
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/datasource"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/log"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend/resource/httpadapter"
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
|
|
"github.com/grafana/grafana/pkg/tsdb/grafana-testdata-datasource/kinds"
|
|
"github.com/grafana/grafana/pkg/tsdb/grafana-testdata-datasource/sims"
|
|
)
|
|
|
|
// ensures that testdata implements all client functions
|
|
// var _ plugins.Client = &Service{}
|
|
|
|
func ProvideService() *Service {
|
|
s := &Service{
|
|
queryMux: datasource.NewQueryTypeMux(),
|
|
scenarios: map[kinds.TestDataQueryType]*Scenario{},
|
|
frame: data.NewFrame("testdata",
|
|
data.NewField("Time", nil, make([]time.Time, 1)),
|
|
data.NewField("Value", nil, make([]float64, 1)),
|
|
data.NewField("Min", nil, make([]float64, 1)),
|
|
data.NewField("Max", nil, make([]float64, 1)),
|
|
),
|
|
labelFrame: data.NewFrame("labeled",
|
|
data.NewField("labels", nil, make([]string, 1)),
|
|
data.NewField("Time", nil, make([]time.Time, 1)),
|
|
data.NewField("Value", nil, make([]float64, 1)),
|
|
),
|
|
logger: backend.NewLoggerWith("logger", "tsdb.testdata"),
|
|
}
|
|
|
|
var err error
|
|
s.sims, err = sims.NewSimulationEngine()
|
|
if err != nil {
|
|
s.logger.Error("Unable to initialize SimulationEngine", "err", err)
|
|
}
|
|
|
|
s.registerScenarios()
|
|
s.resourceHandler = httpadapter.New(s.registerRoutes())
|
|
|
|
return s
|
|
}
|
|
|
|
var (
|
|
_ backend.QueryDataHandler = (*Service)(nil)
|
|
_ backend.CallResourceHandler = (*Service)(nil)
|
|
_ backend.AdmissionHandler = (*Service)(nil)
|
|
_ backend.CollectMetricsHandler = (*Service)(nil)
|
|
)
|
|
|
|
type Service struct {
|
|
logger log.Logger
|
|
scenarios map[kinds.TestDataQueryType]*Scenario
|
|
frame *data.Frame
|
|
labelFrame *data.Frame
|
|
queryMux *datasource.QueryTypeMux
|
|
resourceHandler backend.CallResourceHandler
|
|
sims *sims.SimulationEngine
|
|
}
|
|
|
|
func (s *Service) QueryData(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
|
|
return s.queryMux.QueryData(ctx, req)
|
|
}
|
|
|
|
func (s *Service) CallResource(ctx context.Context, req *backend.CallResourceRequest, sender backend.CallResourceResponseSender) error {
|
|
return s.resourceHandler.CallResource(ctx, req, sender)
|
|
}
|
|
|
|
func (s *Service) CheckHealth(_ context.Context, _ *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
|
|
return &backend.CheckHealthResult{
|
|
Status: backend.HealthStatusOk,
|
|
Message: "Data source is working",
|
|
}, nil
|
|
}
|
|
|
|
// CollectMetricsHandler handles metric collection.
|
|
func (s *Service) CollectMetrics(ctx context.Context, req *backend.CollectMetricsRequest) (*backend.CollectMetricsResult, error) {
|
|
return nil, nil
|
|
}
|