70 lines
2.3 KiB
JavaScript
70 lines
2.3 KiB
JavaScript
/* RateLimit example */
|
|
|
|
const uWS = require('../dist/uws.js');
|
|
const port = 9001;
|
|
|
|
const RateLimit = (limit, interval) => { let now = 0; const last = Symbol(), count = Symbol(); setInterval(() => ++now, interval); return ws => { if (ws[last] != now) { ws[last] = now; ws[count] = 1 } else { return ++ws[count] > limit } } }
|
|
const rateLimit = RateLimit(1, 10000) // limit is: 1 message per 10 seconds.
|
|
const rateLimit2 = RateLimit(10, 2000) // limit per interval (milliseconds)
|
|
|
|
|
|
const app = uWS./*SSL*/App({
|
|
key_file_name: 'misc/key.pem',
|
|
cert_file_name: 'misc/cert.pem',
|
|
passphrase: '1234'
|
|
}).ws('/*', {
|
|
/* Options */
|
|
compression: uWS.SHARED_COMPRESSOR,
|
|
maxPayloadLength: 16 * 1024 * 1024,
|
|
idleTimeout: 10,
|
|
/* Handlers */
|
|
open: (ws) => {
|
|
console.log('A WebSocket connected!');
|
|
},
|
|
message: async (ws, message, isBinary) => {
|
|
|
|
// rateLimit(ws) returns true if over limit
|
|
if (rateLimit(ws)) {
|
|
/*
|
|
Do as you wish, personally, I prefer to inform the client, do Log,
|
|
and close the connection if over limit happened.
|
|
or you can just ignore its message and just return.
|
|
*/
|
|
const dataToSendToClient = {
|
|
action: 'info',
|
|
data: {
|
|
message: {
|
|
english: 'over limit! please slow down.',
|
|
farsi: 'یواشتر'
|
|
}
|
|
}
|
|
}
|
|
await ws.send(JSON.stringify(dataToSendToClient));
|
|
console.warn('rate limit hit!');
|
|
return ws.end();
|
|
}
|
|
|
|
//or maybe different type of message or /endpoint needs different limit
|
|
if (ws.type2 && rateLimit2(ws)) {
|
|
console.warn('type2 is over limit');
|
|
return ws.end();
|
|
}
|
|
|
|
// rest of message code goes here ...
|
|
},
|
|
drain: (ws) => {
|
|
console.log('WebSocket backpressure: ' + ws.getBufferedAmount());
|
|
},
|
|
close: (ws, code, message) => {
|
|
console.log('WebSocket closed');
|
|
}
|
|
}).any('/*', (res, req) => {
|
|
res.end('Nothing to see here!');
|
|
}).listen(port, (token) => {
|
|
if (token) {
|
|
console.log('Listening to port ' + port);
|
|
} else {
|
|
console.log('Failed to listen to port ' + port);
|
|
}
|
|
});
|