Refactor request logger into a class

services/filestore/app.js

@@ -16,6 +16,9 @@ const RequestLogger = require('./app/js/RequestLogger')
 
 const app = express()
 
+const requestLogger = new RequestLogger()
+requestLogger.attach(app)
+
 if (settings.sentry && settings.sentry.dsn) {
   logger.initializeErrorReporting(settings.sentry.dsn)
 }
@@ -26,9 +29,6 @@ if (Metrics.event_loop) {
   Metrics.event_loop.monitor(logger)
 }
 
-app.use(RequestLogger.middleware)
-app.use(RequestLogger.errorHandler)
-
 app.use(function(req, res, next) {
   Metrics.inc('http-request')
   res.logInfo = {}

services/filestore/app/js/FileController.js

@@ -27,8 +27,8 @@ function getFile(req, res, next) {
   }
 
   metrics.inc('getFile')
-  req.setLogMessage('getting file')
-  req.addLogFields({
+  req.requestLogger.setMessage('getting file')
+  req.requestLogger.addFields({
     key,
     bucket,
     format,
@@ -41,7 +41,7 @@ function getFile(req, res, next) {
     if (range) {
       options.start = range.start
       options.end = range.end
-      req.addLogField('range', range)
+      req.requestLogger.addFields({ range })
     }
   }
 
@@ -76,8 +76,8 @@ function getFileHead(req, res, next) {
   const { key, bucket } = req
 
   metrics.inc('getFileSize')
-  req.setLogMessage('getting file size')
-  req.addLogFields({ key, bucket })
+  req.requestLogger.setMessage('getting file size')
+  req.requestLogger.addFields({ key, bucket })
 
   FileHandler.getFileSize(bucket, key, function(err, fileSize) {
     if (err) {
@@ -97,8 +97,8 @@ function insertFile(req, res, next) {
   metrics.inc('insertFile')
   const { key, bucket } = req
 
-  req.setLogMessage('inserting file')
-  req.addLogFields({ key, bucket })
+  req.requestLogger.setMessage('inserting file')
+  req.requestLogger.addFields({ key, bucket })
 
   FileHandler.insertFile(bucket, key, req, function(err) {
     if (err) {
@@ -115,13 +115,13 @@ function copyFile(req, res, next) {
   const oldProjectId = req.body.source.project_id
   const oldFileId = req.body.source.file_id
 
-  req.addLogFields({
+  req.requestLogger.addFields({
     key,
     bucket,
     oldProject_id: oldProjectId,
     oldFile_id: oldFileId
   })
-  req.setLogMessage('copying file')
+  req.requestLogger.setMessage('copying file')
 
   PersistorManager.copyFile(
     bucket,
@@ -146,8 +146,8 @@ function deleteFile(req, res, next) {
   metrics.inc('deleteFile')
   const { key, bucket } = req
 
-  req.addLogFields({ key, bucket })
-  req.setLogMessage('deleting file')
+  req.requestLogger.addFields({ key, bucket })
+  req.requestLogger.setMessage('deleting file')
 
   FileHandler.deleteFile(bucket, key, function(err) {
     if (err) {
@@ -162,8 +162,8 @@ function directorySize(req, res, next) {
   metrics.inc('projectSize')
   const { project_id: projectId, bucket } = req
 
-  req.setLogMessage('getting project size')
-  req.addLogFields({ projectId, bucket })
+  req.requestLogger.setMessage('getting project size')
+  req.requestLogger.addFields({ projectId, bucket })
 
   FileHandler.getDirectorySize(bucket, projectId, function(err, size) {
     if (err) {
@@ -171,7 +171,7 @@ function directorySize(req, res, next) {
     }
 
     res.json({ 'total bytes': size })
-    req.addLogField('size', size)
+    req.requestLogger.addFields({ size })
   })
 }
 

services/filestore/app/js/RequestLogger.js

@@ -1,35 +1,39 @@
 const logger = require('logger-sharelatex')
 const metrics = require('metrics-sharelatex')
 
-module.exports = {
-  errorHandler,
-  middleware
+class RequestLogger {
+  constructor() {
+    this.errorHandler = this.errorHandler.bind(this)
+    this.middleware = this.middleware.bind(this)
+    this._logInfo = {}
+    this._logMessage = 'http request'
   }
 
-function errorHandler(err, req, res, next) {
-  req._logInfo.set('error', err)
+  attach(app) {
+    app.use(this.middleware)
+    app.use(this.errorHandler)
+  }
+
+  errorHandler(err, req, res, next) {
+    this._logInfo.error = err
     res
       .send(err.message)
       .status(500)
       .end()
   }
 
-function middleware(req, res, next) {
-  const startTime = new Date()
+  addFields(fields) {
+    Object.assign(this._logInfo, fields)
+  }
 
-  // methods to allow the setting of additional information to be logged for the request
-  req._logInfo = {}
-  req._logMessage = 'http request'
-  req.addLogField = function(field, value) {
-    req._logInfo[field] = value
-  }
-  req.addLogFields = function(fields) {
-    Object.assign(req._logInfo, fields)
-  }
-  req.setLogMessage = function(message) {
-    req._logMessage = message
+  setMessage(message) {
+    this._logMessage = message
   }
 
+  middleware(req, res, next) {
+    const startTime = new Date()
+    req.requestLogger = this
+
     // override the 'end' method to log and record metrics
     const end = res.end
     res.end = function() {
@@ -73,11 +77,14 @@ function middleware(req, res, next) {
             statusCode: res.statusCode,
             'response-time': responseTime
           },
-        info: req._logInfo
+          info: this._logInfo
         },
-      req._logMessage
+        this._logMessage
       )
     }
 
     next()
   }
+}
+
+module.exports = RequestLogger

services/filestore/test/unit/js/FileControllerTests.js

@@ -73,9 +73,10 @@ describe('FileController', function() {
         file_id: fileId
       },
       headers: {},
-      setLogMessage: sinon.stub(),
-      addLogField: sinon.stub(),
-      addLogFields: sinon.stub()
+      requestLogger: {
+        setMessage: sinon.stub(),
+        addFields: sinon.stub()
+      }
     }
 
     res = {