fix(module): updates decorators type to accept a class

jmcdo29 · Jul 18, 2020 · 1bc3df7 · 1bc3df7
1 parent 8f3ab87
commit 1bc3df7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/packages/nestjs-module/src/decorators/ogma-logger.decorator.ts b/packages/nestjs-module/src/decorators/ogma-logger.decorator.ts
@@ -1,5 +1,6 @@
 import { Inject } from '@nestjs/common';
 import { createProviderToken } from '../ogma.provider';
+import { Type } from '../interfaces';
 
-export const OgmaLogger = (topic: string | (() => any)) =>
+export const OgmaLogger = (topic: string | (() => any) | Type<any>) =>
   Inject(createProviderToken(typeof topic === 'function' ? topic.name : topic));
diff --git a/packages/nestjs-module/src/ogma.provider.ts b/packages/nestjs-module/src/ogma.provider.ts
@@ -89,7 +89,9 @@ export function createProviderToken(topic: string): string {
   return OGMA_SERVICE_TOKEN + ':' + topic;
 }
 
-export function createLoggerProviders(topic: string | (() => any)): Provider[] {
+export function createLoggerProviders(
+  topic: string | (() => any) | Type<any>,
+): Provider[] {
   topic = typeof topic === 'function' ? topic.name : topic;
   const token = createProviderToken(topic);
   return [