mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 05:29:42 -06:00
05709ce411
* chore: add alias for InitTestDB and Session Adds an alias for the sqlstore InitTestDB and Session, and updates tests using these to reduce dependencies on the sqlstore.Store. * next pass of removing sqlstore imports * last little bit * remove mockstore where possible
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package kvstore
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
)
|
|
|
|
const (
|
|
// Wildcard to query all organizations
|
|
AllOrganizations = -1
|
|
)
|
|
|
|
func ProvideService(sqlStore db.DB) KVStore {
|
|
return &kvStoreSQL{
|
|
sqlStore: sqlStore,
|
|
log: log.New("infra.kvstore.sql"),
|
|
}
|
|
}
|
|
|
|
// KVStore is an interface for k/v store.
|
|
type KVStore interface {
|
|
Get(ctx context.Context, orgId int64, namespace string, key string) (string, bool, error)
|
|
Set(ctx context.Context, orgId int64, namespace string, key string, value string) error
|
|
Del(ctx context.Context, orgId int64, namespace string, key string) error
|
|
Keys(ctx context.Context, orgId int64, namespace string, keyPrefix string) ([]Key, error)
|
|
GetAll(ctx context.Context, orgId int64, namespace string) (map[int64]map[string]string, error)
|
|
}
|
|
|
|
// WithNamespace returns a kvstore wrapper with fixed orgId and namespace.
|
|
func WithNamespace(kv KVStore, orgId int64, namespace string) *NamespacedKVStore {
|
|
return &NamespacedKVStore{
|
|
kvStore: kv,
|
|
orgId: orgId,
|
|
namespace: namespace,
|
|
}
|
|
}
|
|
|
|
// NamespacedKVStore is a KVStore wrapper with fixed orgId and namespace.
|
|
type NamespacedKVStore struct {
|
|
kvStore KVStore
|
|
orgId int64
|
|
namespace string
|
|
}
|
|
|
|
func (kv *NamespacedKVStore) Get(ctx context.Context, key string) (string, bool, error) {
|
|
return kv.kvStore.Get(ctx, kv.orgId, kv.namespace, key)
|
|
}
|
|
|
|
func (kv *NamespacedKVStore) Set(ctx context.Context, key string, value string) error {
|
|
return kv.kvStore.Set(ctx, kv.orgId, kv.namespace, key, value)
|
|
}
|
|
|
|
func (kv *NamespacedKVStore) Del(ctx context.Context, key string) error {
|
|
return kv.kvStore.Del(ctx, kv.orgId, kv.namespace, key)
|
|
}
|
|
|
|
func (kv *NamespacedKVStore) Keys(ctx context.Context, keyPrefix string) ([]Key, error) {
|
|
return kv.kvStore.Keys(ctx, kv.orgId, kv.namespace, keyPrefix)
|
|
}
|
|
|
|
// GetAll returns all the keys and values stored per organization. It returns a map of org -> key -> value.
|
|
func (kv *NamespacedKVStore) GetAll(ctx context.Context) (map[int64]map[string]string, error) {
|
|
return kv.kvStore.GetAll(ctx, kv.orgId, kv.namespace)
|
|
}
|