105 lines
3.2 KiB
TypeScript
105 lines
3.2 KiB
TypeScript
///
|
|
/// Copyright © 2016-2022 The Thingsboard Authors
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
///
|
|
|
|
import config from 'config';
|
|
import path from 'path';
|
|
import DailyRotateFile from 'winston-daily-rotate-file';
|
|
|
|
import { LogEntry, logLevel } from 'kafkajs';
|
|
import { createLogger, format, transports } from 'winston';
|
|
import * as Transport from 'winston-transport';
|
|
|
|
const { combine, timestamp, label, printf, splat } = format;
|
|
|
|
const toWinstonLogLevel = (level: logLevel): string => {
|
|
switch(level) {
|
|
case logLevel.ERROR:
|
|
case logLevel.NOTHING:
|
|
return 'error'
|
|
case logLevel.WARN:
|
|
return 'warn'
|
|
case logLevel.INFO:
|
|
return 'info'
|
|
case logLevel.DEBUG:
|
|
return 'debug'
|
|
}
|
|
}
|
|
|
|
const loggerTransports: Array<Transport> = [];
|
|
|
|
if (process.env.NODE_ENV !== 'production' || process.env.DOCKER_MODE === 'true') {
|
|
loggerTransports.push(new transports.Console({
|
|
handleExceptions: true
|
|
}));
|
|
} else {
|
|
const filename = path.join(config.get('logger.path'), config.get('logger.filename'));
|
|
const transport = new (DailyRotateFile)({
|
|
filename: filename,
|
|
datePattern: 'YYYY-MM-DD-HH',
|
|
zippedArchive: true,
|
|
maxSize: '20m',
|
|
maxFiles: '14d',
|
|
handleExceptions: true
|
|
});
|
|
loggerTransports.push(transport);
|
|
}
|
|
|
|
const tbFormat = printf(info => {
|
|
return `${info.timestamp} [${info.label}] ${info.level.toUpperCase()}: ${info.message}`;
|
|
});
|
|
|
|
export function _logger(moduleLabel: string) {
|
|
return createLogger({
|
|
level: config.get('logger.level'),
|
|
format:combine(
|
|
splat(),
|
|
label({ label: moduleLabel }),
|
|
timestamp({format: 'YYYY-MM-DD HH:mm:ss,SSS'}),
|
|
tbFormat
|
|
),
|
|
transports: loggerTransports
|
|
});
|
|
}
|
|
|
|
export function KafkaJsWinstonLogCreator(logLevel: logLevel): (entry: LogEntry) => void {
|
|
const logger = createLogger({
|
|
level: toWinstonLogLevel(logLevel),
|
|
format:combine(
|
|
splat(),
|
|
label({ label: 'kafkajs' }),
|
|
timestamp({format: 'YYYY-MM-DD HH:mm:ss,SSS'}),
|
|
printf(info => {
|
|
var res = `${info.timestamp} [${info.label}] ${info.level.toUpperCase()}: ${info.message}`;
|
|
if (info.extra) {
|
|
res +=`: ${JSON.stringify(info.extra)}`;
|
|
}
|
|
return res;
|
|
}
|
|
)
|
|
),
|
|
transports: loggerTransports
|
|
});
|
|
|
|
return ({ namespace, level, label, log }) => {
|
|
const { message, ...extra } = log;
|
|
logger.log({
|
|
level: toWinstonLogLevel(level),
|
|
message,
|
|
extra,
|
|
});
|
|
}
|
|
}
|