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

framework: mixin: support callback style

Browse Source
This commit is contained in:
undefined committed 2025-03-12 15:49:04 +08:00
1 parent 51c3c4f30f
commit af1b12deed
3 files changed
+9 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

framework/framework/server.ts
+5 -4
View File
@@ -752,8 +752,9 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
this.captureAllRoutes[prefix] = cb;
}
private _applyMixin(Target: any, MixinClass: Partial<any>) {
private _applyMixin(Target: any, MixinClass: Partial<any> | ((t: Partial<any>) => Partial<any>)) {
if (!('prototype' in Target)) throw new Error('Target must be a class.');
if (typeof MixinClass === 'function') MixinClass = MixinClass(Target.prototype);
this.ctx.effect(() => {
let oldValue = null;
for (const val of Object.getOwnPropertyNames(MixinClass)) {
@@ -782,15 +783,15 @@ ${c.response.status} ${endTime - startTime}ms ${c.response.length}`);
});
}
public handlerMixin(MixinClass: Partial<HandlerCommon<C>>) {
public handlerMixin(MixinClass: Partial<HandlerCommon<C>> | ((s: HandlerCommon<C>) => Partial<HandlerCommon<C>>)) {
return this._applyMixin(HandlerCommon, MixinClass);
}
public httpHandlerMixin(MixinClass: Partial<Handler<C>>) {
public httpHandlerMixin(MixinClass: Partial<Handler<C>> | ((s: Handler<C>) => Partial<Handler<C>>)) {
return this._applyMixin(Handler, MixinClass);
}
public wsHandlerMixin(MixinClass: Partial<ConnectionHandler<C>>) {
public wsHandlerMixin(MixinClass: Partial<ConnectionHandler<C>> | ((s: ConnectionHandler<C>) => Partial<ConnectionHandler<C>>)) {
return this._applyMixin(ConnectionHandler, MixinClass);
}