classes!
This commit is contained in:
parent
5d1b3f2678
commit
164a972faa
GPG Key ID:
186A8370EFF937CA
18
app.js
18
app.js
@ -1,6 +1,9 @@
|
||||
#!/usr/bin/env nodejs
|
||||
const app = {};
|
||||
module.exports = app;
|
||||
|
||||
const {P2P} = require('./p2p')
|
||||
|
||||
const p2p = require('./peers')
|
||||
|
||||
const args = process.argv.slice(1);
|
||||
|
||||
@ -18,13 +21,10 @@ if(server_port){
|
||||
process.exit(0)
|
||||
}
|
||||
|
||||
p2p.listen(server_port);
|
||||
|
||||
for(let client of clients_list){
|
||||
p2p.addPeer(client);
|
||||
}
|
||||
app.p2p = new P2P({
|
||||
listenPort: server_port,
|
||||
peers: clients_list
|
||||
})
|
||||
|
||||
|
||||
setTimeout(function(){
|
||||
p2p.broadcast({type:'topic', body:"yolo"})
|
||||
}, 10000);
|
||||
app.pubsub = require('./pubsub')
|
||||
38
client.js
38
client.js
@ -1,38 +0,0 @@
|
||||
const net = require('net');
|
||||
|
||||
var connect = function(address, port){
|
||||
if(!port){
|
||||
let parse = address.split(':');
|
||||
address = parse[0];
|
||||
port = parse[1];
|
||||
}
|
||||
|
||||
let socket = new net.Socket().connect(port, address);
|
||||
|
||||
// socket.on('connect', function(){
|
||||
// console.log('client EVENT connect:', arguments);
|
||||
// });
|
||||
|
||||
// socket.on('close', function(){
|
||||
// console.log('client EVENT close:', arguments);
|
||||
// });
|
||||
|
||||
// socket.on('ready', function(){
|
||||
// console.log('client EVENT ready:', arguments);
|
||||
// });
|
||||
|
||||
// socket.on('data', function(data){
|
||||
// console.log('client EVENT data:',data.toString(), socket.remoteAddress);
|
||||
// });
|
||||
|
||||
|
||||
socket.on('error', function(){
|
||||
console.log('client EVENT error:', arguments);
|
||||
});
|
||||
|
||||
|
||||
return socket;
|
||||
|
||||
}
|
||||
|
||||
module.exports = {connect}
|
||||
196
p2p.js
Normal file
196
p2p.js
Normal file
@ -0,0 +1,196 @@
|
||||
const crypto = require("crypto");
|
||||
const net = require('net');
|
||||
|
||||
class P2P {
|
||||
constructor(args){
|
||||
this.wantedPeers = new Set();
|
||||
this.connectedPeers = {};
|
||||
this.peerID = crypto.randomBytes(16).toString("hex");
|
||||
this.connectInterval = this.__peerInterval();
|
||||
this.onDataCallbacks = [];
|
||||
|
||||
|
||||
if(args.listenPort){
|
||||
this.server = this.__listen(args.listenPort);
|
||||
}
|
||||
|
||||
if(args.peers){
|
||||
this.addPeer(args.peers);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
addPeer(peer){
|
||||
if(Array.isArray(peer)){
|
||||
for(let address of peer){
|
||||
this.addPeer(address);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
if(this.wantedPeers.has(peer)) return true;
|
||||
this.wantedPeers.add(peer);
|
||||
}
|
||||
|
||||
removePeer(peer){
|
||||
if(this.wantedPeers.has(peer)){
|
||||
this.wantedPeers.delete(peer);
|
||||
|
||||
for(let peerID in this.connectedPeers){
|
||||
if(this.connectedPeers[peerID].peerConnectAddress !== peer) continue;
|
||||
this.connectedPeers[peerID].end();
|
||||
delete this.connectedPeers[peerID];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
__connectPeer(address, port) {
|
||||
if(!port){
|
||||
let parse = address.split(':');
|
||||
address = parse[0];
|
||||
port = parse[1];
|
||||
}
|
||||
|
||||
let peer = new net.Socket().connect(port, address);
|
||||
let p2p = this;
|
||||
|
||||
peer.on('connect', function(){
|
||||
console.info(`Peer ${address} is now connected.`);
|
||||
peer.peerConnectAddress = `${address}:${port}`
|
||||
peer.write(JSON.stringify({type:"register", id: p2p.peerID}));
|
||||
});
|
||||
|
||||
peer.on('close', function(){
|
||||
console.info(`Client Peer ${address}, ${peer.peerID} droped.`);
|
||||
delete p2p.connectedPeers[peer.peerID];
|
||||
});
|
||||
|
||||
peer.on('data', function(data){
|
||||
p2p.__read(JSON.parse(data.toString()), peer.remoteAddress, peer);
|
||||
});
|
||||
|
||||
peer.on('error', function(error){
|
||||
if(error.syscall === 'connect' && error.code === 'ECONNREFUSED'){
|
||||
console.info(`Peer ${error.address}:${error.port} connection refussed!`);
|
||||
}else{
|
||||
console.warn('client EVENT error:', arguments);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
__peerInterval(interval){
|
||||
|
||||
return setInterval(function(p2p){
|
||||
let connected = Object.keys(p2p.connectedPeers).map(function(peerID){
|
||||
return p2p.connectedPeers[peerID].peerConnectAddress
|
||||
});
|
||||
|
||||
for(let peer of p2p.wantedPeers){
|
||||
if(!connected.includes(peer)){
|
||||
p2p.__connectPeer(peer);
|
||||
}
|
||||
}
|
||||
}, interval || 1000, this);
|
||||
}
|
||||
|
||||
__listen (port){
|
||||
|
||||
let p2p = this;
|
||||
|
||||
let serverSocket = new net.Server(function (clientSocket) {
|
||||
|
||||
console.info(`server ${port}`)
|
||||
|
||||
|
||||
clientSocket.on('error', function(){
|
||||
console.log('server-client EVENT error:', arguments);
|
||||
});
|
||||
|
||||
|
||||
});
|
||||
|
||||
serverSocket.on('connection', function(clientSocket){
|
||||
|
||||
console.log('server EVENT connection from client:', clientSocket.remoteAddress);
|
||||
clientSocket.write(JSON.stringify({type:"register", id: p2p.peerID}));
|
||||
|
||||
clientSocket.on('data', function(data){
|
||||
p2p.__read(JSON.parse(data.toString()), clientSocket.remoteAddress, clientSocket);
|
||||
});
|
||||
|
||||
clientSocket.on('close', function(){
|
||||
console.info(`server Peer ${clientSocket.remoteAddress} - ${clientSocket.peerID} droped.`);
|
||||
delete p2p.connectedPeers[clientSocket.peerID];
|
||||
});
|
||||
|
||||
});
|
||||
|
||||
serverSocket.on('listening', function(){
|
||||
console.log('p2p server listening on', port,)
|
||||
});
|
||||
|
||||
serverSocket.on('error', function(){
|
||||
console.log('server EVENT error:', arguments);
|
||||
});
|
||||
|
||||
serverSocket.listen(Number(port));
|
||||
|
||||
return serverSocket;
|
||||
}
|
||||
|
||||
broadcast(message, exclude){
|
||||
exclude = [...exclude || [], ...[this.peerID]]
|
||||
let sentTo = []
|
||||
|
||||
for(let _peerID in this.connectedPeers){
|
||||
if(exclude.includes(_peerID)) continue;
|
||||
sentTo.push(_peerID);
|
||||
}
|
||||
|
||||
message.sentTo = [...new Set([...message.sentTo || [], ...sentTo, ...[this.peerID]])]
|
||||
message.from = message.from || this.peerID;
|
||||
|
||||
for(let _peerID of sentTo){
|
||||
this.connectedPeers[_peerID].write(JSON.stringify(message));
|
||||
}
|
||||
}
|
||||
|
||||
__read(message, from, socket){
|
||||
if(message.type === "heartbeat"){
|
||||
console.log('heartbeat from', from);
|
||||
return ;
|
||||
}
|
||||
|
||||
if(message.type === "register"){
|
||||
console.log('registering peer', message.id, socket.remoteAddress)
|
||||
|
||||
if(Object.keys(this.connectedPeers).includes(message.id)){
|
||||
console.log(`Dropping ${message.id}, already connected`)
|
||||
socket.end();
|
||||
}else{
|
||||
socket.peerID = message.id
|
||||
this.connectedPeers[message.id] = socket;
|
||||
}
|
||||
return ;
|
||||
}
|
||||
|
||||
// forward the message to other peers
|
||||
this.broadcast(message, message.sentTo);
|
||||
// console.log('p2p read:', message)
|
||||
|
||||
// pass message to local callbacks
|
||||
this.onDataCallbacks.forEach(function(callback){
|
||||
callback(message);
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
onData(callback){
|
||||
if(callback instanceof Function){
|
||||
this.onDataCallbacks.push(callback);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
module.exports = {P2P};
|
||||
126
peers.js
126
peers.js
@ -1,126 +0,0 @@
|
||||
const TcpServer = require('./server');
|
||||
const TcpClient = require('./client');
|
||||
const crypto = require("crypto");
|
||||
|
||||
const peerID = crypto.randomBytes(16).toString("hex");
|
||||
|
||||
console.log('Peer ID:', peerID)
|
||||
|
||||
let connectedPeers = {}
|
||||
let peers = []
|
||||
|
||||
let addPeer = function(peer){
|
||||
if(peers.includes(peer)) return true;
|
||||
peers.push(peer)
|
||||
}
|
||||
|
||||
setInterval(function(){
|
||||
let connected = Object.keys(connectedPeers).map(function(i){return connectedPeers[i].peerConnectAddress})
|
||||
console.log('interval', connected, Object.keys(connectedPeers))
|
||||
for(let peer of peers){
|
||||
if(!connected.includes(peer)){
|
||||
connectPeer(peer);
|
||||
}
|
||||
}
|
||||
}, 1000);
|
||||
|
||||
|
||||
let connectPeer = function(address, port) {
|
||||
if(!port){
|
||||
let parse = address.split(':');
|
||||
address = parse[0];
|
||||
port = parse[1];
|
||||
}
|
||||
|
||||
let peer = TcpClient.connect(address, port);
|
||||
|
||||
peer.on('connect', function(){
|
||||
console.info(`Peer ${address} is now connected.`);
|
||||
peer.peerConnectAddress = `${address}:${port}`
|
||||
peer.write(JSON.stringify({type:"register", id: peerID}));
|
||||
});
|
||||
|
||||
peer.on('close', function(){
|
||||
delete connectedPeers[peer.peerID];
|
||||
console.info(`Peer ${address} droped.`);
|
||||
});
|
||||
|
||||
peer.on('data', function(data){
|
||||
read(JSON.parse(data.toString()), peer.remoteAddress, peer);
|
||||
});
|
||||
|
||||
}
|
||||
|
||||
let removePeer = function(peer){
|
||||
if(peer[peer.address]){
|
||||
peer.end();
|
||||
delete peer[peer.address]
|
||||
}
|
||||
}
|
||||
|
||||
let listen = function(port){
|
||||
console.log('p2p listen', port)
|
||||
let serverSocket = TcpServer(port);
|
||||
|
||||
serverSocket.on('connection', function(clientSocket){
|
||||
console.log('server EVENT connection from client:', clientSocket.remoteAddress);
|
||||
clientSocket.write(JSON.stringify({type:"register", id: peerID}));
|
||||
|
||||
clientSocket.on('data', function(data){
|
||||
read(JSON.parse(data.toString()), clientSocket.remoteAddress, clientSocket);
|
||||
});
|
||||
|
||||
clientSocket.on('close', function(){
|
||||
delete connectedPeers[clientSocket.peerID];
|
||||
console.info(`Peer ${clientSocket.remoteAddress} droped.`);
|
||||
});
|
||||
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
let broadcast = function(message, exclude){
|
||||
exclude = [...exclude || [], ...[peerID]]
|
||||
let sentTo = []
|
||||
|
||||
for(let _peerID in connectedPeers){
|
||||
if(exclude.includes(_peerID)) continue;
|
||||
sentTo.push(_peerID);
|
||||
}
|
||||
|
||||
message.sentTo = [...new Set([...message.sentTo || [] ,...sentTo, ...[peerID]])]
|
||||
message.from = message.from || peerID;
|
||||
|
||||
for(let _peerID of sentTo){
|
||||
connectedPeers[_peerID].write(JSON.stringify(message));
|
||||
}
|
||||
}
|
||||
|
||||
let read = function(message, from, socket){
|
||||
if(message.type === "heartbeat"){
|
||||
console.log('heartbeat from', from);
|
||||
return ;
|
||||
}
|
||||
|
||||
if(message.type === "register"){
|
||||
console.log('registering net peer', message.id, socket.remoteAddress)
|
||||
if(Object.keys(connectedPeers).includes(message.id)){
|
||||
console.log(`Dropping ${message.id}, already connected`)
|
||||
socket.end();
|
||||
}else{
|
||||
connectedPeers[message.id] = socket;
|
||||
connectedPeers[message.id].peerID = message.id
|
||||
}
|
||||
}
|
||||
|
||||
if(message.type === 'topic'){
|
||||
broadcast(message, message.sentTo);
|
||||
console.log('p2p read:', message)
|
||||
|
||||
// publish message locally
|
||||
//publish(message.top, message.body)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
module.exports = {broadcast, connectPeer, listen, addPeer}
|
||||
11
pubsub.js
Normal file
11
pubsub.js
Normal file
@ -0,0 +1,11 @@
|
||||
const app = require('./app');
|
||||
|
||||
app.p2p.onData(function(data){
|
||||
console.log('app, data:', data)
|
||||
})
|
||||
|
||||
|
||||
setTimeout(function(){
|
||||
app.p2p.broadcast({type:'topic', body:"yolo"})
|
||||
}, 10000);
|
||||
|
||||
51
server.js
51
server.js
@ -1,51 +0,0 @@
|
||||
const net = require('net')
|
||||
|
||||
const server = function(port){
|
||||
|
||||
let serverSocket = new net.Server(function (clientSocket) {
|
||||
|
||||
console.info(`server ${port}`)
|
||||
|
||||
// clientSocket.on('connection', function(){
|
||||
// console.log('server-client EVENT connection:', arguments);
|
||||
// });
|
||||
|
||||
// clientSocket.on('listening', function(){
|
||||
// console.log('server-client EVENT listening:', arguments);
|
||||
// });
|
||||
|
||||
clientSocket.on('error', function(){
|
||||
console.log('server-client EVENT error:', arguments);
|
||||
});
|
||||
|
||||
// clientSocket.on('data', function (data) {
|
||||
// console.log('server-client EVENT data', data.toString(), clientSocket.remoteAddress);
|
||||
// });
|
||||
|
||||
})
|
||||
serverSocket.listen(Number(port));
|
||||
|
||||
// serverSocket.on('connection', function(clientSocket){
|
||||
// console.log('server EVENT connection:', clientSocket.remoteAddress);
|
||||
// });
|
||||
|
||||
// serverSocket.on('listening', function(){
|
||||
// console.log('server EVENT listening:', arguments);
|
||||
// });
|
||||
|
||||
serverSocket.on('error', function(){
|
||||
console.log('server EVENT error:', arguments);
|
||||
});
|
||||
|
||||
// serverSocket.on('data', function (data) {
|
||||
// console.info('here...')
|
||||
// // console.log(arguments, serverSocket);
|
||||
// });
|
||||
|
||||
|
||||
return serverSocket;
|
||||
|
||||
}
|
||||
|
||||
|
||||
module.exports = server;
|
||||
Loading…
x
Reference in New Issue
Block a user