Skip to content

Commit f50d1fc

Browse files
committed
fix: refactoring
1 parent 84ff1e8 commit f50d1fc

File tree

2 files changed

+12
-14
lines changed

2 files changed

+12
-14
lines changed

src/Logger.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import {Injectable, LoggerService} from "@nestjs/common";
1+
import {Inject, Injectable, LoggerService} from "@nestjs/common";
22
import * as util from "node:util";
33
import type {Logger as PinoLogger} from "pino";
44
import crypto from "node:crypto";
@@ -8,7 +8,7 @@ import {PinoMessageSymbol} from "./logger.constants";
88
export class Logger implements LoggerService {
99
private static cache = new Map<string, any>();
1010

11-
constructor(private readonly pinoInstance: PinoLogger) {
11+
constructor(@Inject('PINO_LOGGER_INSTANCE') private readonly pinoInstance: PinoLogger) {
1212
}
1313

1414
debug(message: any, ...optionalParams: any[]): any {

src/logger.module.ts

+10-12
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import pino from "pino";
66

77
export interface LoggerModuleAsyncOptions extends Pick<ModuleMetadata, 'imports'> {
88
inject?: any[];
9-
useFactory?: (...args: any[]) => Promise<LoggerOptions> | LoggerOptions;
9+
useFactory: (...args: any[]) => Promise<LoggerOptions> | LoggerOptions;
1010
}
1111

1212
@Global()
@@ -30,8 +30,8 @@ export class LoggerModule {
3030

3131
return {
3232
module: LoggerModule,
33-
providers: [loggerModuleOptions, pinoProvider, Logger],
34-
exports: [loggerModuleOptions, pinoProvider, Logger]
33+
providers: [Logger, loggerModuleOptions, pinoProvider],
34+
exports: [Logger, loggerModuleOptions, pinoProvider]
3535
};
3636
}
3737

@@ -40,26 +40,24 @@ export class LoggerModule {
4040
Logger
4141
];
4242

43+
providers.push({
44+
provide: PINO_LOGGER_OPTIONS,
45+
useFactory: options.useFactory,
46+
inject: options.inject || [],
47+
});
48+
4349
const pinoProvider: Provider = {
4450
provide: PINO_LOGGER_INSTANCE,
4551
useFactory: async (options: LoggerOptions) => pino(options),
4652
inject: [PINO_LOGGER_OPTIONS],
4753
}
4854
providers.push(pinoProvider);
4955

50-
if (options.useFactory) {
51-
providers.push({
52-
provide: PINO_LOGGER_OPTIONS,
53-
useFactory: options.useFactory,
54-
inject: options.inject || [],
55-
});
56-
}
57-
5856
return {
5957
module: LoggerModule,
6058
imports: options.imports ?? [],
6159
providers,
62-
exports: [pinoProvider, Logger]
60+
exports: [Logger, pinoProvider]
6361
};
6462
}
6563
}

0 commit comments

Comments
 (0)