Skip to content
15 changes: 10 additions & 5 deletions lib/logging.js
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@ var https = require("https");
var moment = require("moment");
var uuid = require("uuid");

const accessor = {};

module.exports = function (id, opts) {

opts = Object.assign({
Expand All @@ -24,7 +26,7 @@ module.exports = function (id, opts) {
keepAlive: true
})
},
credentials: configure.credentials
credentials: opts.credentials
});

var configFile = path.resolve(os.tmpdir(), `leolog_${id.toString()}.json`);
Expand Down Expand Up @@ -125,14 +127,15 @@ module.exports = function (id, opts) {
config.sequenceNumber = data.nextSequenceToken;
fs.writeFile(configFile, JSON.stringify(config, null, 2), callback);
}
})
});
logs = [];
}

});
}

var logger = {
sendEvents,
end: function (callback) {
process.stdout.write = oldStdOut;
process.stderr.write = oldStdErr;
Expand All @@ -149,7 +152,7 @@ module.exports = function (id, opts) {
process.once("beforeExit", () => {
logger.end((err) => {
if (err) {
console.log("Error uploading logs to aws:", err)
console.log("Error uploading logs to aws:", err);
}
console.log("Finished uploading logs", config.logGroupName);
});
Expand All @@ -159,6 +162,8 @@ module.exports = function (id, opts) {
// By catching and logging the error it adds to the event loop and causes "beforeExit" to fire
console.error(err);
});

accessor.logger = logger;
return logger;
};
};

module.exports.accessor = accessor;