Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

core, ui, judge: rewrite http & ws layer (#352)

Browse Source
This commit is contained in:
undefined authored and GitHub committed 2022-05-03 20:22:12 +08:00
1 parent 3d2384f31a
commit 5af082eac8
37 files changed
+868 -889
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/ui-default/components/socket/index.js
+17 -67
View File
@@ -1,88 +1,38 @@
import SockJS from 'sockjs-client';
import $ from 'jquery';
import Notification from 'vj/components/notification/index';
import i18n from 'vj/utils/i18n';
import tpl from 'vj/utils/tpl';
const notification = tpl`<div style="position: fixed; left: 0px; bottom: -100px; width: 100%; \
background: rgb(255, 241, 184); z-index: 2147483647; color: rgb(97, 52, 0); \
text-align: center; font-size: 18px; font-family: Consolas, Menlo, Courier, monospace; \
padding: 8px 0px; box-shadow: rgba(0, 0, 0, 0.1) 0px 4px 8px;" id="disconnect_notification" \
>${i18n('Disconnected from the server, trying to reconnect...')}</div>`;
const $el = $(notification);
$el.appendTo('body');
$el.hide();
function create() {
$el.show();
$el.animate({ bottom: 0 }, 100);
}
function remove() {
$el.animate({ bottom: -100 }, 100, () => $el.remove());
}
import ReconnectingWebSocket from 'reconnecting-websocket';
export default class Sock {
constructor(url, showNotification = true, useShared = false) {
this.url = url;
this.shared = false;
this.isreconnect = false;
this.retryCount = 0;
if (useShared && window.SharedWorker) this.initShared();
else this.init();
this.showNotification = showNotification;
}
onauth() {
remove();
this.retryCount = 0;
if (this.onopen) this.onopen(this.sock);
}
init() {
this.sock = new SockJS(this.url);
// SockJS wouldn't send cookie. hack.
this.sock.onopen = () => this.send(document.cookie);
constructor(url) {
const i = new URL(url, window.location.href);
i.protocol = i.protocol.replace('http', 'ws');
this.url = i.toString();
this.closed = false;
this.sock = new ReconnectingWebSocket(this.url);
this.sock.onopen = () => {
console.log('Connected');
this.onopen?.(this.sock);
};
this.sock.onclose = ({ code, reason }) => {
this.retryCount++;
console.warn('Connection closed, ', code, reason);
if (code >= 4000) this.closed = true;
if (!this.closed) {
this.isreconnect = setTimeout(this.init.bind(this), 3000);
}
if (this.showNotification && this.retryCount > 3) create();
if (this.onclose) this.onclose(code, reason);
this.onclose?.(code, reason);
};
this.sock.onmessage = (message) => {
if (process.env.NODE_ENV !== 'production') console.log('Sock.onmessage: ', message);
const msg = JSON.parse(message.data);
if (msg.event === 'auth') {
if (msg.error === 'PermissionError' || msg.error === 'PrivilegeError') {
if (this.showNotification) Notification.info(i18n('Connect fail: Permission denied.'));
this.closed = true;
} else this.onauth();
} else if (this.onmessage) this.onmessage(message);
this.close();
}
} else this.onmessage?.(message);
};
}
initShared() {
this.shared = true;
const worker = new SharedWorker('/sharedworker.js', { name: 'Hydro Shared Connections Worker' });
worker.port.start();
const path = `${new URL(this.url, window.location.href).href.replace('http', 'ws')}/websocket`;
worker.port.postMessage({ type: 'sharedConn', path, cookie: document.cookie });
worker.port.onmessage = (e) => {
if (e.data.type === 'message') this.onmessage({ data: e.data.payload });
};
this.worker = worker;
}
send(data) {
if (this.shared) this.worker.port.postMessage({ type: 'message', path: this.url, payload: data });
else this.sock.send(data);
this.sock.send(data);
}
close() {
this.closed = true;
if (!this.shared) this.sock.close();
this.sock?.close?.();
}
}