mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 05:29:42 -06:00
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package live
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
|
|
"github.com/centrifugal/centrifuge"
|
|
"github.com/grafana/grafana-plugin-sdk-go/backend"
|
|
"github.com/grafana/grafana/pkg/plugins/plugincontext"
|
|
)
|
|
|
|
type pluginPacketSender struct {
|
|
node *centrifuge.Node
|
|
}
|
|
|
|
func newPluginPacketSender(node *centrifuge.Node) *pluginPacketSender {
|
|
return &pluginPacketSender{node: node}
|
|
}
|
|
|
|
func (p *pluginPacketSender) Send(channel string, packet *backend.StreamPacket) error {
|
|
_, err := p.node.Publish(channel, packet.Data)
|
|
if err != nil {
|
|
return fmt.Errorf("error publishing %s: %w", string(packet.Data), err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
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(user *models.SignedInUser, pluginID string, datasourceUID string) (backend.PluginContext, bool, error) {
|
|
return g.PluginContextProvider.Get(pluginID, datasourceUID, user)
|
|
}
|