mirror of
https://github.com/grafana/grafana.git
synced 2025-08-01 08:52:38 +08:00
4eadb0fec8
* ds-querier: handle execute errors better * fix: change how GetResponseCode works to return 418 if rsp is nil 418 is a bit of an easter egg which in this case works since we don't have an rsp but we do know something went wrong, so a 200 won't work. Also changed this to return the code in the frame, not sure why we weren't. * tests: fix GetResponseCode tests * log no rsp case * bring back og error log
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package v0alpha1
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
data "github.com/grafana/grafana-plugin-sdk-go/experimental/apis/data/v0alpha1"
|
|
)
|
|
|
|
// Generic query request with shared time across all values
|
|
// Copied from: https://github.com/grafana/grafana/blob/main/pkg/api/dtos/models.go#L62
|
|
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
|
|
type QueryDataRequest struct {
|
|
metav1.TypeMeta `json:",inline"`
|
|
|
|
// The time range used when not included on each query
|
|
data.QueryDataRequest `json:",inline"`
|
|
}
|
|
|
|
// Wraps backend.QueryDataResponse, however it includes TypeMeta and implements runtime.Object
|
|
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
|
|
type QueryDataResponse struct {
|
|
metav1.TypeMeta `json:",inline"`
|
|
|
|
// Backend wrapper (external dependency)
|
|
backend.QueryDataResponse `json:",inline"`
|
|
}
|
|
|
|
// GetResponseCode return the right status code for the response by checking the responses.
|
|
func GetResponseCode(rsp *backend.QueryDataResponse) int {
|
|
if rsp == nil {
|
|
return http.StatusTeapot // rsp is nil, so we return a teapot
|
|
}
|
|
for _, res := range rsp.Responses {
|
|
if res.Error != nil && res.Status != 0 {
|
|
return int(res.Status)
|
|
}
|
|
|
|
if res.Error != nil {
|
|
return http.StatusTeapot // Status is nil but we have an error, so we return a teapot
|
|
}
|
|
}
|
|
return http.StatusOK
|
|
}
|
|
|
|
// Defines a query behavior in a datasource. This is a similar model to a CRD where the
|
|
// payload describes a valid query
|
|
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
|
|
type QueryTypeDefinition struct {
|
|
metav1.TypeMeta `json:",inline"`
|
|
metav1.ObjectMeta `json:"metadata,omitempty"`
|
|
|
|
Spec data.QueryTypeDefinitionSpec `json:"spec,omitempty"`
|
|
}
|
|
|
|
// +k8s:deepcopy-gen:interfaces=k8s.io/apimachinery/pkg/runtime.Object
|
|
type QueryTypeDefinitionList struct {
|
|
metav1.TypeMeta `json:",inline"`
|
|
metav1.ListMeta `json:"metadata,omitempty"`
|
|
|
|
Items []QueryTypeDefinition `json:"items"`
|
|
}
|