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: upgrade to context tracing

Browse Source
This commit is contained in:
undefined committed 2024-10-05 20:38:20 +08:00
1 parent e79e978b15
commit 28b3abf441
9 files changed
+42 -38
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

framework/framework/server.ts
+10 -12
View File
@@ -338,7 +338,6 @@ export class WebService extends Service {
HandlerCommon = HandlerCommon;
Handler = Handler;
ConnectionHandler = ConnectionHandler;
handlerCtxBase = this.ctx;
constructor(ctx: Context, public config: WebServiceConfig) {
super(ctx, 'server', true);
@@ -439,9 +438,6 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
}
socket.close();
});
this.ctx.inject(['server'], (c) => {
this.handlerCtxBase = c;
});
}
async listen() {
@@ -460,7 +456,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
private async handleHttp(ctx: KoaContext, HandlerClass, checker) {
const { args } = ctx.HydroContext;
Object.assign(args, ctx.params);
const h = new HandlerClass(ctx, this.handlerCtxBase);
const h = new HandlerClass(ctx, this.ctx);
ctx.handler = h;
const method = ctx.method.toLowerCase();
try {
@@ -534,7 +530,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
private async handleWS(ctx: KoaContext, HandlerClass, checker, conn, layer) {
const { args } = ctx.HydroContext;
const h = new HandlerClass(ctx, this.handlerCtxBase);
const h = new HandlerClass(ctx, this.ctx);
await this.ctx.parallel('connection/create', h);
ctx.handler = h;
h.conn = conn;
@@ -599,8 +595,8 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
}
private register(type: 'route' | 'conn', routeName: string, path: string, HandlerClass: any, ...permPrivChecker) {
const name = HandlerClass.name;
if (!HandlerClass?.[kHandler] || !isClass(HandlerClass)) throw new Error('Invalid registration.');
const name = HandlerClass.name;
if (this.registrationCount[name] && this.registry[name] !== HandlerClass) {
logger.warn('Route with name %s already exists.', name);
}
@@ -617,7 +613,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
const dispose = router.disposeLastOp;
// @ts-ignore
this.ctx.parallel(`handler/register/${name}`, HandlerClass);
this[Context.current]?.on('dispose', () => {
this.ctx.on('dispose', () => {
this.registrationCount[name]--;
if (!this.registrationCount[name]) delete this.registry[name];
dispose();
@@ -650,7 +646,7 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
const dispose = () => {
this[name] = this[name].filter((i) => i !== layer);
};
this[Context.current]?.on('dispose', dispose);
this[Context.origin]?.on('dispose', dispose);
return dispose;
}
@@ -683,9 +679,11 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
public registerRenderer(name: string, func: RendererFunction) {
if (this.renderers[name]) logger.warn('Renderer %s already exists.', name);
this.renderers[name] = func;
this[Context.current]?.on('dispose', () => {
delete this.renderers[name];
this.ctx.effect(() => {
this.renderers[name] = func;
return () => {
delete this.renderers[name];
};
});
}
}