Skip to content

Commit 3158854

Browse files
committed
feat: expand register add loadUnit
1 parent e94c045 commit 3158854

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

plugin/controller/lib/AppLoadUnitControllerHook.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ export class AppLoadUnitControllerHook implements LifecycleHook<LoadUnitLifecycl
2525
throw new Error(`not find controller implement for ${String(proto.name)} which type is ${metadata.type}`);
2626
}
2727
ControllerMetadataManager.instance.addController(metadata);
28-
await register.register(this.rootProtoManager);
28+
await register.register(this.rootProtoManager, obj);
2929
}
3030
}
3131
}
Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
1+
import { LoadUnit } from '@eggjs/tegg-metadata';
12
import { RootProtoManager } from './RootProtoManager';
23

34
export interface ControllerRegister {
4-
register(rootProtoManager: RootProtoManager): Promise<void>;
5+
register(rootProtoManager: RootProtoManager, loadUnit?: LoadUnit): Promise<void>;
56
}

0 commit comments

Comments
 (0)