98 lines
3.0 KiB
Go
98 lines
3.0 KiB
Go
package clientspace
|
|
|
|
import (
|
|
"context"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/app"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/app/logger"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacesyncproto"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/storage"
|
|
config2 "github.com/anytypeio/go-anytype-infrastructure-experiments/common/config"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/net/rpc/server"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/ocache"
|
|
"time"
|
|
)
|
|
|
|
const CName = "client.clientspace"
|
|
|
|
var log = logger.NewNamed(CName)
|
|
|
|
func New() Service {
|
|
return &service{}
|
|
}
|
|
|
|
type Service interface {
|
|
GetSpace(ctx context.Context, id string) (commonspace.Space, error)
|
|
CreateSpace(ctx context.Context, payload commonspace.SpaceCreatePayload) (commonspace.Space, error)
|
|
DeriveSpace(ctx context.Context, payload commonspace.SpaceDerivePayload) (commonspace.Space, error)
|
|
app.ComponentRunnable
|
|
}
|
|
|
|
type service struct {
|
|
conf config2.Space
|
|
spaceCache ocache.OCache
|
|
commonSpace commonspace.Service
|
|
spaceStorageProvider storage.SpaceStorageProvider
|
|
}
|
|
|
|
func (s *service) Init(a *app.App) (err error) {
|
|
s.conf = a.MustComponent(config2.CName).(*config2.Config).Space
|
|
s.commonSpace = a.MustComponent(commonspace.CName).(commonspace.Service)
|
|
s.spaceStorageProvider = a.MustComponent(storage.CName).(storage.SpaceStorageProvider)
|
|
s.spaceCache = ocache.New(
|
|
func(ctx context.Context, id string) (value ocache.Object, err error) {
|
|
return s.commonSpace.GetSpace(ctx, id)
|
|
},
|
|
ocache.WithLogger(log.Sugar()),
|
|
ocache.WithGCPeriod(time.Minute),
|
|
ocache.WithTTL(time.Duration(s.conf.GCTTL)*time.Second),
|
|
)
|
|
return spacesyncproto.DRPCRegisterSpace(a.MustComponent(server.CName).(server.DRPCServer), &rpcHandler{s})
|
|
}
|
|
|
|
func (s *service) Name() (name string) {
|
|
return CName
|
|
}
|
|
|
|
func (s *service) Run(ctx context.Context) (err error) {
|
|
return
|
|
}
|
|
|
|
func (s *service) CreateSpace(ctx context.Context, payload commonspace.SpaceCreatePayload) (container commonspace.Space, err error) {
|
|
id, err := s.commonSpace.CreateSpace(ctx, payload)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
obj, err := s.spaceCache.Get(ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return obj.(commonspace.Space), nil
|
|
}
|
|
|
|
func (s *service) DeriveSpace(ctx context.Context, payload commonspace.SpaceDerivePayload) (container commonspace.Space, err error) {
|
|
id, err := s.commonSpace.DeriveSpace(ctx, payload)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
obj, err := s.spaceCache.Get(ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return obj.(commonspace.Space), nil
|
|
}
|
|
|
|
func (s *service) GetSpace(ctx context.Context, id string) (container commonspace.Space, err error) {
|
|
v, err := s.spaceCache.Get(ctx, id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return v.(commonspace.Space), nil
|
|
}
|
|
|
|
func (s *service) Close(ctx context.Context) (err error) {
|
|
return s.spaceCache.Close()
|
|
}
|