Will Browne
5b67ae1876
* remove secrets manager code * remove unused struct * add test toggles back * rollback golden file changes * add missing TestMain * update betterer
77 lines
2.2 KiB
Go
77 lines
2.2 KiB
Go
package kvstore
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/grafana/grafana/pkg/services/secrets"
|
|
)
|
|
|
|
const (
|
|
// Wildcard to query all organizations
|
|
AllOrganizations = -1
|
|
)
|
|
|
|
func ProvideService(
|
|
sqlStore db.DB,
|
|
secretsService secrets.Service,
|
|
) (SecretsKVStore, error) {
|
|
store := NewSQLSecretsKVStore(sqlStore, secretsService, log.New("secrets.kvstore"))
|
|
return WithCache(store, 5*time.Second, 5*time.Minute), nil
|
|
}
|
|
|
|
// SecretsKVStore is an interface for k/v store.
|
|
type SecretsKVStore interface {
|
|
Get(ctx context.Context, orgId int64, namespace string, typ string) (string, bool, error)
|
|
Set(ctx context.Context, orgId int64, namespace string, typ string, value string) error
|
|
Del(ctx context.Context, orgId int64, namespace string, typ string) error
|
|
Keys(ctx context.Context, orgId int64, namespace string, typ string) ([]Key, error)
|
|
Rename(ctx context.Context, orgId int64, namespace string, typ string, newNamespace string) error
|
|
GetAll(ctx context.Context) ([]Item, error)
|
|
}
|
|
|
|
// WithType returns a kvstore wrapper with fixed orgId and type.
|
|
func With(kv SecretsKVStore, orgId int64, namespace string, typ string) *FixedKVStore {
|
|
return &FixedKVStore{
|
|
kvStore: kv,
|
|
OrgId: orgId,
|
|
Namespace: namespace,
|
|
Type: typ,
|
|
}
|
|
}
|
|
|
|
// FixedKVStore is a SecretsKVStore wrapper with fixed orgId, namespace and type.
|
|
type FixedKVStore struct {
|
|
kvStore SecretsKVStore
|
|
OrgId int64
|
|
Namespace string
|
|
Type string
|
|
}
|
|
|
|
func (kv *FixedKVStore) Get(ctx context.Context) (string, bool, error) {
|
|
return kv.kvStore.Get(ctx, kv.OrgId, kv.Namespace, kv.Type)
|
|
}
|
|
|
|
func (kv *FixedKVStore) Set(ctx context.Context, value string) error {
|
|
return kv.kvStore.Set(ctx, kv.OrgId, kv.Namespace, kv.Type, value)
|
|
}
|
|
|
|
func (kv *FixedKVStore) Del(ctx context.Context) error {
|
|
return kv.kvStore.Del(ctx, kv.OrgId, kv.Namespace, kv.Type)
|
|
}
|
|
|
|
func (kv *FixedKVStore) Keys(ctx context.Context) ([]Key, error) {
|
|
return kv.kvStore.Keys(ctx, kv.OrgId, kv.Namespace, kv.Type)
|
|
}
|
|
|
|
func (kv *FixedKVStore) Rename(ctx context.Context, newNamespace string) error {
|
|
err := kv.kvStore.Rename(ctx, kv.OrgId, kv.Namespace, kv.Type, newNamespace)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
kv.Namespace = newNamespace
|
|
return nil
|
|
}
|