mirror of
https://github.com/grafana/grafana.git
synced 2025-02-10 23:55:47 -06:00
c989f0fbbe
* convert SQLs to use sdk contracts * make draft * postgres * intermedia * get datasourceinfo filled at the beginning of the service * move the interval into package because of cyclict import and fix all postgres tests * fix mysql test * fix mssql * fix the test for pr https://github.com/grafana/grafana/issues/35839 * fix some issue about intervalv2 package * update sql test * wire migration for SQLs * add sqls to the background process * make it register instead of register and start * revert formatting * fix tests * fix linter * remove integration test * Postgres test fix Co-authored-by: Marcus Efraimsson <marcus.efraimsson@gmail.com>
77 lines
2.7 KiB
Go
77 lines
2.7 KiB
Go
package tsdb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/httpclient"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/plugins/backendplugin"
|
|
"github.com/grafana/grafana/pkg/services/oauthtoken"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudmonitoring"
|
|
_ "github.com/grafana/grafana/pkg/tsdb/postgres"
|
|
)
|
|
|
|
// NewService returns a new Service.
|
|
func NewService(
|
|
cfg *setting.Cfg, pluginManager plugins.Manager, backendPluginManager backendplugin.Manager,
|
|
oauthTokenService *oauthtoken.Service, httpClientProvider httpclient.Provider, cloudMonitoringService *cloudmonitoring.Service,
|
|
) *Service {
|
|
s := newService(cfg, pluginManager, backendPluginManager, oauthTokenService)
|
|
|
|
// register backend data sources using legacy plugin
|
|
// contracts/non-SDK contracts
|
|
s.registry["stackdriver"] = cloudMonitoringService.NewExecutor
|
|
|
|
return s
|
|
}
|
|
|
|
func newService(cfg *setting.Cfg, manager plugins.Manager, backendPluginManager backendplugin.Manager,
|
|
oauthTokenService oauthtoken.OAuthTokenService) *Service {
|
|
return &Service{
|
|
Cfg: cfg,
|
|
PluginManager: manager,
|
|
BackendPluginManager: backendPluginManager,
|
|
// nolint:staticcheck // plugins.DataPlugin deprecated
|
|
registry: map[string]func(*models.DataSource) (plugins.DataPlugin, error){},
|
|
OAuthTokenService: oauthTokenService,
|
|
}
|
|
}
|
|
|
|
// Service handles data requests to data sources.
|
|
type Service struct {
|
|
Cfg *setting.Cfg
|
|
PluginManager plugins.Manager
|
|
BackendPluginManager backendplugin.Manager
|
|
OAuthTokenService oauthtoken.OAuthTokenService
|
|
|
|
//nolint: staticcheck // plugins.DataPlugin deprecated
|
|
registry map[string]func(*models.DataSource) (plugins.DataPlugin, error)
|
|
}
|
|
|
|
//nolint: staticcheck // plugins.DataPlugin deprecated
|
|
func (s *Service) HandleRequest(ctx context.Context, ds *models.DataSource, query plugins.DataQuery) (plugins.DataResponse, error) {
|
|
if factory, exists := s.registry[ds.Type]; exists {
|
|
var err error
|
|
plugin, err := factory(ds)
|
|
if err != nil {
|
|
//nolint: staticcheck // plugins.DataPlugin deprecated
|
|
return plugins.DataResponse{}, fmt.Errorf("could not instantiate endpoint for data plugin %q: %w",
|
|
ds.Type, err)
|
|
}
|
|
|
|
return plugin.DataQuery(ctx, ds, query)
|
|
}
|
|
|
|
return dataPluginQueryAdapter(ds.Type, s.BackendPluginManager, s.OAuthTokenService).DataQuery(ctx, ds, query)
|
|
}
|
|
|
|
// RegisterQueryHandler registers a query handler factory.
|
|
// This is only exposed for tests!
|
|
//nolint: staticcheck // plugins.DataPlugin deprecated
|
|
func (s *Service) RegisterQueryHandler(name string, factory func(*models.DataSource) (plugins.DataPlugin, error)) {
|
|
s.registry[name] = factory
|
|
}
|