merge backend datasources and datasources

This commit is contained in:
Carl Bergquist 2017-12-20 16:24:49 +01:00
parent 8b37e27dde
commit 20284f7745
3 changed files with 23 additions and 97 deletions

View File

@ -1,85 +0,0 @@
package plugins
import (
"context"
"encoding/json"
"os/exec"
"path"
"github.com/grafana/grafana/pkg/models"
"github.com/grafana/grafana/pkg/plugins/backend"
"github.com/grafana/grafana/pkg/tsdb"
"github.com/grafana/grafana/pkg/log"
proto "github.com/grafana/grafana/pkg/tsdb/models"
shared "github.com/grafana/grafana/pkg/tsdb/models/proxy"
plugin "github.com/hashicorp/go-plugin"
)
type BackendDatasource struct {
*PluginBase
Executable string
log log.Logger
client *plugin.Client
}
var handshakeConfig = plugin.HandshakeConfig{
ProtocolVersion: 1,
MagicCookieKey: "GRAFANA_BACKEND_DATASOURCE",
MagicCookieValue: "55d2200a-6492-493a-9353-73b728d468aa",
}
func (p *BackendDatasource) initBackendPlugin(log log.Logger) error {
p.log = log.New("plugin-id", p.Id)
p.client = plugin.NewClient(&plugin.ClientConfig{
HandshakeConfig: handshakeConfig,
Plugins: map[string]plugin.Plugin{p.Id: &shared.TsdbPluginImpl{}},
Cmd: exec.Command(path.Join(p.PluginDir, p.Executable)),
AllowedProtocols: []plugin.Protocol{plugin.ProtocolGRPC},
Logger: backend.LogWrapper{Logger: p.log},
})
rpcClient, err := p.client.Client()
if err != nil {
return err
}
raw, err := rpcClient.Dispense(p.Id)
if err != nil {
return err
}
plugin := raw.(shared.TsdbPlugin)
response, err := plugin.Query(context.Background(), &proto.TsdbQuery{})
if err != nil {
p.log.Error("Response from plugin. ", "response", response)
} else {
p.log.Info("Response from plugin. ", "response", response)
}
tsdb.RegisterTsdbQueryEndpoint(p.Id, func(dsInfo *models.DataSource) (tsdb.TsdbQueryEndpoint, error) {
return &shared.TsdbWrapper{TsdbPlugin: plugin}, nil
})
return nil
}
func (p *BackendDatasource) Kill() {
p.client.Kill()
}
func (p *BackendDatasource) Load(decoder *json.Decoder, pluginDir string) error {
if err := decoder.Decode(&p); err != nil {
return err
}
if err := p.registerPlugin(pluginDir); err != nil {
return err
}
BackendDatasources[p.Id] = p
return nil
}

View File

@ -2,10 +2,13 @@ package plugins
import ( import (
"encoding/json" "encoding/json"
"fmt"
"os" "os"
"os/exec" "os/exec"
"path" "path"
"path/filepath" "path/filepath"
"runtime"
"strings"
"github.com/grafana/grafana/pkg/log" "github.com/grafana/grafana/pkg/log"
"github.com/grafana/grafana/pkg/models" "github.com/grafana/grafana/pkg/models"
@ -55,13 +58,21 @@ func (p *DataSourcePlugin) Load(decoder *json.Decoder, pluginDir string) error {
return nil return nil
} }
var handshakeConfig = plugin.HandshakeConfig{
ProtocolVersion: 1,
MagicCookieKey: "GRAFANA_BACKEND_DATASOURCE",
MagicCookieValue: "55d2200a-6492-493a-9353-73b728d468aa",
}
func (p *DataSourcePlugin) initBackendPlugin(log log.Logger) error { func (p *DataSourcePlugin) initBackendPlugin(log log.Logger) error {
p.log = log.New("plugin-id", p.Id) p.log = log.New("plugin-id", p.Id)
cmd := path.Join(p.PluginDir, fmt.Sprintf("%s_%s_%s", p.Executable, strings.ToLower(runtime.GOOS), strings.ToLower(runtime.GOARCH)))
p.client = plugin.NewClient(&plugin.ClientConfig{ p.client = plugin.NewClient(&plugin.ClientConfig{
HandshakeConfig: handshakeConfig, HandshakeConfig: handshakeConfig,
Plugins: map[string]plugin.Plugin{p.Id: &shared.TsdbPluginImpl{}}, Plugins: map[string]plugin.Plugin{p.Id: &shared.TsdbPluginImpl{}},
Cmd: exec.Command(path.Join(p.PluginDir, p.Executable)), Cmd: exec.Command(cmd),
AllowedProtocols: []plugin.Protocol{plugin.ProtocolGRPC}, AllowedProtocols: []plugin.Protocol{plugin.ProtocolGRPC},
Logger: backend.LogWrapper{Logger: p.log}, Logger: backend.LogWrapper{Logger: p.log},
}) })
@ -86,5 +97,7 @@ func (p *DataSourcePlugin) initBackendPlugin(log log.Logger) error {
} }
func (p *DataSourcePlugin) Kill() { func (p *DataSourcePlugin) Kill() {
if p.client != nil {
p.client.Kill() p.client.Kill()
} }
}

View File

@ -23,7 +23,7 @@ var (
StaticRoutes []*PluginStaticRoute StaticRoutes []*PluginStaticRoute
Apps map[string]*AppPlugin Apps map[string]*AppPlugin
Plugins map[string]*PluginBase Plugins map[string]*PluginBase
BackendDatasources map[string]*BackendDatasource //BackendDatasources map[string]*BackendDatasource
PluginTypes map[string]interface{} PluginTypes map[string]interface{}
GrafanaLatestVersion string GrafanaLatestVersion string
@ -52,7 +52,7 @@ func NewPluginManager() (*PluginManager, error) {
func (p *PluginManager) Run(ctx context.Context) error { func (p *PluginManager) Run(ctx context.Context) error {
<-ctx.Done() <-ctx.Done()
for _, p := range BackendDatasources { for _, p := range DataSources {
p.Kill() p.Kill()
} }
@ -68,12 +68,10 @@ func Init() error {
Panels = make(map[string]*PanelPlugin) Panels = make(map[string]*PanelPlugin)
Apps = make(map[string]*AppPlugin) Apps = make(map[string]*AppPlugin)
Plugins = make(map[string]*PluginBase) Plugins = make(map[string]*PluginBase)
BackendDatasources = make(map[string]*BackendDatasource)
PluginTypes = map[string]interface{}{ PluginTypes = map[string]interface{}{
"panel": PanelPlugin{}, "panel": PanelPlugin{},
"datasource": DataSourcePlugin{}, "datasource": DataSourcePlugin{},
"app": AppPlugin{}, "app": AppPlugin{},
//"backend-datasource": BackendDatasource{},
} }
plog.Info("Starting plugin search") plog.Info("Starting plugin search")