98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
//go:generate mockgen -destination mock_objectsync/mock_objectsync.go github.com/anytypeio/any-sync/commonspace/objectsync ActionQueue
|
|
package objectsync
|
|
|
|
import (
|
|
"context"
|
|
"github.com/anytypeio/any-sync/app/logger"
|
|
"github.com/anytypeio/any-sync/app/ocache"
|
|
"github.com/anytypeio/any-sync/commonspace/object/syncobjectgetter"
|
|
"github.com/anytypeio/any-sync/commonspace/objectsync/synchandler"
|
|
"github.com/anytypeio/any-sync/commonspace/spacesyncproto"
|
|
"go.uber.org/zap"
|
|
"time"
|
|
)
|
|
|
|
var log = logger.NewNamed("commonspace.objectsync")
|
|
|
|
type ObjectSync interface {
|
|
ocache.ObjectLastUsage
|
|
synchandler.SyncHandler
|
|
MessagePool() MessagePool
|
|
ActionQueue() ActionQueue
|
|
|
|
Init(getter syncobjectgetter.SyncObjectGetter)
|
|
Close() (err error)
|
|
}
|
|
|
|
type objectSync struct {
|
|
spaceId string
|
|
|
|
streamPool MessagePool
|
|
objectGetter syncobjectgetter.SyncObjectGetter
|
|
actionQueue ActionQueue
|
|
|
|
syncCtx context.Context
|
|
cancelSync context.CancelFunc
|
|
}
|
|
|
|
func NewObjectSync(streamManager StreamManager, spaceId string) (objectSync ObjectSync) {
|
|
msgPool := newMessagePool(streamManager, func(ctx context.Context, senderId string, message *spacesyncproto.ObjectSyncMessage) (err error) {
|
|
return objectSync.HandleMessage(ctx, senderId, message)
|
|
})
|
|
syncCtx, cancel := context.WithCancel(context.Background())
|
|
objectSync = newObjectSync(
|
|
spaceId,
|
|
msgPool,
|
|
syncCtx,
|
|
cancel)
|
|
return
|
|
}
|
|
|
|
func newObjectSync(
|
|
spaceId string,
|
|
streamPool MessagePool,
|
|
syncCtx context.Context,
|
|
cancel context.CancelFunc,
|
|
) *objectSync {
|
|
return &objectSync{
|
|
streamPool: streamPool,
|
|
spaceId: spaceId,
|
|
syncCtx: syncCtx,
|
|
cancelSync: cancel,
|
|
actionQueue: NewDefaultActionQueue(),
|
|
}
|
|
}
|
|
|
|
func (s *objectSync) Init(objectGetter syncobjectgetter.SyncObjectGetter) {
|
|
s.objectGetter = objectGetter
|
|
s.actionQueue.Run()
|
|
}
|
|
|
|
func (s *objectSync) Close() (err error) {
|
|
s.actionQueue.Close()
|
|
s.cancelSync()
|
|
return
|
|
}
|
|
|
|
func (s *objectSync) LastUsage() time.Time {
|
|
// TODO: [che]
|
|
return time.Now()
|
|
}
|
|
|
|
func (s *objectSync) HandleMessage(ctx context.Context, senderId string, message *spacesyncproto.ObjectSyncMessage) (err error) {
|
|
log.With(zap.String("peerId", senderId), zap.String("objectId", message.ObjectId)).Debug("handling message")
|
|
obj, err := s.objectGetter.GetObject(ctx, message.ObjectId)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return obj.HandleMessage(ctx, senderId, message)
|
|
}
|
|
|
|
func (s *objectSync) MessagePool() MessagePool {
|
|
return s.streamPool
|
|
}
|
|
|
|
func (s *objectSync) ActionQueue() ActionQueue {
|
|
return s.actionQueue
|
|
}
|