修复socket连接出错

This commit is contained in:
masnn committed 2020-04-09 20:10:38 +08:00
1 parent a2600d5cc4
commit f259106aa3
1 file changed
+5 -1
+5 -1
View File
@@ -53,6 +53,7 @@ function POST(route, ...handler) {
function SOCKET(prefix, middlewares, handler) {
const sock = sockjs.createServer({ prefix });
sock.on('connection', async conn => {
try {
conn.cookies = {
get(name) {
return conn.cookies[name];
@@ -64,7 +65,7 @@ function SOCKET(prefix, middlewares, handler) {
conn.request = {
headers: conn.headers
};
let p = conn.url.split('?')[1].split('&');
let p = (conn.url.split('?')[1] || '').split('&');
for (let i in p) p[i] = p[i].split('=');
for (let i in p) conn.params[p[i][0]] = decodeURIComponent(p[i][1]);
await new Promise((resolve, reject) => {
@@ -89,6 +90,9 @@ function SOCKET(prefix, middlewares, handler) {
conn.write(JSON.stringify(data));
};
handler(conn);
} catch (e) {
console.error(e);
}
});
sock.installHandlers(server);
}