Container.ts
135 lines
| 4.4 KiB
| video/mp2t
|
TypeScriptLexer
cin
|
r49 | import { ActivationContext } from "./ActivationContext"; | ||
| import { ValueDescriptor } from "./ValueDescriptor"; | ||||
| import { ActivationError } from "./ActivationError"; | ||||
|
cin
|
r131 | import { ServiceMap, Descriptor, PartialServiceMap, ContainerProvided, Resolver, ContainerServiceMap, ContainerKeys, ContainerResolve, ILifetimeManager } from "./interfaces"; | ||
|
cin
|
r49 | import { TraceSource } from "../log/TraceSource"; | ||
|
cin
|
r120 | import { Configuration, RegistrationMap } from "./Configuration"; | ||
|
cin
|
r49 | import { Cancellation } from "../Cancellation"; | ||
|
cin
|
r129 | import { MapOf, IDestroyable } from "../interfaces"; | ||
|
cin
|
r118 | import { isDescriptor } from "./traits"; | ||
|
cin
|
r131 | import { LifetimeManager } from "./LifetimeManager"; | ||
| import { each } from "../safe"; | ||||
|
cin
|
r49 | |||
| const trace = TraceSource.get("@implab/core/di/ActivationContext"); | ||||
|
cin
|
r129 | export class Container<S extends object = any> implements Resolver<S>, IDestroyable { | ||
|
cin
|
r120 | readonly _services: ContainerServiceMap<S>; | ||
|
cin
|
r49 | |||
|
cin
|
r114 | readonly _cache: MapOf<any>; | ||
|
cin
|
r131 | readonly _lifetimeManager: ILifetimeManager; | ||
|
cin
|
r114 | 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
|
r131 | this._lifetimeManager = new LifetimeManager(); | ||
|
cin
|
r49 | } | ||
| getRootContainer() { | ||||
| return this._root; | ||||
| } | ||||
| getParent() { | ||||
| return this._parent; | ||||
| } | ||||
|
cin
|
r131 | getLifetimeManager() { | ||
| return this._lifetimeManager; | ||||
| } | ||||
|
cin
|
r120 | resolve<K extends ContainerKeys<S>>(name: K, def?: ContainerResolve<S, K>): ContainerResolve<S, K> { | ||
|
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
|
r131 | const context = new ActivationContext<S>(this, this._services, String(name), d); | ||
|
cin
|
r114 | try { | ||
|
cin
|
r131 | return d.activate(context); | ||
|
cin
|
r114 | } catch (error) { | ||
| throw new ActivationError(name.toString(), context.getStack(), error); | ||||
| } | ||||
|
cin
|
r49 | } | ||
| } | ||||
| /** | ||||
| * @deprecated use resolve() method | ||||
| */ | ||||
|
cin
|
r120 | getService<K extends ContainerKeys<S>>(name: K, def?: ContainerResolve<S, K>) { | ||
|
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
|
r131 | each(data, (v, k) => this.register(k, v)); | ||
|
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); | ||||
| } | ||||
|
cin
|
r129 | destroy() { | ||
| return this.dispose(); | ||||
| } | ||||
|
cin
|
r49 | 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
|
r120 | async configure(config: string | RegistrationMap<S>, 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
|
r120 | createChildContainer<S2 extends object = S>(): Container<S & S2> { | ||
|
cin
|
r114 | return new Container<S & S2>(this as any); | ||
|
cin
|
r49 | } | ||
| } | ||||