64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package synctree
|
|
|
|
import (
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/diffservice"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacesyncproto"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/syncservice"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/nodeconf"
|
|
"github.com/anytypeio/go-anytype-infrastructure-experiments/common/pkg/ocache"
|
|
"time"
|
|
)
|
|
|
|
type SyncClient interface {
|
|
syncservice.StreamPool
|
|
RequestFactory
|
|
ocache.ObjectLastUsage
|
|
BroadcastAsyncOrSendResponsible(message *spacesyncproto.ObjectSyncMessage) (err error)
|
|
}
|
|
|
|
type syncClient struct {
|
|
syncservice.StreamPool
|
|
RequestFactory
|
|
spaceId string
|
|
notifiable diffservice.HeadNotifiable
|
|
configuration nodeconf.Configuration
|
|
}
|
|
|
|
func newSyncClient(
|
|
spaceId string,
|
|
pool syncservice.StreamPool,
|
|
notifiable diffservice.HeadNotifiable,
|
|
factory RequestFactory,
|
|
configuration nodeconf.Configuration) SyncClient {
|
|
return &syncClient{
|
|
StreamPool: pool,
|
|
RequestFactory: factory,
|
|
notifiable: notifiable,
|
|
configuration: configuration,
|
|
spaceId: spaceId,
|
|
}
|
|
}
|
|
|
|
func (s *syncClient) LastUsage() time.Time {
|
|
return s.StreamPool.LastUsage()
|
|
}
|
|
|
|
func (s *syncClient) BroadcastAsync(message *spacesyncproto.ObjectSyncMessage) (err error) {
|
|
s.notifyIfNeeded(message)
|
|
return s.StreamPool.BroadcastAsync(message)
|
|
}
|
|
|
|
func (s *syncClient) BroadcastAsyncOrSendResponsible(message *spacesyncproto.ObjectSyncMessage) (err error) {
|
|
if s.configuration.IsResponsible(s.spaceId) {
|
|
return s.SendAsync(s.configuration.NodeIds(s.spaceId), message)
|
|
}
|
|
return s.BroadcastAsync(message)
|
|
}
|
|
|
|
func (s *syncClient) notifyIfNeeded(message *spacesyncproto.ObjectSyncMessage) {
|
|
if message.GetContent().GetHeadUpdate() != nil {
|
|
update := message.GetContent().GetHeadUpdate()
|
|
s.notifiable.UpdateHeads(message.TreeId, update.Heads)
|
|
}
|
|
}
|