Added initiator field to peer to detect in the peer initiated the connected.

This commit is contained in:
anthdm 2018-01-26 21:42:43 +01:00
parent 95a7ac1d22
commit 058459c65d
3 changed files with 12 additions and 10 deletions

View file

@ -13,13 +13,16 @@ type Peer struct {
send chan *Message
// verack is true if this node has sended it's version.
verack bool
// whether we or him made the initial connection.
initiator bool
}
// NewPeer returns a (TCP) Peer.
func NewPeer(conn net.Conn) *Peer {
func NewPeer(conn net.Conn, init bool) *Peer {
return &Peer{
conn: conn,
send: make(chan *Message),
conn: conn,
send: make(chan *Message),
initiator: init,
}
}
@ -35,6 +38,7 @@ func (p *Peer) writeLoop() {
for {
msg := <-p.send
rpcLogger.Printf("OUT :: %+v", msg)
if err := msg.encode(p.conn); err != nil {
log.Printf("encode error: %s", err)
}