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

server: add handlerCtxBase

Browse Source
This commit is contained in:
undefined committed 2024-06-11 15:12:56 +08:00
1 parent 457d820bb3
commit 58d0c75995
1 file changed
+6 -2
Download Patch File Download Diff File Expand all files Collapse all files
framework/framework/server.ts
+6 -2
View File
@@ -322,6 +322,7 @@ export class WebService extends Service {
HandlerCommon = HandlerCommon; HandlerCommon = HandlerCommon;
Handler = Handler; Handler = Handler;
ConnectionHandler = ConnectionHandler; ConnectionHandler = ConnectionHandler;
handlerCtxBase = this.ctx;
constructor(ctx: Context, public config: WebServiceConfig) { constructor(ctx: Context, public config: WebServiceConfig) {
super(ctx, 'server', true); super(ctx, 'server', true);
@@ -420,6 +421,9 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
} }
socket.close(); socket.close();
}); });
this.ctx.inject(['server'], (c) => {
this.handlerCtxBase = c;
});
} }
async listen() { async listen() {
@@ -438,7 +442,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
private async handleHttp(ctx: KoaContext, HandlerClass, checker) { private async handleHttp(ctx: KoaContext, HandlerClass, checker) {
const { args } = ctx.HydroContext; const { args } = ctx.HydroContext;
Object.assign(args, ctx.params); Object.assign(args, ctx.params);
const h = new HandlerClass(ctx, this.ctx); const h = new HandlerClass(ctx, this.handlerCtxBase);
ctx.handler = h; ctx.handler = h;
const method = ctx.method.toLowerCase(); const method = ctx.method.toLowerCase();
try { try {
@@ -511,7 +515,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
private async handleWS(ctx: KoaContext, HandlerClass, checker, conn, layer) { private async handleWS(ctx: KoaContext, HandlerClass, checker, conn, layer) {
const { args } = ctx.HydroContext; const { args } = ctx.HydroContext;
const h = new HandlerClass(ctx, this.ctx); const h = new HandlerClass(ctx, this.handlerCtxBase);
await this.ctx.parallel('connection/create', h); await this.ctx.parallel('connection/create', h);
ctx.handler = h; ctx.handler = h;
h.conn = conn; h.conn = conn;