49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
|
|
simplejson "github.com/bitly/go-simplejson"
|
|
"github.com/grafana/grafana_plugin_model/go/datasource"
|
|
hclog "github.com/hashicorp/go-hclog"
|
|
plugin "github.com/hashicorp/go-plugin"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type ZabbixDatasource struct {
|
|
plugin.NetRPCUnsupportedPlugin
|
|
logger hclog.Logger
|
|
}
|
|
|
|
func (ds *ZabbixDatasource) Query(ctx context.Context, tsdbReq *datasource.DatasourceRequest) (*datasource.DatasourceResponse, error) {
|
|
queryType, err := GetQueryType(tsdbReq)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dsInfo := tsdbReq.GetDatasource()
|
|
ds.logger.Debug("createRequest", "dsInfo", dsInfo)
|
|
|
|
ds.logger.Debug("createRequest", "queryType", queryType)
|
|
|
|
switch queryType {
|
|
case "zabbixAPI":
|
|
return ds.ZabbixAPIQuery(ctx, tsdbReq)
|
|
default:
|
|
return nil, errors.New("Query is not implemented yet")
|
|
}
|
|
}
|
|
|
|
func GetQueryType(tsdbReq *datasource.DatasourceRequest) (string, error) {
|
|
queryType := "query"
|
|
if len(tsdbReq.Queries) > 0 {
|
|
firstQuery := tsdbReq.Queries[0]
|
|
queryJson, err := simplejson.NewJson([]byte(firstQuery.ModelJson))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
queryType = queryJson.Get("queryType").MustString("query")
|
|
}
|
|
return queryType, nil
|
|
}
|