Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: fix use MultiInstanceProto from other modules
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,28 @@ | ||
| import { LifecycleHook } from '@eggjs/tegg-lifecycle'; | ||
| import { EggProtoImplClass, PrototypeUtil } from '@eggjs/core-decorator'; | ||
| import { | ||
| EggPrototypeCreatorFactory, EggPrototypeFactory, | ||
| LoadUnit, | ||
| LoadUnitLifecycleContext, | ||
| } from '@eggjs/tegg-metadata'; | ||
|
|
||
| export class LoadUnitMultiInstanceProtoHook implements LifecycleHook<LoadUnitLifecycleContext, LoadUnit> { | ||
| multiInstanceClazzSet: Set<EggProtoImplClass> = new Set(); | ||
|
|
||
| async preCreate(ctx: LoadUnitLifecycleContext, loadUnit: LoadUnit): Promise<void> { | ||
| const clazzList = ctx.loader.load(); | ||
| const multiInstanceClazzList = Array.from(this.multiInstanceClazzSet); | ||
| for (const clazz of clazzList) { | ||
| if (PrototypeUtil.isEggMultiInstancePrototype(clazz)) { | ||
| this.multiInstanceClazzSet.add(clazz); | ||
| } | ||
| } | ||
| for (const clazz of multiInstanceClazzList) { | ||
| const protos = await EggPrototypeCreatorFactory.createProto(clazz, loadUnit); | ||
| for (const proto of protos) { | ||
| EggPrototypeFactory.instance.registerPrototype(proto, loadUnit); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,254 @@ | ||
| import assert from 'node:assert'; | ||
| import util from 'node:util'; | ||
| import { Graph, GraphNode, GraphNodeObj, ModuleConfigUtil, ModuleReference } from '@eggjs/tegg-common-util'; | ||
| import { | ||
| AccessLevel, | ||
| EggProtoImplClass, | ||
| EggPrototypeName, | ||
| INIT_TYPE_TRY_ORDER, | ||
| InitTypeQualifierAttribute, | ||
| LoadUnitNameQualifierAttribute, | ||
| PrototypeUtil, | ||
| QualifierInfo, | ||
| QualifierUtil, | ||
| } from '@eggjs/core-decorator'; | ||
|
|
||
| export interface InstanceClazzMeta { | ||
| name: PropertyKey; | ||
| qualifiers: QualifierInfo[]; | ||
| accessLevel: AccessLevel, | ||
| instanceModule: GraphNode<ModuleNode>; | ||
| ownerModule: GraphNode<ModuleNode>; | ||
| } | ||
|
|
||
| export type ClazzMetaMap = Record<EggPrototypeName, InstanceClazzMeta[]>; | ||
|
|
||
| function verifyQualifier(clazzQualifiers: QualifierInfo[], qualifier: QualifierInfo) { | ||
| const selfQualifiers = clazzQualifiers.find(t => t.attribute === qualifier.attribute); | ||
| return selfQualifiers?.value === qualifier.value; | ||
| } | ||
|
|
||
| function verifyQualifiers(clazzQualifiers: QualifierInfo[], qualifiers: QualifierInfo[]) { | ||
| for (const qualifier of qualifiers) { | ||
| if (!verifyQualifier(clazzQualifiers, qualifier)) { | ||
| return false; | ||
| } | ||
| } | ||
| return true; | ||
| } | ||
|
|
||
| export class ClazzMap { | ||
| private clazzMap: ClazzMetaMap; | ||
|
|
||
| constructor(graph: Graph<ModuleNode>) { | ||
| this.build(graph); | ||
| } | ||
|
|
||
| private build(graph: Graph<ModuleNode>) { | ||
| /** | ||
| * 1. iterate all module get all MultiInstanceClazz | ||
| * 2. iterate MultiInstanceClazz and all module get object meta | ||
| * 3. iterate object meta and build clazz map | ||
| */ | ||
| const clazzMap: ClazzMetaMap = {}; | ||
| for (const ownerNode of graph.nodes.values()) { | ||
| for (const clazz of ownerNode.val.getClazzList()) { | ||
| const qualifiers = QualifierUtil.getProtoQualifiers(clazz); | ||
| if (PrototypeUtil.isEggMultiInstancePrototype(clazz)) { | ||
| for (const instanceNode of graph.nodes.values()) { | ||
| const property = PrototypeUtil.getMultiInstanceProperty(clazz, { | ||
| unitPath: instanceNode.val.moduleConfig.path, | ||
| }); | ||
| assert(property, `multi instance property not found for ${clazz.name}`); | ||
| for (const info of property.objects) { | ||
| clazzMap[info.name] = clazzMap[info.name] || []; | ||
| clazzMap[info.name].push({ | ||
| name: info.name, | ||
| accessLevel: PrototypeUtil.getAccessLevel(clazz, { | ||
| unitPath: instanceNode.val.moduleConfig.path, | ||
| }) as AccessLevel, | ||
| qualifiers: [ | ||
| ...qualifiers, | ||
| ...info.qualifiers, | ||
| ], | ||
| instanceModule: instanceNode, | ||
| ownerModule: ownerNode, | ||
| }); | ||
| } | ||
| } | ||
| } else { | ||
| const property = PrototypeUtil.getProperty(clazz); | ||
| assert(property, `property not found for ${clazz.name}`); | ||
| clazzMap[property.name] = clazzMap[property.name] || []; | ||
| clazzMap[property.name].push({ | ||
| name: property.name, | ||
| accessLevel: PrototypeUtil.getAccessLevel(clazz, { | ||
| unitPath: ownerNode.val.moduleConfig.path, | ||
| }) as AccessLevel, | ||
| qualifiers, | ||
| ownerModule: ownerNode, | ||
| instanceModule: ownerNode, | ||
| }); | ||
| } | ||
| } | ||
| } | ||
| this.clazzMap = clazzMap; | ||
| } | ||
|
|
||
| findDependencyModule(objName: EggPrototypeName, properQualifiers: QualifierInfo[], intoModule: GraphNode<ModuleNode>): GraphNode<ModuleNode>[] { | ||
| const result: Set<GraphNode<ModuleNode>> = new Set(); | ||
| const objInfo = this.clazzMap[objName]; | ||
| if (!objInfo) { | ||
| return []; | ||
| } | ||
| let mayObjs = objInfo.filter(obj => { | ||
| // 1. check accessLevel | ||
| if (obj.instanceModule !== intoModule && obj.accessLevel === AccessLevel.PRIVATE) { | ||
| return false; | ||
| } | ||
| // 2. check qualifier | ||
| return verifyQualifiers(obj.qualifiers, properQualifiers); | ||
| }); | ||
|
|
||
| // 3. auto set init type qualifier | ||
| if (mayObjs.length > 1) { | ||
| const initTypeQualifiers = INIT_TYPE_TRY_ORDER.map(type => ({ | ||
| attribute: InitTypeQualifierAttribute, | ||
| value: type, | ||
| })); | ||
| for (const initTypeQualifier of initTypeQualifiers) { | ||
| const mayInitTypeObjs = mayObjs.filter(obj => { | ||
| return verifyQualifiers(obj.qualifiers, [ | ||
| ...properQualifiers, | ||
| initTypeQualifier, | ||
| ]); | ||
| }); | ||
| if (mayInitTypeObjs.length > 0) { | ||
| mayObjs = mayInitTypeObjs; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| // 4. auto set load unit name qualifier | ||
| if (mayObjs.length > 1) { | ||
| const moduleNameQualifiers = { | ||
| attribute: LoadUnitNameQualifierAttribute, | ||
| value: intoModule.val.name, | ||
| }; | ||
| const mayLoadUnitNameObjs = mayObjs.filter(obj => { | ||
| return verifyQualifiers(obj.qualifiers, [ | ||
| ...properQualifiers, | ||
| moduleNameQualifiers, | ||
| ]); | ||
| }); | ||
| if (mayLoadUnitNameObjs.length > 0) { | ||
| mayObjs = mayLoadUnitNameObjs; | ||
| } | ||
| } | ||
|
|
||
| if (mayObjs.length > 1) { | ||
| const message = util.format('multi class found for %s@%o in module %j', | ||
| objName, | ||
| properQualifiers, | ||
| mayObjs.map(t => { | ||
| return t.instanceModule.val.moduleConfig.path; | ||
| })); | ||
| throw new Error(message); | ||
| } | ||
|
|
||
| for (const obj of mayObjs) { | ||
| result.add(obj.instanceModule); | ||
| result.add(obj.ownerModule); | ||
| } | ||
| return Array.from(result); | ||
| } | ||
| } | ||
|
|
||
| export class ModuleNode implements GraphNodeObj { | ||
| readonly id: string; | ||
| readonly name: string; | ||
| readonly moduleConfig: ModuleReference; | ||
| private readonly clazzList: EggProtoImplClass[]; | ||
|
|
||
| constructor(moduleConfig: ModuleReference) { | ||
| this.moduleConfig = moduleConfig; | ||
| this.id = moduleConfig.path; | ||
| this.name = ModuleConfigUtil.readModuleNameSync(moduleConfig.path); | ||
| this.clazzList = []; | ||
| } | ||
|
|
||
| addClazz(clazz: EggProtoImplClass) { | ||
| if (!this.clazzList.includes(clazz)) { | ||
| this.clazzList.push(clazz); | ||
| } | ||
| const defaultQualifier = [{ | ||
| attribute: InitTypeQualifierAttribute, | ||
| value: PrototypeUtil.getInitType(clazz, { | ||
| unitPath: this.moduleConfig.path, | ||
| })!, | ||
| }, { | ||
| attribute: LoadUnitNameQualifierAttribute, | ||
| value: this.name, | ||
| }]; | ||
| for (const qualifier of defaultQualifier) { | ||
| QualifierUtil.addProtoQualifier(clazz, qualifier.attribute, qualifier.value); | ||
| } | ||
| } | ||
|
|
||
| toString() { | ||
| return `${this.name}@${this.moduleConfig.path}`; | ||
| } | ||
|
|
||
| getClazzList(): readonly EggProtoImplClass[] { | ||
| return this.clazzList; | ||
| } | ||
| } | ||
|
|
||
| export class AppGraph { | ||
| private graph: Graph<ModuleNode>; | ||
| private clazzMap: ClazzMap; | ||
| moduleConfigList: Array<ModuleReference>; | ||
|
|
||
| constructor() { | ||
| this.graph = new Graph<ModuleNode>(); | ||
| } | ||
|
|
||
| addNode(moduleNode: ModuleNode) { | ||
| if (!this.graph.addVertex(new GraphNode(moduleNode))) { | ||
| throw new Error(`duplicate module: ${moduleNode}`); | ||
| } | ||
| } | ||
|
|
||
| build() { | ||
| this.clazzMap = new ClazzMap(this.graph); | ||
|
|
||
| // 1. iterate all modules | ||
| for (const node of this.graph.nodes.values()) { | ||
| // 2. iterate all class | ||
| for (const clazz of node.val.getClazzList()) { | ||
| const injectObjects = PrototypeUtil.getInjectObjects(clazz); | ||
| // 3. iterate all inject objects | ||
| for (const injectObject of injectObjects) { | ||
| const properQualifiers = QualifierUtil.getProperQualifiers(clazz, injectObject.refName); | ||
| // 4. find dependency module | ||
| const dependencyModules = this.clazzMap.findDependencyModule(injectObject.objName, properQualifiers, node); | ||
| for (const moduleNode of dependencyModules) { | ||
| // 5. add edge | ||
| if (node !== moduleNode) { | ||
| this.graph.addEdge(node, moduleNode); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| sort() { | ||
| const loopPath = this.graph.loopPath(); | ||
| if (loopPath) { | ||
| throw new Error('module has recursive deps: ' + loopPath); | ||
| } | ||
| this.moduleConfigList = this.graph.sort() | ||
| .map(t => t.val.moduleConfig); | ||
| } | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.