94 lines
No EOL
2.6 KiB
JavaScript
94 lines
No EOL
2.6 KiB
JavaScript
const _io = require("socket.io");
|
|
const {getCompressionConfig} = require("./helper");
|
|
const {logger} = require('./logger');
|
|
|
|
let io;
|
|
|
|
const getServer = function () {
|
|
return io;
|
|
}
|
|
|
|
let redisClient;
|
|
const useRedis = process.env.redis === "true";
|
|
|
|
if (useRedis) {
|
|
const {createClient} = require("redis");
|
|
const REDIS_URL = (process.env.REDIS_URL || "localhost:6379").replace(/((^\w+:|^)\/\/|^)/, 'redis://');
|
|
redisClient = createClient({url: REDIS_URL});
|
|
redisClient.on("error", (error) => logger.error(`Redis error : ${error}`));
|
|
void redisClient.connect();
|
|
}
|
|
|
|
const processSocketsList = function (sockets) {
|
|
let res = []
|
|
for (let socket of sockets) {
|
|
let {handshake} = socket;
|
|
res.push({handshake});
|
|
}
|
|
return res
|
|
}
|
|
|
|
const doFetchAllSockets = async function () {
|
|
if (useRedis) {
|
|
try {
|
|
let cachedResult = await redisClient.get('fetchSocketsResult');
|
|
if (cachedResult) {
|
|
return JSON.parse(cachedResult);
|
|
}
|
|
let result = await io.fetchSockets();
|
|
let cachedString = JSON.stringify(processSocketsList(result));
|
|
await redisClient.set('fetchSocketsResult', cachedString, {EX: 5});
|
|
return result;
|
|
} catch (error) {
|
|
logger.error('Error setting value with expiration:', error);
|
|
}
|
|
}
|
|
return await io.fetchSockets();
|
|
}
|
|
|
|
const fetchSockets = async function (roomID) {
|
|
if (!io) {
|
|
return [];
|
|
}
|
|
if (!roomID) {
|
|
return await doFetchAllSockets();
|
|
}
|
|
return await io.in(roomID).fetchSockets();
|
|
}
|
|
|
|
const createSocketIOServer = function (server, prefix) {
|
|
if (io) {
|
|
return io;
|
|
}
|
|
if (process.env.uws !== "true") {
|
|
io = _io(server, {
|
|
maxHttpBufferSize: (parseFloat(process.env.maxHttpBufferSize) || 5) * 1e6,
|
|
cors: {
|
|
origin: "*",
|
|
methods: ["GET", "POST", "PUT"],
|
|
credentials: true
|
|
},
|
|
path: (prefix ? prefix : '') + '/socket',
|
|
...getCompressionConfig()
|
|
});
|
|
} else {
|
|
io = new _io.Server({
|
|
maxHttpBufferSize: (parseFloat(process.env.maxHttpBufferSize) || 5) * 1e6,
|
|
cors: {
|
|
origin: "*",
|
|
methods: ["GET", "POST", "PUT"],
|
|
credentials: true
|
|
},
|
|
path: (prefix ? prefix : '') + '/socket',
|
|
...getCompressionConfig()
|
|
});
|
|
io.attachApp(server);
|
|
}
|
|
return io;
|
|
}
|
|
|
|
module.exports = {
|
|
createSocketIOServer,
|
|
getServer,
|
|
fetchSockets,
|
|
} |