Use VecWriter for enqueue_message method.

Author: arik-so

lightning/src/ln/peer_handler.rs

@@ -461,8 +461,9 @@ impl<Descriptor: SocketDescriptor, CM: Deref, L: Deref> PeerManager<Descriptor,
 
 	/// Append a message to a peer's pending outbound/write buffer, and update the map of peers needing sends accordingly
 	fn enqueue_message<M: Encode + Writeable>(&self, peers_needing_send: &mut HashSet<Descriptor>, peer: &mut Peer, descriptor: Descriptor, message: &M) {
-		let mut encoded_message: Vec<u8> = vec![];
-		wire::write(message, &mut encoded_message).unwrap(); // crash if the write failed
+		let mut buffer = VecWriter(Vec::new());
+		wire::write(message, &mut buffer).unwrap(); // crash if the write failed
+		let encoded_message = buffer.0;
 
 		log_trace!(self.logger, "Enqueueing message of type {} to {}", message.type_id(), log_pubkey!(peer.their_node_id.unwrap()));
 		peer.pending_outbound_buffer.push_back(peer.channel_encryptor.encrypt_message(&encoded_message[..]));