Container.ts
140 lines
| 4.5 KiB
| video/mp2t
|
TypeScriptLexer
cin
|
r49 | import { ActivationContext } from "./ActivationContext"; | ||
| import { ValueDescriptor } from "./ValueDescriptor"; | ||||
| import { ActivationError } from "./ActivationError"; | ||||
|
cin
|
r118 | import { ServiceMap, Descriptor, PartialServiceMap, ContainerServices, Resolver } from "./interfaces"; | ||
|
cin
|
r49 | import { TraceSource } from "../log/TraceSource"; | ||
| import { Configuration } from "./Configuration"; | ||||
| import { Cancellation } from "../Cancellation"; | ||||
|
cin
|
r114 | import { MapOf } from "../interfaces"; | ||
|
cin
|
r118 | import { isDescriptor } from "./traits"; | ||
|
cin
|
r49 | |||
| const trace = TraceSource.get("@implab/core/di/ActivationContext"); | ||||
|
cin
|
r115 | export class Container<S = any> implements Resolver<S> { | ||
|
cin
|
r118 | readonly _services: PartialServiceMap<ContainerServices<S>>; | ||
|
cin
|
r49 | |||
|
cin
|
r114 | readonly _cache: MapOf<any>; | ||
| readonly _cleanup: (() => void)[]; | ||||
|
cin
|
r49 | |||
|
cin
|
r114 | readonly _root: Container<S>; | ||
|
cin
|
r49 | |||
|
cin
|
r114 | readonly _parent?: Container<S>; | ||
|
cin
|
r49 | |||
|
cin
|
r114 | _disposed: boolean; | ||
|
cin
|
r49 | |||
|
cin
|
r114 | constructor(parent?: Container<S>) { | ||
|
cin
|
r49 | this._parent = parent; | ||
| this._services = parent ? Object.create(parent._services) : {}; | ||||
| this._cache = {}; | ||||
| this._cleanup = []; | ||||
| this._root = parent ? parent.getRootContainer() : this; | ||||
|
cin
|
r115 | this._services.container = new ValueDescriptor(this) as any; | ||
|
cin
|
r114 | this._disposed = false; | ||
|
cin
|
r49 | } | ||
| getRootContainer() { | ||||
| return this._root; | ||||
| } | ||||
| getParent() { | ||||
| return this._parent; | ||||
| } | ||||
|
cin
|
r115 | resolve<K extends keyof ContainerServices<S>, T extends ContainerServices<S>[K] = ContainerServices<S>[K]>(name: K, def?: T): T { | ||
|
cin
|
r49 | trace.debug("resolve {0}", name); | ||
| const d = this._services[name]; | ||||
| if (d === undefined) { | ||||
|
cin
|
r114 | if (def !== undefined) | ||
|
cin
|
r49 | return def; | ||
| else | ||||
| throw new Error("Service '" + name + "' isn't found"); | ||||
|
cin
|
r114 | } else { | ||
|
cin
|
r49 | |||
|
cin
|
r114 | const context = new ActivationContext<S>(this, this._services); | ||
| try { | ||||
|
cin
|
r115 | return context.activate(d as Descriptor<S, T>, name.toString()); | ||
|
cin
|
r114 | } catch (error) { | ||
| throw new ActivationError(name.toString(), context.getStack(), error); | ||||
| } | ||||
|
cin
|
r49 | } | ||
| } | ||||
| /** | ||||
| * @deprecated use resolve() method | ||||
| */ | ||||
|
cin
|
r115 | getService<K extends keyof S, T extends ContainerServices<S>[K] = ContainerServices<S>[K]>(name: K, def?: T) { | ||
|
cin
|
r114 | return this.resolve(name, def); | ||
|
cin
|
r49 | } | ||
|
cin
|
r115 | register<K extends keyof S>(name: K, service: Descriptor<S, S[K]>): this; | ||
| register(services: PartialServiceMap<S>): this; | ||||
| register<K extends keyof S>(nameOrCollection: K | ServiceMap<S>, service?: Descriptor<S, S[K]>) { | ||||
|
cin
|
r49 | if (arguments.length === 1) { | ||
|
cin
|
r114 | const data = nameOrCollection as ServiceMap<S>; | ||
|
cin
|
r115 | |||
|
cin
|
r114 | for (const name in data) { | ||
| if (Object.prototype.hasOwnProperty.call(data, name)) { | ||||
|
cin
|
r115 | this.register(name, data[name] as Descriptor<S, S[keyof S]>); | ||
|
cin
|
r114 | } | ||
| } | ||||
|
cin
|
r49 | } else { | ||
| if (!isDescriptor(service)) | ||||
| throw new Error("The service parameter must be a descriptor"); | ||||
|
cin
|
r115 | this._services[nameOrCollection as K] = service as any; | ||
|
cin
|
r49 | } | ||
| return this; | ||||
| } | ||||
|
cin
|
r114 | onDispose(callback: () => void) { | ||
|
cin
|
r49 | if (!(callback instanceof Function)) | ||
| throw new Error("The callback must be a function"); | ||||
| this._cleanup.push(callback); | ||||
| } | ||||
| dispose() { | ||||
|
cin
|
r114 | if (this._disposed) | ||
| return; | ||||
| this._disposed = true; | ||||
| for (const f of this._cleanup) | ||||
| f(); | ||||
|
cin
|
r49 | } | ||
| /** | ||||
| * @param{String|Object} config | ||||
| * The configuration of the contaier. Can be either a string or an object, | ||||
| * if the configuration is an object it's treated as a collection of | ||||
| * services which will be registed in the contaier. | ||||
| * | ||||
| * @param{Function} opts.contextRequire | ||||
| * The function which will be used to load a configuration or types for services. | ||||
| * | ||||
| */ | ||||
|
cin
|
r59 | async configure(config: string | object, opts?: any, ct = Cancellation.none) { | ||
|
cin
|
r115 | const c = new Configuration<S>(this); | ||
|
cin
|
r49 | |||
| if (typeof (config) === "string") { | ||||
|
cin
|
r59 | return c.loadConfiguration(config, opts && opts.contextRequire, ct); | ||
|
cin
|
r49 | } else { | ||
| return c.applyConfiguration(config, opts && opts.contextRequire, ct); | ||||
| } | ||||
| } | ||||
|
cin
|
r114 | createChildContainer<S2 extends { container?: Container<S & S2> } = S>(): Container<S & S2> { | ||
| return new Container<S & S2>(this as any); | ||||
|
cin
|
r49 | } | ||
|
cin
|
r107 | has(id: string | number) { | ||
|
cin
|
r49 | return id in this._cache; | ||
| } | ||||
|
cin
|
r107 | get(id: string | number) { | ||
|
cin
|
r49 | return this._cache[id]; | ||
| } | ||||
|
cin
|
r107 | store(id: string | number, value: any) { | ||
|
cin
|
r49 | return (this._cache[id] = value); | ||
| } | ||||
| } | ||||