keyring_store.go

package etcd

import (
	"context"
	"fmt"
	"path"

	clientv3 "go.etcd.io/etcd/client/v3"

	corev1 "github.com/rancher/opni/pkg/apis/core/v1"
	"github.com/rancher/opni/pkg/keyring"
	"github.com/rancher/opni/pkg/storage"
)

type etcdKeyringStore struct {
	client *clientv3.Client
	ref    *corev1.Reference
	prefix string
}

func (ks *etcdKeyringStore) Put(ctx context.Context, keyring keyring.Keyring) error {
	k, err := keyring.Marshal()
	if err != nil {
		return fmt.Errorf("failed to marshal keyring: %w", err)
	}
	_, err = ks.client.Put(ctx, path.Join(ks.prefix, keyringKey, ks.ref.Id), string(k))
	if err != nil {
		return fmt.Errorf("failed to put keyring: %w", err)
	}
	return nil
}

func (ks *etcdKeyringStore) Get(ctx context.Context) (keyring.Keyring, error) {
	resp, err := ks.client.Get(ctx, path.Join(ks.prefix, keyringKey, ks.ref.Id))
	if err != nil {
		return nil, fmt.Errorf("failed to get keyring: %w", err)
	}
	if len(resp.Kvs) == 0 {
		return nil, storage.ErrNotFound
	}
	k, err := keyring.Unmarshal(resp.Kvs[0].Value)
	if err != nil {
		return nil, fmt.Errorf("failed to unmarshal keyring: %w", err)
	}
	return k, nil
}

func (ks *etcdKeyringStore) Delete(ctx context.Context) error {
	resp, err := ks.client.Delete(ctx, path.Join(ks.prefix, keyringKey, ks.ref.Id))
	if err != nil {
		return fmt.Errorf("failed to delete keyring: %w", err)
	}
	if resp.Deleted == 0 {
		return storage.ErrNotFound
	}
	return nil
}