[6a3a178] | 1 |
|
---|
| 2 |
|
---|
| 3 | const debug = require('debug')('log4js:tcp');
|
---|
| 4 | const net = require('net');
|
---|
| 5 |
|
---|
| 6 | function appender(config, layout) {
|
---|
| 7 | let canWrite = false;
|
---|
| 8 | const buffer = [];
|
---|
| 9 | let socket;
|
---|
| 10 | let shutdownAttempts = 3;
|
---|
| 11 | let endMsg = '__LOG4JS__';
|
---|
| 12 |
|
---|
| 13 | function write(loggingEvent) {
|
---|
| 14 | debug('Writing log event to socket');
|
---|
| 15 | canWrite = socket.write(`${layout(loggingEvent)}${endMsg}`, 'utf8');
|
---|
| 16 | }
|
---|
| 17 |
|
---|
| 18 | function emptyBuffer() {
|
---|
| 19 | let evt;
|
---|
| 20 | debug('emptying buffer');
|
---|
| 21 | /* eslint no-cond-assign:0 */
|
---|
| 22 | while ((evt = buffer.shift())) {
|
---|
| 23 | write(evt);
|
---|
| 24 | }
|
---|
| 25 | }
|
---|
| 26 |
|
---|
| 27 | function createSocket() {
|
---|
| 28 | debug(`appender creating socket to ${config.host || 'localhost'}:${config.port || 5000}`);
|
---|
| 29 | endMsg = `${config.endMsg || '__LOG4JS__'}`;
|
---|
| 30 | socket = net.createConnection(config.port || 5000, config.host || 'localhost');
|
---|
| 31 | socket.on('connect', () => {
|
---|
| 32 | debug('socket connected');
|
---|
| 33 | emptyBuffer();
|
---|
| 34 | canWrite = true;
|
---|
| 35 | });
|
---|
| 36 | socket.on('drain', () => {
|
---|
| 37 | debug('drain event received, emptying buffer');
|
---|
| 38 | canWrite = true;
|
---|
| 39 | emptyBuffer();
|
---|
| 40 | });
|
---|
| 41 | socket.on('timeout', socket.end.bind(socket));
|
---|
| 42 | // don't bother listening for 'error', 'close' gets called after that anyway
|
---|
| 43 | socket.on('close', createSocket);
|
---|
| 44 | }
|
---|
| 45 |
|
---|
| 46 | createSocket();
|
---|
| 47 |
|
---|
| 48 | function log(loggingEvent) {
|
---|
| 49 | if (canWrite) {
|
---|
| 50 | write(loggingEvent);
|
---|
| 51 | } else {
|
---|
| 52 | debug('buffering log event because it cannot write at the moment');
|
---|
| 53 | buffer.push(loggingEvent);
|
---|
| 54 | }
|
---|
| 55 | }
|
---|
| 56 |
|
---|
| 57 | log.shutdown = function (cb) {
|
---|
| 58 | debug('shutdown called');
|
---|
| 59 | if (buffer.length && shutdownAttempts) {
|
---|
| 60 | debug('buffer has items, waiting 100ms to empty');
|
---|
| 61 | shutdownAttempts -= 1;
|
---|
| 62 | setTimeout(() => {
|
---|
| 63 | log.shutdown(cb);
|
---|
| 64 | }, 100);
|
---|
| 65 | } else {
|
---|
| 66 | socket.removeAllListeners('close');
|
---|
| 67 | socket.end(cb);
|
---|
| 68 | }
|
---|
| 69 | };
|
---|
| 70 | return log;
|
---|
| 71 | }
|
---|
| 72 |
|
---|
| 73 | function configure(config, layouts) {
|
---|
| 74 | debug(`configure with config = ${config}`);
|
---|
| 75 | let layout = function (loggingEvent) {
|
---|
| 76 | return loggingEvent.serialise();
|
---|
| 77 | };
|
---|
| 78 | if (config.layout) {
|
---|
| 79 | layout = layouts.layout(config.layout.type, config.layout);
|
---|
| 80 | }
|
---|
| 81 | return appender(config, layout);
|
---|
| 82 | }
|
---|
| 83 |
|
---|
| 84 | module.exports.configure = configure;
|
---|