82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"git.sad.ovh/sophie/nuke/internal/actorify"
|
|
)
|
|
|
|
type unit struct{}
|
|
|
|
type ActorifiedStore struct {
|
|
Interface
|
|
|
|
deleteActor *actorify.Actor[string, unit]
|
|
getActor *actorify.Actor[string, []byte]
|
|
setActor *actorify.Actor[*actorSetReq, unit]
|
|
cancel context.CancelFunc
|
|
}
|
|
|
|
type actorSetReq struct {
|
|
key string
|
|
value []byte
|
|
expiry time.Duration
|
|
}
|
|
|
|
func NewActorifiedStore(backend Interface) *ActorifiedStore {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
result := &ActorifiedStore{
|
|
Interface: backend,
|
|
cancel: cancel,
|
|
}
|
|
|
|
result.deleteActor = actorify.New(ctx, result.actorDelete)
|
|
result.getActor = actorify.New(ctx, backend.Get)
|
|
result.setActor = actorify.New(ctx, result.actorSet)
|
|
|
|
return result
|
|
}
|
|
|
|
func (a *ActorifiedStore) Close() { a.cancel() }
|
|
|
|
func (a *ActorifiedStore) Delete(ctx context.Context, key string) error {
|
|
if _, err := a.deleteActor.Call(ctx, key); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *ActorifiedStore) Get(ctx context.Context, key string) ([]byte, error) {
|
|
return a.getActor.Call(ctx, key)
|
|
}
|
|
|
|
func (a *ActorifiedStore) Set(ctx context.Context, key string, value []byte, expiry time.Duration) error {
|
|
if _, err := a.setActor.Call(ctx, &actorSetReq{
|
|
key: key,
|
|
value: value,
|
|
expiry: expiry,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *ActorifiedStore) actorDelete(ctx context.Context, key string) (unit, error) {
|
|
if err := a.Interface.Delete(ctx, key); err != nil {
|
|
return unit{}, err
|
|
}
|
|
|
|
return unit{}, nil
|
|
}
|
|
|
|
func (a *ActorifiedStore) actorSet(ctx context.Context, req *actorSetReq) (unit, error) {
|
|
if err := a.Interface.Set(ctx, req.key, req.value, req.expiry); err != nil {
|
|
return unit{}, err
|
|
}
|
|
|
|
return unit{}, nil
|
|
}
|