Andres Martinez Gotor
99 lines
3.0 KiB
Go
99 lines
3.0 KiB
Go
package datasourcecheck
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
advisor "github.com/grafana/grafana/apps/advisor/pkg/apis/advisor/v0alpha1"
|
|
"github.com/grafana/grafana/apps/advisor/pkg/app/checks"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/registry/apis/datasource"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
"github.com/grafana/grafana/pkg/services/pluginsintegration/pluginstore"
|
|
"github.com/grafana/grafana/pkg/util"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
func New(
|
|
datasourceSvc datasources.DataSourceService,
|
|
pluginStore pluginstore.Store,
|
|
pluginContextProvider datasource.PluginContextWrapper,
|
|
pluginClient plugins.Client,
|
|
) checks.Check {
|
|
return &check{
|
|
DatasourceSvc: datasourceSvc,
|
|
PluginStore: pluginStore,
|
|
PluginContextProvider: pluginContextProvider,
|
|
PluginClient: pluginClient,
|
|
}
|
|
}
|
|
|
|
type check struct {
|
|
DatasourceSvc datasources.DataSourceService
|
|
PluginStore pluginstore.Store
|
|
PluginContextProvider datasource.PluginContextWrapper
|
|
PluginClient plugins.Client
|
|
}
|
|
|
|
func (c *check) Type() string {
|
|
return "datasource"
|
|
}
|
|
|
|
func (c *check) Run(ctx context.Context, obj *advisor.CheckSpec) (*advisor.CheckV0alpha1StatusReport, error) {
|
|
// Optionally read the check input encoded in the object
|
|
// fmt.Println(obj.Data)
|
|
|
|
dss, err := c.DatasourceSvc.GetAllDataSources(ctx, &datasources.GetAllDataSourcesQuery{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dsErrs := []advisor.CheckReportError{}
|
|
for _, ds := range dss {
|
|
// Data source UID validation
|
|
err := util.ValidateUID(ds.UID)
|
|
if err != nil {
|
|
dsErrs = append(dsErrs, advisor.CheckReportError{
|
|
Severity: advisor.CheckReportErrorSeverityLow,
|
|
Reason: fmt.Sprintf("Invalid UID '%s' for data source %s", ds.UID, ds.Name),
|
|
Action: "Check the <a href='https://grafana.com/docs/grafana/latest/upgrade-guide/upgrade-v11.2/#grafana-data-source-uid-format-enforcement' target=_blank>documentation</a> for more information.",
|
|
})
|
|
}
|
|
|
|
// Health check execution
|
|
pCtx, err := c.PluginContextProvider.PluginContextForDataSource(ctx, &backend.DataSourceInstanceSettings{
|
|
Type: ds.Type,
|
|
UID: ds.UID,
|
|
APIVersion: ds.APIVersion,
|
|
})
|
|
if err != nil {
|
|
klog.ErrorS(err, "Error creating plugin context", "datasource", ds.Name)
|
|
continue
|
|
}
|
|
req := &backend.CheckHealthRequest{
|
|
PluginContext: pCtx,
|
|
Headers: map[string]string{},
|
|
}
|
|
resp, err := c.PluginClient.CheckHealth(ctx, req)
|
|
if err != nil {
|
|
fmt.Println("Error checking health", err)
|
|
continue
|
|
}
|
|
if resp.Status != backend.HealthStatusOk {
|
|
dsErrs = append(dsErrs, advisor.CheckReportError{
|
|
Severity: advisor.CheckReportErrorSeverityHigh,
|
|
Reason: fmt.Sprintf("Health check failed for %s", ds.Name),
|
|
Action: fmt.Sprintf(
|
|
"Go to the <a href='/connections/datasources/edit/%s'>data source configuration</a>"+
|
|
" and address the issues reported.", ds.UID),
|
|
})
|
|
}
|
|
}
|
|
|
|
return &advisor.CheckV0alpha1StatusReport{
|
|
Count: int64(len(dss)),
|
|
Errors: dsErrs,
|
|
}, nil
|
|
}
|