mirror of
https://github.com/grafana/grafana.git
synced 2024-11-29 12:14:08 -06:00
38 lines
827 B
Go
38 lines
827 B
Go
package pipeline
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/grafana/grafana/pkg/services/live/orgchannel"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/data"
|
|
)
|
|
|
|
// FrameStorage keeps last channel frame in memory. Not usable in HA setup.
|
|
type FrameStorage struct {
|
|
mu sync.RWMutex
|
|
frames map[string]*data.Frame
|
|
}
|
|
|
|
func NewFrameStorage() *FrameStorage {
|
|
return &FrameStorage{
|
|
frames: map[string]*data.Frame{},
|
|
}
|
|
}
|
|
|
|
func (s *FrameStorage) Set(orgID int64, channel string, frame *data.Frame) error {
|
|
key := orgchannel.PrependOrgID(orgID, channel)
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
s.frames[key] = frame
|
|
return nil
|
|
}
|
|
|
|
func (s *FrameStorage) Get(orgID int64, channel string) (*data.Frame, bool, error) {
|
|
key := orgchannel.PrependOrgID(orgID, channel)
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
f, ok := s.frames[key]
|
|
return f, ok, nil
|
|
}
|