overleaf/libraries/metrics/statsd/http.coffee

os = require("os")
yn = require("yn")

STACKDRIVER_LOGGING = yn(process.env['STACKDRIVER_LOGGING'])

module.exports.monitor = (logger) ->
	return (req, res, next) ->
		Metrics = require("./metrics")
		startTime = process.hrtime()
		end = res.end
		res.end = () ->
			end.apply(this, arguments)
			responseTime = process.hrtime(startTime)
			responseTimeMs = Math.round(responseTime[0] * 1000 + responseTime[1] / 1000)
			if req.route?.path?
				routePath = req.route.path.toString().replace(/\//g, '_').replace(/\:/g, '').slice(1)
				key = "http-requests.#{routePath}.#{req.method}.#{res.statusCode}"
				Metrics.timing(key, responseTimeMs)
				remoteIp = req.ip || req.socket?.socket?.remoteAddress || req.socket?.remoteAddress
				reqUrl = req.originalUrl || req.url
				referrer = req.headers['referer'] || req.headers['referrer']
				if STACKDRIVER_LOGGING
					info =
						httpRequest:
							requestMethod: req.method
							requestUrl: reqUrl
							requestSize: req.headers["content-length"]
							status: res.statusCode
							responseSize: res._headers?["content-length"]
							userAgent: req.headers["user-agent"]
							remoteIp: remoteIp
							referer: referrer
							latency:
								seconds: responseTime[0]
								nanos: responseTime[1]
							protocol: req.protocol
				else
					info =
						req:
							url: reqUrl
							method: req.method
							referrer: referrer
							"remote-addr": remoteIp
							"user-agent": req.headers["user-agent"]
							"content-length": req.headers["content-length"]
						res:
							"content-length": res._headers?["content-length"]
							statusCode: res.statusCode
						"response-time": responseTimeMs
				if res.statusCode >= 500
					logger.error(info, "%s %s", req.method, reqUrl)
				else if res.statusCode >= 400 and res.statusCode < 500
					logger.warn(info, "%s %s", req.method, reqUrl)
				else
					logger.info(info, "%s %s", req.method, reqUrl)
		next()