82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
package node
|
|
|
|
import (
|
|
"lux/proto"
|
|
|
|
ipnet "net"
|
|
)
|
|
|
|
type LuxNodeSync struct {
|
|
Synced map[proto.LuxID]struct{}
|
|
Neighbors map[proto.LuxID]*ipnet.UDPAddr
|
|
SyncState LuxNodeState
|
|
}
|
|
|
|
func NewLuxNodeSync() LuxNodeSync {
|
|
return LuxNodeSync{
|
|
Synced: make(map[proto.LuxID]struct{}),
|
|
Neighbors: make(map[proto.LuxID]*ipnet.UDPAddr),
|
|
SyncState: LuxNodeState{},
|
|
}
|
|
}
|
|
|
|
func (sync *LuxNodeSync) Read(rd *proto.LuxBuffer) error {
|
|
// read synced list
|
|
syncNum, err := rd.ReadUint16()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for i := 0; i < int(syncNum); i++ {
|
|
id := proto.LuxID{}
|
|
if err := id.Read(rd); err != nil {
|
|
return err
|
|
}
|
|
|
|
sync.Synced[id] = struct{}{}
|
|
}
|
|
|
|
// read neighbors
|
|
neighborNum, err := rd.ReadUint16()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for i := 0; i < int(neighborNum); i++ {
|
|
id := proto.LuxID{}
|
|
if err := id.Read(rd); err != nil {
|
|
return err
|
|
}
|
|
|
|
addrPort, err := rd.ReadIPPort()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sync.Neighbors[id] = ipnet.UDPAddrFromAddrPort(addrPort)
|
|
}
|
|
|
|
// read sync state
|
|
err = sync.SyncState.Read(rd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (sync *LuxNodeSync) Write(wd *proto.LuxBuffer) {
|
|
// write synced list
|
|
wd.WriteUint16(uint16(len(sync.Synced)))
|
|
for id := range sync.Synced {
|
|
id.Write(wd)
|
|
}
|
|
|
|
// write neighbors
|
|
wd.WriteUint16(uint16(len(sync.Neighbors)))
|
|
for id, udpAddr := range sync.Neighbors {
|
|
id.Write(wd)
|
|
wd.WriteIPPort(udpAddr.AddrPort())
|
|
}
|
|
|
|
// write sync state
|
|
sync.SyncState.Write(wd)
|
|
}
|