mirror of https://github.com/grafana/grafana.git
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package live
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/centrifugal/centrifuge"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana/pkg/plugins/plugincontext"
|
|
)
|
|
|
|
type pluginChannelPublisher struct {
|
|
node *centrifuge.Node
|
|
}
|
|
|
|
func newPluginChannelPublisher(node *centrifuge.Node) *pluginChannelPublisher {
|
|
return &pluginChannelPublisher{node: node}
|
|
}
|
|
|
|
func (p *pluginChannelPublisher) Publish(channel string, data []byte) error {
|
|
_, err := p.node.Publish(channel, data)
|
|
return err
|
|
}
|
|
|
|
type pluginPresenceGetter struct {
|
|
node *centrifuge.Node
|
|
}
|
|
|
|
func newPluginPresenceGetter(node *centrifuge.Node) *pluginPresenceGetter {
|
|
return &pluginPresenceGetter{node: node}
|
|
}
|
|
|
|
func (p *pluginPresenceGetter) GetNumSubscribers(channel string) (int, error) {
|
|
res, err := p.node.PresenceStats(channel)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return res.NumClients, nil
|
|
}
|
|
|
|
type pluginContextGetter struct {
|
|
PluginContextProvider *plugincontext.Provider
|
|
}
|
|
|
|
func newPluginContextGetter(pluginContextProvider *plugincontext.Provider) *pluginContextGetter {
|
|
return &pluginContextGetter{
|
|
PluginContextProvider: pluginContextProvider,
|
|
}
|
|
}
|
|
|
|
func (g *pluginContextGetter) GetPluginContext(ctx context.Context, pluginID string, datasourceUID string) (backend.PluginContext, bool, error) {
|
|
user, ok := getContextSignedUser(ctx)
|
|
if !ok {
|
|
return backend.PluginContext{}, false, fmt.Errorf("no signed user found in context")
|
|
}
|
|
return g.PluginContextProvider.Get(pluginID, datasourceUID, user)
|
|
}
|