websocket connect
parent
e2733bfee7
commit
6f1cc69031
@ -0,0 +1,55 @@
|
||||
// const callService = require('../../core/services/call');
|
||||
// const websocketManager = require('../../core/websocket/manager');
|
||||
const { callService, websocketManager } = require('../../core'); // Import from core/index.js
|
||||
|
||||
exports.getIn = async ctx => {
|
||||
// Wait for at least one connection to be established (or handle the case where no connections are available)
|
||||
// await new Promise(resolve => {
|
||||
// const checkConnections = () => {
|
||||
// if (websocketManager.getConnections().length > 0) {
|
||||
// resolve();
|
||||
// } else {
|
||||
// setTimeout(checkConnections, 100); // Check again after 100ms
|
||||
// }
|
||||
// };
|
||||
// checkConnections();
|
||||
// });
|
||||
|
||||
const availableWs = websocketManager.getAvailableConnection();
|
||||
|
||||
if (!availableWs) {
|
||||
// ctx.status = 503;
|
||||
// ctx.body = { message: 'No available connections' };
|
||||
ctx.assert(availableWs, 503, 'No available connections');
|
||||
return;
|
||||
}
|
||||
// return availableWs;
|
||||
|
||||
const { sessionId, url } = callService.createSession(availableWs);
|
||||
return { sessionId, url, message: 'Connection established' }; // availableWs
|
||||
};
|
||||
|
||||
exports.sendMsg = ctx => {
|
||||
const { sessionId, message } = ctx.request.body;
|
||||
if (!sessionId || !message) {
|
||||
// ctx.status = 400;
|
||||
// ctx.body = { message: 'Session ID and message are required' };
|
||||
ctx.assert(sessionId, 400, 'Session ID and message are required');
|
||||
return;
|
||||
}
|
||||
const wsToSend = callService.getSession(sessionId);
|
||||
if (!wsToSend) {
|
||||
// ctx.status = 404;
|
||||
// ctx.body = { message: 'Session not found' };
|
||||
ctx.assert(wsToSend, 400, 'Session not found');
|
||||
return;
|
||||
}
|
||||
try {
|
||||
wsToSend.send(message);
|
||||
ctx.body = { message: 'Message sent' };
|
||||
} catch (error) {
|
||||
console.error('Error sending message:', error);
|
||||
ctx.status = 500;
|
||||
ctx.body = { message: 'Failed to send message' };
|
||||
}
|
||||
};
|
||||
@ -0,0 +1 @@
|
||||
module.exports = require('./message.routes');
|
||||
@ -0,0 +1,54 @@
|
||||
'use strict';
|
||||
|
||||
const generateId = require('../../utils/generateId.util');
|
||||
|
||||
/**
|
||||
* Mock database, replace this with your db models import, required to perform query to your database.
|
||||
*/
|
||||
const db = {
|
||||
users: [
|
||||
{
|
||||
id: 'bff28903-042e-47c2-b9ee-07c3954989ec',
|
||||
name: 'Marco',
|
||||
created_at: 1558536830937,
|
||||
},
|
||||
{
|
||||
id: 'dca01a32-36e6-4886-af75-8e7caa0162a9',
|
||||
name: 'Leonardo',
|
||||
created_at: 1558536843742,
|
||||
},
|
||||
{
|
||||
id: 'dca01a32-36e6-4886-af75-8e7caa0162a9',
|
||||
name: 'Berta',
|
||||
created_at: 1558536863550,
|
||||
},
|
||||
],
|
||||
};
|
||||
function sleep(ms) {
|
||||
return new Promise(resolve => setTimeout(resolve, ms));
|
||||
}
|
||||
exports.getOne = async ctx => {
|
||||
const { id } = ctx.params;
|
||||
const user = db.users.find(user => user.id === id);
|
||||
await sleep(500);
|
||||
ctx.assert(user, 404, "The requested user doesn't exist");
|
||||
return user;
|
||||
};
|
||||
|
||||
exports.getAll = async ctx => {
|
||||
return db.users;
|
||||
};
|
||||
|
||||
exports.createOne = async ctx => {
|
||||
const { name } = ctx.request.body;
|
||||
ctx.assert(name, 400, 'The user info is malformed!');
|
||||
const id = generateId();
|
||||
const newUser = {
|
||||
id,
|
||||
name,
|
||||
timestamp: Date.now(),
|
||||
};
|
||||
db.users.push(newUser);
|
||||
const createdUser = db.users.find(user => user.id === id);
|
||||
return createdUser;
|
||||
};
|
||||
@ -0,0 +1,16 @@
|
||||
'use strict';
|
||||
|
||||
const controller = require('./message.controller');
|
||||
|
||||
module.exports = Router => {
|
||||
const router = new Router({
|
||||
prefix: `/messages`,
|
||||
});
|
||||
|
||||
router
|
||||
.get('/:id', controller.getOne)
|
||||
.get('/', controller.getAll)
|
||||
.post('/', controller.createOne);
|
||||
|
||||
return router;
|
||||
};
|
||||
@ -1 +0,0 @@
|
||||
module.exports = require('./whatsapp.routes');
|
||||
@ -1,54 +0,0 @@
|
||||
'use strict';
|
||||
|
||||
const generateId = require('../../utils/generateId.util');
|
||||
|
||||
/**
|
||||
* Mock database, replace this with your db models import, required to perform query to your database.
|
||||
*/
|
||||
const db = {
|
||||
users: [
|
||||
{
|
||||
id: 'bff28903-042e-47c2-b9ee-07c3954989ec',
|
||||
name: 'Marco',
|
||||
created_at: 1558536830937,
|
||||
},
|
||||
{
|
||||
id: 'dca01a32-36e6-4886-af75-8e7caa0162a9',
|
||||
name: 'Leonardo',
|
||||
created_at: 1558536843742,
|
||||
},
|
||||
{
|
||||
id: 'dca01a32-36e6-4886-af75-8e7caa0162a9',
|
||||
name: 'Berta',
|
||||
created_at: 1558536863550,
|
||||
},
|
||||
],
|
||||
};
|
||||
function sleep(ms) {
|
||||
return new Promise(resolve => setTimeout(resolve, ms));
|
||||
}
|
||||
exports.getOne = async ctx => {
|
||||
const { userId } = ctx.params;
|
||||
const user = db.users.find(user => user.id === userId);
|
||||
await sleep(500);
|
||||
ctx.assert(user, 404, "The requested user doesn't exist");
|
||||
return user;
|
||||
};
|
||||
|
||||
exports.qrcode = async ctx => {
|
||||
return `data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAMgAAADICAIAAAAiOjnJAAAAAXNSR0IArs4c6QAADgtJREFUeAHt2tFuXTcSRNH8/09nHnwdJLFX5C13D8+FmJeMi3V2FZstDWDkjz/e5J8/h/7RdYWXX7o40itH/qt/cgJ6mKorXhz5pYsjvXLkv/onJ6CHqbrixZFfujjSK0f+q39yAnqYqiteHPmliyO9cuS/+icnoIepuuLFkV+6ONIrR/6rf3ICepiqK14c+aWLI71y5L/6Jyegh6m64sWRX7o40itH/qt/cgJ6mKorXhz5pYsjvXLkfxtdg9jWNSDlbvuVW3X1lC5+9YszpasP9angylEhcbb9yq26ekoXv/rFmdLVh/pUcOWokDjbfuVWXT2li1/94kzp6kN9KrhyVEicbb9yq66e0sWvfnGmdPWhPhVcOSokzrZfuVVXT+niV784U7r6UJ8KrhwVEmfbr9yqq6d08atfnCldfahPBVeOComz7Vdu1dVTuvjVL86Urj7Up4IrR4XE2fYrt+rqKV386hdnSlcf6grmB/Gg8uXf1nWt7Vzxax/5q177kD8GQkLly7+to/4f27ni1z7yV732IX8MhITKl39bR/27WK/Baz7U9WD8IB5Uvvzbuq61nSt+7SN/1Wsf8sdASKh8+bd11L+/sV6D13yo68H4QTyofPm3dV1rO1f82kf+qtc+5I+BkFD58m/rqH9/Y70Gr/lQ14Pxg3hQ+fJv67rWdq74tY/8Va99yB8DIaHyqx+xlLf5Clau9MqRv+q1D/ljICRUfvUjlvI2X8HKlV458le99iF/DISEyq9+xFLe5itYudIrR/6q1z7kj4GQUPnVj1jK23wFK1d65chf9dqH/DEQEiq/+hFLeZuvYOVKrxz5q177kD8GQkLlVz9iKW/zFaxc6ZUjf9VrH/LHQEio/OpHLOVtvoKVK71y5K967UP+GAgJlV/9iKW8zVewcqVXjvxVr33IHwMhofLll47Y9b8xr7mn/MqVXucsDh+AH8SDWlR+6aoj/5Rec0/5lStd85Gf+hgICZUvv3TE8gdGnKrX3FN+5UrXHOSnPgZCQuXLLx2xd7E0mA/0OmfixkBIqHz5pSP2LpYG84Fe50zcGAgJlS+/dMTexdJgPtDrnIkbAyGh8uWXjti7WBrMB3qdM3FjICRUvvzSEXsXS4P5QK9zJk6gbV2FlCu/9KdxpnrqXtu6+lPfLiS+ClX/u3Cmemo+27r6U98uJL4KVf+7cKZ6aj7buvpT3y4kvgpV/7twpnpqPtu6+lPfLiS+ClX/u3Cmemo+27r6U98uJL4KVf+7cKZ6aj7buvpT3y4kvgpV/7twpnpqPtu6+lPfLiS+ClX/u3Cmemo+27r6v42uAekCX82vOVz9gwl8tUWp9/1gfPdYE6iD/mp+ze3qH0zgqy1Kve8H47vHmkAd9Ffza25X/2ACX21R6n0/GN891gTqoL+aX3O7+gcT+GqLUu/7wfj+/8e6wLvrU5OcmoP6TPGfxln/LypPXVgPWfWp/sqd4j+NcxdLL/7Spx5MMVP8p3HuYunF72L91q7exbqL9VsLpI/vYt3F0m78ln4X6y7Wby2QPr6LdRdLu/Fb+gdj3TtW673Eb+SaO+WvHM2hcp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7m173G9HrhqeCaO+WvHN23cp7mH/uLUw1Ieh3EFGc7t/Lll17nII70ypf/LtZrwhwQDvQw0oHJ85/iTPVUn3yxWkjBpzjbuZUvv/Q6T3GkV778d7FeE+aAcKCHkQ5Mnv8UZ6qn+uSL1UIKPsXZzq18+aXXeYojvfLlv4v1mjAHhAM9jHRg8vynOFM91SdfrBZS8CnOdm7lyy+9zlMc6ZUv/12s14Q5IBzoYaQDk+c/xZnqyQuoqPRaSJyqK1e6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/vcqQtMcerDKFe6+PJLF2dKf/tcXaDq2wOtfab8ulflizOl1z5TfvZfD2Dyzw+m+kxxft7yD/7FsnLFmdKVu62z/1QwA+LBVJ8pjupXvjhTeu0z5Wf/9QAm//xgqs8U5+ct72+s7/PVfPKv9O/Af/+bAfHg39zTf1b92kucKb32mfKz/3oAk39+MNVnivPzlvc31vf5aj73N9b3CeHfGhzslMWZ0hm8fMD+U7kMiAdTfaY4ql/54kzptc+UP/dXsEDyV138bV09p3LFlz6VK86pXP5fZC2qC0gXf1vf7iO+9He/L/vXC8tfdRZaPlDPqVjxpU/linMq9/7Gek1eD1N1PaT0yq/+U7l3se5i1V39NX/daPmr/mvt5l3qOZUkvvSpXHFO5d7fWK/J62GqroeUXvnVfyr3LtZdrLqrO379BEhXi+qvHPFP6VP9K+eUX7nU68MIJI780sV5mj7Vv3JO+ZVLvT6YQOLIL12cp+lT/SvnlF+51OuDCSSO/NLFeZo+1b9yTvmVS70+mEDiyC9dnKfpU/0r55RfudTrgwkkjvzSxXmaPtW/ck75lUu9PphA4sgvXZyn6VP9K+eUX7nU64MJJI780sV5mj7Vv3JO+fkXpPVhTl1APdVnSq+52/56r9qn8u9i5Yl9+6A+zLa/XqP2qfy7WHlid7F+aWTa3KorTJxTfuVW/d3vVfvX+dzfWHli3z6oD7Ptr9eofSr/Llae2F2sXxqZNrfqChPnlF+5VX/3e9X+dT73N1ae2LcP6sNs++s1ap/K52JVUC1a/eojjvRTnJorf9U1h6rX3LtYr4lNDVocPUz1iyNd/KqLT10B/AAHlVP9iOUPRuXLL732mfKLI139qy4+dQXwAxxUTvUj9i6WBvPSNeeqfxDz47ECfnT+t1I51a90caSf4tRc+auuOVS95vInvoJUVJzqr5zKl1967TPlF0e6+lddfOoK4Ac4qJzqRyx/MCpffum1z5RfHOnqX3XxqSuAH+CgcqofsXexNJiXrjlX/YOYH4/XA36M/E/lVB/l/mfZnxxOcX6C/pRU+8gvnaX0gXSChg6UK30olr/5Kn+753Yf9ZfOPvpAOkFDB8qVPhR7F+s1SM1ZOuevD6QTNHSgXOlDsXexXoPUnKVz/vpAOkFDB8qVPhR7F+s1SM1ZOuevD6QTNHSgXOlDsXexXoPUnKVz/vpAOkFDB8qVPhR7F+s1SM1ZOuevD6QTNHSgXOlDsXexXoPUnKVzcPrgaXpdoNp/ij/FOdU/59YPnubffrAp/hSnzv9Ybi36NP/24Kb4U5w6/2O5tejT/NuDm+JPcer8j+XWok/zbw9uij/FqfM/lluLPs2/Pbgp/hSnzv9Ybi36NP/24Kb4U5w6/2O5tejT/NuDm+JPcer83yW39hzza6AKkH9KV6505b6LXz3fXp96GHGqXgcqvjhP86vn2+tTgxan6nWg4ovzNL96vr0+NWhxql4HKr44T/Or59vrU4MWp+p1oOKL8zS/er69PjVocapeByq+OE/zq+fb61ODFqfqdaDii/M0v3q+vT41aHGqXgcqvjhP86tn1nWxbV1Fa6440sWv/sqZ4m/n1p7yH/sPAFVIg5MujvTKkV+6cqWLU3XxpYs/5b+L9ZpwHWh9mCn+dm7tKf9drLtY/9hVLco/TH/7g/x3se5i/W1N/vxTi/IP09/+IP9drLtYf1uTu1h/DYM/MTj468N//Q/Y8w+eONL/VePTfxRfuoKm/HlwKlT1egHxxZFeOfJLV650caouvnTxp/xcLAVUffsC23zdt+aKs60f67kdXPlP8+vha09xtvVjPbeDK/9pfj187SnOtn6s53Zw5T/Nr4evPcXZ1o/13A6u/Kf59fC1pzjb+rGe28GV/zS/Hr72FGdbP9ZzO7jyn+bXw9ee4mzrx3puB1f+0/x6+NpTnG39WM/t4Mo/5Z/KrZy6WNv82of+7aKVf8o/lVs5fBgcbPMR2+XtopV/yj+VWzn1xbb5tQ/920Ur/5R/Krdy+DA42OYjtsvbRSv/lH8qt3Lqi23zax/6t4tW/in/VG7l8GFwsM1HbJe3i1b+Kf9UbuXUF9vm1z70bxet/FP+qdzK4cPgYJuP2C5vF638bX+f0MwXute2XturT+V8uf/QLw9o6AM92LZe66tP5dzFyhP73Ad6sG29tlWfyrmLlSf2uQ/0YNt6bas+lXMXK0/scx/owbb12lZ9KucuVp7Y5z7Qg23rta36VM5drDyxz32gB9vWa1v1qZy7WHlin/tAD7at17bqUzlcLAVM6SoqvvxTunKlK7f6xdnW1VN67iPQtq6iypV/SleudOVWvzjbunpKz30E2tZVVLnyT+nKla7c6hdnW1dP6bmPQNu6iipX/ildudKVW/3ibOvqKT33EWhbV1Hlyj+lK1e6cqtfnG1dPaXnPgJt6yqqXPmndOVKV271i7Otq6f03EegbV1FlSv/lK5c6cqtfnG2dfWUnvsItK2rqHLln9KVK1251S/Otq6e0rf7rPOfdjH1mdLXB3oDvk1AD3ZqPuozpZ+615fL1YOdGoT6TOmn7vXlcvVgpwahPlP6qXt9uVw92KlBqM+UfupeXy5XD3ZqEOozpZ+615fL1YOdGoT6TOmn7vXlcvVgpwahPlP6qXtN5f4PQtV4Huhvn8YAAAAASUVORK5CYII=`;
|
||||
};
|
||||
|
||||
exports.createOne = async ctx => {
|
||||
const { name } = ctx.request.body;
|
||||
ctx.assert(name, 400, 'The user info is malformed!');
|
||||
const id = generateId();
|
||||
const newUser = {
|
||||
id,
|
||||
name,
|
||||
timestamp: Date.now(),
|
||||
};
|
||||
db.users.push(newUser);
|
||||
const createdUser = db.users.find(user => user.id === id);
|
||||
return createdUser;
|
||||
};
|
||||
@ -1,16 +0,0 @@
|
||||
'use strict';
|
||||
|
||||
const controller = require('./whatsapp.controller');
|
||||
|
||||
module.exports = Router => {
|
||||
const router = new Router({
|
||||
prefix: `/whatsapp`,
|
||||
});
|
||||
|
||||
router
|
||||
.get('/qrcode', controller.qrcode)
|
||||
.get('/qrcode/:l', controller.qrcode)
|
||||
.post('/', controller.createOne);
|
||||
|
||||
return router;
|
||||
};
|
||||
@ -0,0 +1,26 @@
|
||||
'use strict';
|
||||
|
||||
const serverNodes = Array.from({ length: 1 }, (_, i) => ({
|
||||
name: `wai-server-${i + 1}`,
|
||||
port: 8080 + i,
|
||||
// host: `ws://remote-websocket-service-${i + 1}.com`,
|
||||
host: `wss://p9axztuwd7x8a7.mycht.cn/whatsapp_server?opisn=404&_spam=${Date.now().toString()}`,
|
||||
protocol: 'WhatsApp',
|
||||
// path: '/ws',
|
||||
// protocol: 'ws',
|
||||
// subprotocol: 'protoo',
|
||||
// maxPayload: 1048576, // 1MB
|
||||
// allowRequest: (req, next) => next(null, true),
|
||||
// compression: true,
|
||||
// json: true,
|
||||
// binary: true,
|
||||
// heartbeatInterval: 5000,
|
||||
// heartbeatTimeout: 10000,
|
||||
// closeTimeout: 5000,
|
||||
}));
|
||||
|
||||
const config = {
|
||||
websockets: serverNodes,
|
||||
};
|
||||
|
||||
module.exports = config;
|
||||
@ -0,0 +1,12 @@
|
||||
// core/index.js
|
||||
const createCallService = require('./services/call');
|
||||
const createWebsocketManager = require('./websocket/manager');
|
||||
|
||||
// Create the instances here
|
||||
const callService = createCallService();
|
||||
const websocketManager = createWebsocketManager(callService);
|
||||
|
||||
module.exports = {
|
||||
callService, // Export the INSTANCE
|
||||
websocketManager, // Export the INSTANCE
|
||||
};
|
||||
@ -0,0 +1,31 @@
|
||||
const generateId = require('../../utils/generateId.util');
|
||||
|
||||
module.exports = () => {
|
||||
const sessions = new Map();
|
||||
|
||||
const createSession = ws => {
|
||||
const sessionId = generateId();
|
||||
sessions.set(ws, sessionId);
|
||||
return { sessionId, url: ws.url };
|
||||
};
|
||||
|
||||
const getSession = sessionId => {
|
||||
for (const [ws, storedSessionId] of sessions) {
|
||||
if (storedSessionId === sessionId) {
|
||||
return ws;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
};
|
||||
|
||||
const removeSession = ws => {
|
||||
sessions.delete(ws);
|
||||
};
|
||||
|
||||
return {
|
||||
createSession,
|
||||
getSession,
|
||||
removeSession,
|
||||
sessions,
|
||||
};
|
||||
};
|
||||
@ -0,0 +1,60 @@
|
||||
const WebSocket = require('ws');
|
||||
const websocketServerNodes = require('../../config').websockets;
|
||||
|
||||
module.exports = callService => {
|
||||
let connections = [];
|
||||
|
||||
const connect = () => {
|
||||
websocketServerNodes.forEach((node, index) => {
|
||||
createConnection(node, index);
|
||||
});
|
||||
};
|
||||
|
||||
const createConnection = (node, index) => {
|
||||
try {
|
||||
const ws = new WebSocket(node.host, [node.protocol]);
|
||||
ws.on('open', () => {
|
||||
console.log(`Connected to WebSocket ${index}: ${node.protocol} ${node.host} ${node.protocol}`);
|
||||
// connections.push(ws);
|
||||
connections = [...connections, ws];
|
||||
console.log('Current connections:', connections.length);
|
||||
});
|
||||
|
||||
ws.on('close', () => {
|
||||
console.log(`WebSocket ${index} disconnected. Reconnecting...`);
|
||||
connections.splice(connections.indexOf(ws), 1);
|
||||
setTimeout(() => createConnection(node, index), 3000);
|
||||
});
|
||||
|
||||
ws.on('error', error => {
|
||||
console.error(`WebSocket ${index} error:`, error);
|
||||
});
|
||||
|
||||
ws.on('message', message => {
|
||||
console.log(`Received from WebSocket ${index}: ${message}`);
|
||||
});
|
||||
} catch (error) {
|
||||
console.error(`Error connecting to WebSocket ${index}:`, error);
|
||||
}
|
||||
};
|
||||
|
||||
const getAvailableConnection = () => {
|
||||
// console.log('Available connections length:', connections.length);
|
||||
for (const ws of connections) {
|
||||
// console.log('Session has ws:', callService.sessions.has(ws));
|
||||
if (!callService.sessions.has(ws)) {
|
||||
// console.log('Found available ws:', ws); // Add this line
|
||||
return ws;
|
||||
}
|
||||
}
|
||||
console.log('No available connections found.');
|
||||
return null;
|
||||
};
|
||||
const getConnections = () => connections; // Add this getter
|
||||
|
||||
return {
|
||||
connect,
|
||||
getAvailableConnection,
|
||||
getConnections,
|
||||
};
|
||||
};
|
||||
Loading…
Reference in New Issue