mirror of https://github.com/grafana/grafana.git
124 lines
2.6 KiB
Go
124 lines
2.6 KiB
Go
package backendplugin
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/genproto/pluginv2"
|
|
)
|
|
|
|
// HealthStatus is the status of the plugin.
|
|
type HealthStatus int
|
|
|
|
const (
|
|
// HealthStatusUnknown means the status of the plugin is unknown.
|
|
HealthStatusUnknown HealthStatus = iota
|
|
// HealthStatusOk means the status of the plugin is good.
|
|
HealthStatusOk
|
|
// HealthStatusError means the plugin is in an error state.
|
|
HealthStatusError
|
|
)
|
|
|
|
var healthStatusNames = map[int]string{
|
|
0: "UNKNOWN",
|
|
1: "OK",
|
|
2: "ERROR",
|
|
}
|
|
|
|
func (hs HealthStatus) String() string {
|
|
s, exists := healthStatusNames[int(hs)]
|
|
if exists {
|
|
return s
|
|
}
|
|
return strconv.Itoa(int(hs))
|
|
}
|
|
|
|
// CheckHealthResult check health result.
|
|
type CheckHealthResult struct {
|
|
Status HealthStatus
|
|
Message string
|
|
JSONDetails []byte
|
|
}
|
|
|
|
func checkHealthResultFromProto(protoResp *pluginv2.CheckHealthResponse) *CheckHealthResult {
|
|
status := HealthStatusUnknown
|
|
switch protoResp.Status {
|
|
case pluginv2.CheckHealthResponse_ERROR:
|
|
status = HealthStatusError
|
|
case pluginv2.CheckHealthResponse_OK:
|
|
status = HealthStatusOk
|
|
}
|
|
|
|
return &CheckHealthResult{
|
|
Status: status,
|
|
Message: protoResp.Message,
|
|
JSONDetails: protoResp.JsonDetails,
|
|
}
|
|
}
|
|
|
|
func collectMetricsResultFromProto(protoResp *pluginv2.CollectMetricsResponse) *CollectMetricsResult {
|
|
var prometheusMetrics []byte
|
|
|
|
if protoResp.Metrics != nil {
|
|
prometheusMetrics = protoResp.Metrics.Prometheus
|
|
}
|
|
|
|
return &CollectMetricsResult{
|
|
PrometheusMetrics: prometheusMetrics,
|
|
}
|
|
}
|
|
|
|
// CollectMetricsResult collect metrics result.
|
|
type CollectMetricsResult struct {
|
|
PrometheusMetrics []byte
|
|
}
|
|
|
|
// CallResourceResult call resource result.
|
|
type CallResourceResult struct {
|
|
Status int
|
|
Headers map[string][]string
|
|
Body []byte
|
|
}
|
|
|
|
type callResourceResultStream interface {
|
|
Recv() (*CallResourceResult, error)
|
|
Close() error
|
|
}
|
|
|
|
type callResourceResultStreamImpl struct {
|
|
stream pluginv2.Resource_CallResourceClient
|
|
}
|
|
|
|
func (s *callResourceResultStreamImpl) Recv() (*CallResourceResult, error) {
|
|
protoResp, err := s.stream.Recv()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
respHeaders := map[string][]string{}
|
|
for key, values := range protoResp.Headers {
|
|
respHeaders[key] = values.Values
|
|
}
|
|
|
|
return &CallResourceResult{
|
|
Headers: respHeaders,
|
|
Body: protoResp.Body,
|
|
Status: int(protoResp.Code),
|
|
}, nil
|
|
}
|
|
|
|
func (s *callResourceResultStreamImpl) Close() error {
|
|
return s.stream.CloseSend()
|
|
}
|
|
|
|
type singleCallResourceResult struct {
|
|
result *CallResourceResult
|
|
}
|
|
|
|
func (s *singleCallResourceResult) Recv() (*CallResourceResult, error) {
|
|
return s.result, nil
|
|
}
|
|
|
|
func (s *singleCallResourceResult) Close() error {
|
|
return nil
|
|
}
|