51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
|
// Copyright (c) 2016 Bitfury Group Limited
|
||
|
// Distributed under the MIT software license, see the accompanying
|
||
|
// file LICENSE or http://www.opensource.org/licenses/mit-license.php
|
||
|
|
||
|
package lnwire
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"github.com/BitfuryLightning/tools/rt"
|
||
|
"encoding/gob"
|
||
|
"io"
|
||
|
)
|
||
|
|
||
|
type NeighborUpdMessage struct {
|
||
|
RoutingMessageBase
|
||
|
DiffBuff *rt.DifferenceBuffer
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) Decode(r io.Reader, pver uint32) error{
|
||
|
decoder := gob.NewDecoder(r)
|
||
|
diffBuff := new(rt.DifferenceBuffer)
|
||
|
err := decoder.Decode(diffBuff)
|
||
|
msg.DiffBuff = diffBuff
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) Encode(w io.Writer, pver uint32) error{
|
||
|
encoder := gob.NewEncoder(w)
|
||
|
err := encoder.Encode(msg.DiffBuff)
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) Command() uint32{
|
||
|
return CmdNeighborUpdMessage
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) MaxPayloadLength(uint32) uint32{
|
||
|
// TODO: Insert some estimations
|
||
|
return 1000000
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) Validate() error{
|
||
|
// TODO: Add validation
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (msg *NeighborUpdMessage) String() string {
|
||
|
return fmt.Sprintf("NeighborUpdMessage{%v %v %v}", msg.SenderID, msg.ReceiverID, *msg.DiffBuff)
|
||
|
}
|
||
|
|
||
|
var _ Message = (*NeighborUpdMessage)(nil)
|