Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 41 additions & 8 deletions pkg/scd/store/raftstore/store.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,37 +3,70 @@ package raftstore
import (
"context"

dsserr "github.com/interuss/dss/pkg/errors"
"github.com/interuss/dss/pkg/memstore"
"github.com/interuss/dss/pkg/raftstore"
"github.com/interuss/dss/pkg/raftstore/consensus"
"github.com/interuss/dss/pkg/scd/actions"
"github.com/interuss/dss/pkg/scd/repos"
scdmemstore "github.com/interuss/dss/pkg/scd/store/memstore"
scdraftparams "github.com/interuss/dss/pkg/scd/store/raftstore/params"
"github.com/interuss/stacktrace"
"go.uber.org/zap"
)

// repo is a full implementation of scd.repos.Repository for Raft-based storage.
type repo struct{}
type repo struct {
consensus *consensus.Consensus
memStore *memstore.Store[repos.Repository]
memRepo repos.Repository
}

func Init(ctx context.Context, logger *zap.Logger, locality string) (*raftstore.Store[repos.Repository], error) {
params, err := scdraftparams.GetConnectParameters()
if err != nil {
return nil, stacktrace.Propagate(err, "failed to get scd raft parameters")
}
return raftstore.Init(ctx, logger.With(zap.String("service", "scd")), locality, params, &repo{}, actions.Registry)

memStore, err := scdmemstore.Init(ctx, logger)
if err != nil {
return nil, stacktrace.Propagate(err, "failed to initialize scd memstore")
}

r := &repo{memStore: memStore, memRepo: memStore.GetRepo()}
store, err := raftstore.Init(ctx, logger.With(zap.String("service", "scd")), locality, params, r, actions.Registry)
if err != nil {
return nil, stacktrace.Propagate(err, "failed to initialize scd raftstore")
}

r.consensus = store.Consensus

return store, nil
}

func (r *repo) GetRepo() repos.Repository { return r }

func (r *repo) GetSnapshot() ([]byte, error) {
return nil, stacktrace.NewErrorWithCode(dsserr.NotImplemented, "not implemented yet")
return r.memStore.GetSnapshot()
}

func (r *repo) RestoreFromSnapshot([]byte) error {
return stacktrace.NewErrorWithCode(dsserr.NotImplemented, "not implemented yet")
func (r *repo) RestoreFromSnapshot(data []byte) error {
return r.memStore.RestoreFromSnapshot(data)
}

func (r *repo) Apply(_ context.Context, _ consensus.Proposal) (any, error) {
return nil, stacktrace.NewErrorWithCode(dsserr.NotImplemented, "not implemented yet")
func (r *repo) Apply(ctx context.Context, proposal consensus.Proposal) (any, error) {
switch proposal.RequestType {

default:
handler, ok := actions.Registry[string(proposal.RequestType)]
if !ok {
return nil, stacktrace.NewError("unrecognized request type: %s", proposal.RequestType)
}

request, err := handler.Decode(proposal.Value)
if err != nil {
return nil, stacktrace.Propagate(err, "failed to decode %s payload", proposal.RequestType)
}

return handler.Execute(ctx, r.memRepo, request)
}
}
Loading