OpenTSDB: Move health check to the backend (#114082)

* add feature toggle

* move health check to backend

* add tests
This commit is contained in:
Gareth
2025-12-04 17:11:33 +09:00
committed by GitHub
parent c8853f50cf
commit dc8bb66a45
10 changed files with 165 additions and 5 deletions
+8
View File
@@ -1961,6 +1961,14 @@ var (
FrontendOnly: true,
Owner: grafanaDataProSquad,
},
{
Name: "opentsdbBackendMigration",
Description: "Run queries through the data source backend",
Stage: FeatureStageGeneralAvailability,
Owner: grafanaOSSBigTent,
Expression: "false",
RequiresRestart: true,
},
{
Name: "ttlPluginInstanceManager",
Description: "Enable TTL plugin instance manager",
+1
View File
@@ -266,6 +266,7 @@ panelTimeSettings,experimental,@grafana/dashboards-squad,false,false,false
kubernetesAnnotations,experimental,@grafana/grafana-backend-services-squad,false,false,false
awsDatasourcesHttpProxy,experimental,@grafana/aws-datasources,false,false,false
transformationsEmptyPlaceholder,preview,@grafana/datapro,false,false,true
opentsdbBackendMigration,GA,@grafana/oss-big-tent,false,true,false
ttlPluginInstanceManager,experimental,@grafana/plugins-platform-backend,false,false,true
lokiQueryLimitsContext,experimental,@grafana/observability-logs,false,false,true
rudderstackUpgrade,experimental,@grafana/grafana-frontend-platform,false,false,true
1 Name Stage Owner requiresDevMode RequiresRestart FrontendOnly
266 kubernetesAnnotations experimental @grafana/grafana-backend-services-squad false false false
267 awsDatasourcesHttpProxy experimental @grafana/aws-datasources false false false
268 transformationsEmptyPlaceholder preview @grafana/datapro false false true
269 opentsdbBackendMigration GA @grafana/oss-big-tent false true false
270 ttlPluginInstanceManager experimental @grafana/plugins-platform-backend false false true
271 lokiQueryLimitsContext experimental @grafana/observability-logs false false true
272 rudderstackUpgrade experimental @grafana/grafana-frontend-platform false false true
+4
View File
@@ -762,6 +762,10 @@ const (
// Enables http proxy settings for aws datasources
FlagAwsDatasourcesHttpProxy = "awsDatasourcesHttpProxy"
// FlagOpentsdbBackendMigration
// Run queries through the data source backend
FlagOpentsdbBackendMigration = "opentsdbBackendMigration"
// FlagKubernetesAlertingHistorian
// Adds support for Kubernetes alerting historian APIs
FlagKubernetesAlertingHistorian = "kubernetesAlertingHistorian"
+14
View File
@@ -2499,6 +2499,20 @@
"expression": "true"
}
},
{
"metadata": {
"name": "opentsdbBackendMigration",
"resourceVersion": "1763456634837",
"creationTimestamp": "2025-11-18T09:03:54Z"
},
"spec": {
"description": "Run queries through the data source backend",
"stage": "GA",
"codeowner": "@grafana/oss-big-tent",
"requiresRestart": true,
"expression": "false"
}
},
{
"metadata": {
"name": "otelLogsFormatting",
+60
View File
@@ -94,6 +94,66 @@ func newInstanceSettings(httpClientProvider *httpclient.Provider) datasource.Ins
}
}
func (s *Service) CheckHealth(ctx context.Context, req *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
logger := logger.FromContext(ctx)
dsInfo, err := s.getDSInfo(ctx, req.PluginContext)
if err != nil {
return &backend.CheckHealthResult{
Status: backend.HealthStatusError,
Message: err.Error(),
}, nil
}
u, err := url.Parse(dsInfo.URL)
if err != nil {
return &backend.CheckHealthResult{
Status: backend.HealthStatusError,
Message: err.Error(),
}, nil
}
u.Path = path.Join(u.Path, "api/suggest")
query := u.Query()
query.Set("q", "cpu")
query.Set("type", "metrics")
u.RawQuery = query.Encode()
httpReq, err := http.NewRequestWithContext(ctx, http.MethodGet, u.String(), nil)
if err != nil {
return &backend.CheckHealthResult{
Status: backend.HealthStatusError,
Message: err.Error(),
}, nil
}
res, err := dsInfo.HTTPClient.Do(httpReq)
if err != nil {
return &backend.CheckHealthResult{
Status: backend.HealthStatusError,
Message: err.Error(),
}, nil
}
defer func() {
if err := res.Body.Close(); err != nil {
logger.Error("Failed to close response body", "error", err)
}
}()
if res.StatusCode != 200 {
return &backend.CheckHealthResult{
Status: backend.HealthStatusError,
Message: fmt.Sprintf("OpenTSDB suggest endpoint returned status %d", res.StatusCode),
}, nil
}
return &backend.CheckHealthResult{
Status: backend.HealthStatusOk,
Message: "Data source is working",
}, nil
}
func (s *Service) QueryData(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
logger := logger.FromContext(ctx)
+52
View File
@@ -18,6 +18,58 @@ import (
"github.com/stretchr/testify/require"
)
func TestCheckHealth(t *testing.T) {
tests := []struct {
name string
httpStatusCode int
expectedStatus backend.HealthStatus
expectedMessage string
}{
{
name: "successful health check",
httpStatusCode: 200,
expectedStatus: backend.HealthStatusOk,
expectedMessage: "Data source is working",
},
{
name: "http error",
httpStatusCode: 500,
expectedStatus: backend.HealthStatusError,
expectedMessage: "OpenTSDB suggest endpoint returned status 500",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
assert.Equal(t, "/api/suggest", r.URL.Path)
assert.Equal(t, "cpu", r.URL.Query().Get("q"))
assert.Equal(t, "metrics", r.URL.Query().Get("type"))
w.WriteHeader(tt.httpStatusCode)
}))
defer server.Close()
pluginCtx := backend.PluginContext{
DataSourceInstanceSettings: &backend.DataSourceInstanceSettings{
URL: server.URL,
JSONData: []byte(`{}`),
},
}
im := datasource.NewInstanceManager(newInstanceSettings(httpclient.NewProvider()))
service := &Service{im: im}
ctx := backend.WithPluginContext(context.Background(), pluginCtx)
result, err := service.CheckHealth(ctx, &backend.CheckHealthRequest{
PluginContext: pluginCtx,
})
assert.NoError(t, err)
assert.Equal(t, tt.expectedStatus, result.Status)
assert.Contains(t, result.Message, tt.expectedMessage)
})
}
}
func TestOpenTsdbExecutor(t *testing.T) {
service := &Service{}
+6 -1
View File
@@ -10,7 +10,8 @@ import (
)
var (
_ backend.QueryDataHandler = (*Datasource)(nil)
_ backend.QueryDataHandler = (*Datasource)(nil)
_ backend.CheckHealthHandler = (*Datasource)(nil)
)
type Datasource struct {
@@ -26,3 +27,7 @@ func NewDatasource(context.Context, backend.DataSourceInstanceSettings) (instanc
func (d *Datasource) QueryData(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
return d.Service.QueryData(ctx, req)
}
func (d *Datasource) CheckHealth(ctx context.Context, req *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
return d.Service.CheckHealth(ctx, req)
}