add lifecycle in service decorator

botflux · May 5, 2021 · 1348dfa · 1348dfa
1 parent d799b85
commit 1348dfa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/V2/Implementation/ServiceLoader.ts b/src/V2/Implementation/ServiceLoader.ts
@@ -1,8 +1,8 @@
 import {ContainerBuilderInterface, LifeCycle, ServiceKey, SyncServiceProviderInterface} from '../Interfaces'
 
-export const Service = (serviceKey: ServiceKey): ClassDecorator => target => {
+export const Service = (serviceKey: ServiceKey, lifeCycle: LifeCycle): ClassDecorator => target => {
     // @ts-ignore
-    Reflect.defineMetadata('Service@ServiceKey', serviceKey, target)
+    Reflect.defineMetadata('Service@ServiceKey', {serviceKey, lifeCycle}, target)
 }
 export const Inject = (serviceKey: ServiceKey): ParameterDecorator => (target, propertyKey, parameterIndex) => {
     // @ts-ignore
@@ -11,11 +11,13 @@ export const Inject = (serviceKey: ServiceKey): ParameterDecorator => (target, p
     // @ts-ignore
     Reflect.defineMetadata('Inject@ServiceKey', injectionTokens, target)
 }
+
 type Constructor<T> = { new(...args: any[]): T }
+
 export const reflectServiceLoader = (services: Constructor<unknown>[]) => (containerBuilder: ContainerBuilderInterface) => {
     const servicesFactories = services.map(constructor => {
         // @ts-ignore
-        const serviceKey = Reflect.getOwnMetadata('Service@ServiceKey', constructor)
+        const { serviceKey, lifeCycle } = Reflect.getOwnMetadata('Service@ServiceKey', constructor)
 
         // @ts-ignore
         const injectionTokens: { [key: string]: string } = Reflect.getOwnMetadata('Inject@ServiceKey', constructor) || {}
@@ -27,8 +29,8 @@ export const reflectServiceLoader = (services: Constructor<unknown>[]) => (conta
             })
 
             return new constructor(...factoryParameters)
-        }]
+        }, lifeCycle]
     })
 
-    servicesFactories.forEach(([serviceKey, factory]) => containerBuilder.addFactory(serviceKey, factory, LifeCycle.Singleton))
+    servicesFactories.forEach(([serviceKey, factory, lifeCycle]) => containerBuilder.addFactory(serviceKey, factory, lifeCycle))
 }
diff --git a/tests/V2/ReflectionContainer.spec.ts b/tests/V2/ReflectionContainer.spec.ts
@@ -1,11 +1,12 @@
 import 'reflect-metadata'
 import {createContainerBuilder} from '../../src/V2/Implementation/Container'
 import {Inject, reflectServiceLoader, Service} from '../../src/V2/Implementation/ServiceLoader'
+import {LifeCycle} from '../../src/V2/Interfaces'
 
-@Service("MyService")
+@Service("MyService", LifeCycle.Singleton)
 class MyService {}
 
-@Service("AnotherService")
+@Service("AnotherService", LifeCycle.Singleton)
 class AnotherService {
     constructor(
         @Inject("MyService") public readonly myService: MyService