79 lines
2.2 KiB
JavaScript
79 lines
2.2 KiB
JavaScript
'use strict'
|
|
|
|
const split = require('split2');
|
|
const assingDeep = require('assign-deep');
|
|
const { Transform } = require('stream');
|
|
const DEV_ENVS = ['test', 'development'];
|
|
const logs = new Map();
|
|
|
|
const LEVELS = {
|
|
10: 'trace',
|
|
20: 'debug',
|
|
30: 'info',
|
|
40: 'warn',
|
|
50: 'error',
|
|
60: 'fatal'
|
|
}
|
|
|
|
module.exports = function logCollector () {
|
|
return new Transform({
|
|
transform (chunk, enc, callback) {
|
|
let entry;
|
|
|
|
try {
|
|
entry = JSON.parse(chunk);
|
|
const { level, time } = entry;
|
|
|
|
if (level === undefined && time === undefined) {
|
|
throw new Error('Entry log is not valid');
|
|
}
|
|
} catch (e) {
|
|
if (DEV_ENVS.includes(process.env.NODE_ENV)) {
|
|
this.push(chunk + '\n');
|
|
}
|
|
return callback();
|
|
}
|
|
|
|
const { id, end } = entry;
|
|
|
|
if (id === undefined) {
|
|
entry.level = LEVELS[entry.level];
|
|
this.push(`${JSON.stringify(entry)}\n`);
|
|
return callback();
|
|
}
|
|
|
|
if (end === true) {
|
|
const accEntry = logs.get(id);
|
|
accEntry.level = LEVELS[accEntry.level];
|
|
accEntry.time = entry.time;
|
|
this.push(`${JSON.stringify(accEntry)}\n`);
|
|
logs.delete(id);
|
|
return callback();
|
|
}
|
|
|
|
if (logs.has(id)) {
|
|
const accEntry = logs.get(id);
|
|
|
|
if (accEntry.level > entry.level) {
|
|
delete entry.level
|
|
}
|
|
|
|
let error;
|
|
if (hasProperty(accEntry, 'error') && hasProperty(entry, 'error')) {
|
|
logs.set(id, assingDeep({}, accEntry, entry, { error: accEntry.error.concat(entry.error) }));
|
|
} else {
|
|
logs.set(id, assingDeep({}, accEntry, entry));
|
|
}
|
|
} else {
|
|
logs.set(id, entry);
|
|
}
|
|
|
|
callback();
|
|
}
|
|
})
|
|
}
|
|
|
|
function hasProperty(obj, prop) {
|
|
return Object.prototype.hasOwnProperty.call(obj, prop)
|
|
}
|