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

core: gateway: add metadata field

Browse Source
This commit is contained in:
undefined committed 2025-06-16 13:22:21 +08:00
1 parent 436733f621
commit 1bff3ecce0
2 files changed
+16 -11
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/handler/connection.ts
+15 -10
View File
@@ -33,24 +33,29 @@ class WebsocketEventsConnectionManagerHandler extends ConnectionHandler {
}
async message(payload: any) {
if (!['resume', 'subscribe'].includes(payload.operation)) return;
const accept = [];
const reject = [];
const session = payload?.credential
const session = payload.credential
? await TokenModel.get(payload.credential, TokenModel.TYPE_SESSION)
: null;
if (!session && payload.operation !== 'resume') return;
if (payload.operation === 'resume' && !this.privileged) return;
const user = payload.operation === 'resume' ? null : await UserModel.getById('system', session.uid);
for (const channel of payload.channels) {
const op = payload.operation || '';
if (!session && op !== 'resume') return;
if (op === 'resume' && !this.privileged) return;
const user = op === 'resume' ? null : await UserModel.getById('system', session.uid);
for (const channel of payload.channels || []) {
try {
const result = payload.operation === 'resume'
const result = op === 'resume'
? { ok: true, channel }
// eslint-disable-next-line no-await-in-loop
: await this.ctx.bail('subscription/subscribe', channel, user);
if (result?.ok) accept.push(result.channel);
else reject.push(channel);
: await this.ctx.bail('subscription/subscribe', channel, user, this.privileged ? payload.metadata || {} : {});
if (result?.ok) {
// eslint-disable-next-line no-await-in-loop
await this.accept(result.channel);
accept.push(result.channel);
} else reject.push(channel);
} catch (e) {
logger.error('Error subscribing to channel %s for user %s: %s', channel, user._id, e);
logger.error('Error subscribing to channel %s for user %s: %s', channel, user?._id, e);
reject.push(channel);
}
}
packages/hydrooj/src/service/bus.ts
+1 -1
View File
@@ -23,7 +23,7 @@ export interface EventMap {
'app/reload': (entries: Set<string>) => VoidReturn
'subscription/init': (h: ConnectionHandler<Context>, privileged: boolean) => VoidReturn
'subscription/subscribe': (channel: string, user: User) => VoidReturn
'subscription/subscribe': (channel: string, user: User, metadata: Record<string, string>) => VoidReturn
'subscription/enable': (channel: string, h: ConnectionHandler<Context>, privileged: boolean) => VoidReturn
'app/watch/change': (path: string) => VoidReturn