跳转到主要内容

为您的数据源后端添加查询路由器或复用器

通常您在您的后端插件中实现 QueryData 方法以进行数据查询。但如果您需要支持不同类型的查询,比如指标、日志和跟踪,该怎么办呢?这就是查询 路由器(也称为 复用器)的用途所在。

插件开发要求是在客户端填充您的查询模型客户端的 queryType 属性,请参阅 DataQuery 接口。

在查询中填充 queryType 并将其发送到您的后端插件后,以下是如何使用 datasource.QueryTypeMux 来复用或路由不同查询类型的示例。

按照这种方式实现,每个查询处理器都可以使用 json.Unmarshal 将每个查询 JSON 字段在 DataQuery 中解析到某个 Go 结构体,如本示例所示

package mydatasource

import (
"context"

"github.com/grafana/grafana-plugin-sdk-go/backend"
"github.com/grafana/grafana-plugin-sdk-go/backend/datasource"
)

type MyDatasource struct {
queryHandler backend.QueryDataHandler
}

func New() *MyDatasource {
ds := &MyDatasource{}
queryTypeMux := datasource.NewQueryTypeMux()
queryTypeMux.HandleFunc("metrics", ds.handleMetricsQuery)
queryTypeMux.HandleFunc("logs", ds.handleLogsQuery)
queryTypeMux.HandleFunc("traces", ds.handleTracesQuery)
queryTypeMux.HandleFunc("", ds.handleQueryFallback)
ds.queryHandler := queryTypeMux
return ds
}

func (d *MyDatasource) QueryData(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
return d.queryHandler.QueryData(ctx, req)
}

// handleMetricsQuery handle queries of query type "metrics".
// All queries in backend.QueryDataRequest is guaranteed to only
// include queries with queryType "metrics".
func (d *MyDatasource) handleMetricsQuery(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
// implementation...
}

// handleLogsQuery handle queries of query type "logs".
// All queries in backend.QueryDataRequest is guaranteed to only
// include queries with queryType "logs".
func (d *MyDatasource) handleLogsQuery(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
// implementation...
}

// handleTracesQuery handle queries of query type "logs".
// All queries in backend.QueryDataRequest is guaranteed to only
// include queries with queryType "traces".
func (d *MyDatasource) handleTracesQuery(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
// implementation...
}

// handleQueryFallback handle queries without a matching query type handler registered.
func (d *MyDatasource) handleQueryFallback(ctx context.Context, req *backend.QueryDataRequest) (*backend.QueryDataResponse, error) {
// implementation...
}

高级用法

在 Grafana 的内置 TestData 数据源中,可以找到一个使用 QueryTypeMux 的例子。请参考此代码中的实现示例