From 9826d125e3e83fc012819c061779cc54c03223c6 Mon Sep 17 00:00:00 2001
From: Henry Oswald <henry.oswald@sharelatex.com>
Date: Thu, 15 Oct 2015 22:22:20 +0100
Subject: [PATCH] added basic health check to docstore

---
 services/docstore/app.coffee | 41 ++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/services/docstore/app.coffee b/services/docstore/app.coffee
index 3a01bead03..3a6b06f7a8 100644
--- a/services/docstore/app.coffee
+++ b/services/docstore/app.coffee
@@ -25,6 +25,47 @@ app.del  '/project/:project_id/doc/:doc_id', HttpController.deleteDoc
 app.post  '/project/:project_id/archive', HttpController.archiveAllDocs
 app.post  '/project/:project_id/unarchive', HttpController.unArchiveAllDocs
 
+
+ObjectId = require("mongojs").ObjectId
+request = require("request")
+async = require("async")
+_ = require("underscore")
+crypto = require("crypto")
+
+app.get "/health_check", (req, res)->
+	doc_id = ObjectId()
+	project_id = ObjectId()
+	url = "http://localhost:#{port}/project/#{project_id}/doc/#{doc_id}"
+	lines = ["smoke test - delete me", "#{crypto.randomBytes(32).toString("hex")}"]
+	logger.log lines:lines, url:url, doc_id:doc_id, project_id:project_id, "running health check"
+	jobs = [
+		(cb)->
+			opts = 
+				url:url
+				json: {lines: lines}
+			request.post(opts, cb)
+		(cb)->
+			request.get {url:url, json:true}, (err, res, body)->
+				if res.statusCode != 200
+					cb("status code not 200, its #{res.statusCode}")
+				else if _.isEqual(body.lines, lines) and body._id == doc_id.toString()
+					cb()
+				else
+					cb("lines not equal ")
+		(cb)->
+			request.del url, cb
+	]
+	async.series jobs, (err)->
+		if err?
+			logger.err err:err, "error running health check"
+			res.send 500
+		else
+			res.send()
+
+
+
+
+
 app.get '/status', (req, res)->
 	res.send('docstore is alive')