-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.js
More file actions
135 lines (113 loc) · 3.62 KB
/
server.js
File metadata and controls
135 lines (113 loc) · 3.62 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
var testDataSize = 500;
var testData = [];
var shift = 2.0 * Math.PI * Math.random();
var scale = 0.5;
var currentDataIndex = 0;
var clients = [];
for (var i = 0; i < testDataSize; i++) {
testData[i] = 0;
}
for (var k = 1; k < 1000; k++) {
var a = 1000.0 * (Math.random() - 0.5) / k;
var b = 1000.0 * (Math.random() - 0.5) / k;
for (var i = 0; i < testDataSize; i++) {
testData[i] += a * Math.sin(shift + scale * 2.0 * Math.PI * k * i / testDataSize);
testData[i] += b * Math.cos(shift + scale * 2.0 * Math.PI * k * i / testDataSize);
}
}
function sendNextPrice() {
const buf = Buffer.allocUnsafe(8);
buf.writeDoubleBE(testData[currentDataIndex], 0);
//Go through all users
database.collection("users").find({}).toArray(function(err, clientList) {
if (err) throw err;
for (var i = 0; i < clientList.length; i++) {
socket.send([buf], clientList[i].port, clientList[i].address, (err) => {
if (err != null) {
console.log("ERROR:", err);
}
});
}
});
currentDataIndex = (currentDataIndex + 1) % testDataSize;
}
function checkClientStatus() {
database.collection("users").deleteMany({lastContact: {$lt: new Date().getTime() - 5000}}, function(err, obj) {
if (err) throw err;
console.log(obj.result.n + " users deleted");
});
}
var mongoClient = require('mongodb').MongoClient;
var dbURL = "mongodb://localhost:27017/mydb";
const dgram = require('dgram');
const socket = dgram.createSocket('udp4');
function initDatabase(database) {
//Delete this once we want data persistence
database.createCollection("users", function(err, res) {
if (err) throw err;
//Delete this once we want data persistence
//If I try dropping a non-existant database, it has a fit
//so the easiest way to fix it is to just create it.
database.collection("users").drop(function(err, delOK) {
if (err) throw err;
database.createCollection("users", function(err, res) {
if (err) throw err;
});
});
});
}
var database;
mongoClient.connect(dbURL, function(err, db) {
if (err) throw err;
database = db;
initDatabase(database);
});
socket.on('error', (err) => {
console.log(`server error:\n${err.stack}`);
socket.close();
});
socket.on('message', (msg, rinfo) => {
var id = msg.readDoubleBE(0);
var key = msg.readDoubleBE(8);
console.log(`server got: ${id} from ${rinfo.address}:${rinfo.port}`);
//console.log("Server got %d from ${rinfo.address}:${rinfo.port}", msg);
var newClientObj = {
address: rinfo.address,
port: rinfo.port,
lastContact: new Date().getTime()
};
database.collection("users").update({address: rinfo.address}, newClientObj, {
callback: function(err, result) {
if (err) throw err;
if (result === null) {
database.collection("users").insertOne(myobj, function(err, res) {
if (err) throw err;
console.log("1 record inserted");
});
}
else {
console.log("ELAPSED TIME: " + (new Date().getTime() - result.lastContact));
result.lastContact = new Date().getTime();
console.log("NEW ELAPSED TIME: " + (new Date().getTime() - result.lastContact));
}
console.log("Found:", result);
},
upsert: true});
/*
//Output collection for debugging
database.collection("users").find({}).toArray(function(err, result) {
if (err) throw err;
console.log("\n\n\n-------------\n\n\n")
console.log(result);
console.log("\n\n\n-------------\n\n\n")
});*/
});
socket.on('listening', () => {
const address = socket.address();
socket.setBroadcast(true);
console.log(`server listening ${address.address}:${address.port}`);
setInterval(sendNextPrice, 100);
setInterval(checkClientStatus, 1000);
});
socket.bind(5149);
// server listening 0.0.0.0:5149