mirror of
https://github.com/grafana/grafana.git
synced 2025-09-25 05:14:00 +08:00
7536647ab6
* introduce registry write/read separation * internal + external registries * fix tests * fixup * rename * move interfaces * back to plugins.Store * fix registry name * remove context.TODOs * remove some ctx for now * tidy * remove dupe logic * update naming * move from manager.go to store * amend logger name * new store writer svc * restrict changes * more simplifying * move interfaces around * remove unused * fix linter * tidy * add registry test * fix tests * revert testdata changes * revert testdata changes #1 * revert testdata changes #2 * revert testdata changes #3 * revert testdata changes #4 * revert testdata changes #5 * revert testdata changes * fixup testdata * remove unused log * update naming in test * adjust ctx in test
82 lines
1.4 KiB
Go
82 lines
1.4 KiB
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
)
|
|
|
|
type InMemory struct {
|
|
store map[string]*plugins.Plugin
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
func ProvideService() *InMemory {
|
|
return NewInMemory()
|
|
}
|
|
|
|
func NewInMemory() *InMemory {
|
|
return &InMemory{
|
|
store: make(map[string]*plugins.Plugin),
|
|
}
|
|
}
|
|
|
|
func (i *InMemory) Plugin(_ context.Context, pluginID string) (*plugins.Plugin, bool) {
|
|
return i.plugin(pluginID)
|
|
}
|
|
|
|
func (i *InMemory) Plugins(_ context.Context) []*plugins.Plugin {
|
|
i.mu.RLock()
|
|
defer i.mu.RUnlock()
|
|
|
|
res := make([]*plugins.Plugin, 0)
|
|
for _, p := range i.store {
|
|
res = append(res, p)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func (i *InMemory) Add(_ context.Context, p *plugins.Plugin) error {
|
|
if i.isRegistered(p.ID) {
|
|
return fmt.Errorf("plugin %s is already registered", p.ID)
|
|
}
|
|
|
|
i.mu.Lock()
|
|
i.store[p.ID] = p
|
|
i.mu.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (i *InMemory) Remove(_ context.Context, pluginID string) error {
|
|
if !i.isRegistered(pluginID) {
|
|
return fmt.Errorf("plugin %s is not registered", pluginID)
|
|
}
|
|
|
|
i.mu.Lock()
|
|
delete(i.store, pluginID)
|
|
i.mu.Unlock()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (i *InMemory) plugin(pluginID string) (*plugins.Plugin, bool) {
|
|
i.mu.RLock()
|
|
defer i.mu.RUnlock()
|
|
p, exists := i.store[pluginID]
|
|
|
|
if !exists {
|
|
return nil, false
|
|
}
|
|
|
|
return p, true
|
|
}
|
|
|
|
func (i *InMemory) isRegistered(pluginID string) bool {
|
|
_, exists := i.plugin(pluginID)
|
|
return exists
|
|
}
|