use the messages object associated with the peer

Author: xloem

lib/pool.js

@@ -1,7 +1,6 @@
 'use strict';
 var P2P = require('bitcore-p2p');
 var Peer = P2P.Peer;
-var Messages = new P2P.Messages();
 var bitcore = require('bitcore-lib');
 var BlockHeader = bitcore.BlockHeader;
 var EventEmitter = require('events').EventEmitter;
@@ -93,7 +92,7 @@ Pool.prototype._setLoaderPeer = function(peer) {
   this.peers.loader = peer;
   var lastHashIdx = this.chain.index.hashes.length - 1;
   // TODO: move this to an on('set-loader-peer') instead of being in here
-  peer.sendMessage(new Messages.GetHeaders([this.chain.index.hashes[lastHashIdx]]));
+  peer.sendMessage(new peer.messages.GetHeaders([this.chain.index.hashes[lastHashIdx]]));
 }
 
 Pool.prototype._handlePeerConnect = function(peer) {
@@ -116,7 +115,7 @@ Pool.prototype._handlePeerReady = function(peer, addr) {
   // TODO: not ideal to stick this on the peer object
   // But its the simplest for now
   peer._pingTimer = (setInterval(function(){
-    peer.sendMessage(new Messages.Ping());
+    peer.sendMessage(peer.messages.Ping());
   },this.ping.interval));
 
   //TODO: Smarter loader peer choosing
@@ -162,7 +161,7 @@ Pool.prototype._handlePeerInv = function(peer, message) {
   // If we got new blocks, GetHeaders
   if(blockHashes.length) {
     var lastHashIdx = this.chain.index.hashes.length - 1;
-    peer.sendMessage(new Messages.GetHeaders([this.chain.index.hashes[lastHashIdx]]));
+    peer.sendMessage(new peer.messages.GetHeaders([this.chain.index.hashes[lastHashIdx]]));
   }
 }
 
@@ -177,7 +176,7 @@ Pool.prototype._handlePeerHeaders = function(peer, message) {
   // If we got 2000 messages, assume we still have more to get
   if(message.headers.length == 2000) {
     var lastHeader = message.headers[message.headers.length - 1];
-    peer.sendMessage(new Messages.GetHeaders([lastHeader.id]));
+    peer.sendMessage(new peer.messages.GetHeaders([lastHeader.id]));
   } else {
     this.emit('chain-full');
   }