forked from extern/smegmesh
44 lines
780 B
Go
44 lines
780 B
Go
package crdt
|
|
|
|
import (
|
|
"github.com/automerge/automerge-go"
|
|
logging "github.com/tim-beatham/wgmesh/pkg/log"
|
|
)
|
|
|
|
type AutomergeSync struct {
|
|
state *automerge.SyncState
|
|
manager *CrdtNodeManager
|
|
}
|
|
|
|
func (a *AutomergeSync) GenerateMessage() ([]byte, bool) {
|
|
msg, valid := a.state.GenerateMessage()
|
|
|
|
if !valid {
|
|
return nil, false
|
|
}
|
|
|
|
return msg.Bytes(), true
|
|
}
|
|
|
|
func (a *AutomergeSync) RecvMessage(msg []byte) error {
|
|
_, err := a.state.ReceiveMessage(msg)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *AutomergeSync) Complete() {
|
|
logging.Log.WriteInfof("Sync Completed")
|
|
a.manager.SaveChanges()
|
|
}
|
|
|
|
func NewAutomergeSync(manager *CrdtNodeManager) *AutomergeSync {
|
|
return &AutomergeSync{
|
|
state: automerge.NewSyncState(manager.doc),
|
|
manager: manager,
|
|
}
|
|
}
|