forked from Runnable/compose-test-repo-5.1
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathserver.js
71 lines (66 loc) · 2.07 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
'use strict'
const http = require('http')
const r = require('rethinkdb')
require('rethinkdb-init')(r)
const j = (obj) => JSON.stringify(obj, null, ' ')
const checkIfDBConnectionIsValid = (rethinkdbHost) => {
return r.init({
host: rethinkdbHost,
db: process.env.DB_NAME || 'hello_node_rethinkdb'
}, [
rethinkdbHost,
process.env.TABLE_NAME || 'master'
])
.then(function (conn) {
console.log(`Getting db list for '${rethinkdbHost}' ...`)
return Promise.all([r.dbList().run(conn), r.tableList().run(conn)])
})
.spread(function (dbList, tableList) {
console.log(`DB and table list for ${rethinkdbHost} ...`, dbList, tableList)
return { dbList, tableList }
})
}
http.createServer(function (req, res) {
let opts = {
IS_MIRRORED_DOCKERFILE: process.env.IS_MIRRORED_DOCKERFILE,
RUNNABLE_CONTAINER_ID: process.env.RUNNABLE_CONTAINER_ID,
RETHINKDB_3: process.env.RETHINKDB_3,
RETHINKDB_4: process.env.RETHINKDB_4,
HOSTNAME: process.env.HOSTNAME
}
if (process.env.NAME) {
opts.NAME = process.env.NAME
}
res.writeHead(200, {'Content-Type': 'application/json'})
if (!process.env.RETHINKDB_3 || !process.env.RETHINKDB_4) {
console.log('One of the necessary `RETHINKDB` ENV vars not set')
res.end(j({
message: 'Hello: Not enough RethinkDB Variables set',
opts
}))
}
console.log('Connecting to Rethinkdb...')
return Promise.all([
// Hosts and ENVs set in docker-compose.yml file
checkIfDBConnectionIsValid('rethinkdb1'),
checkIfDBConnectionIsValid('some-weird-host'),
checkIfDBConnectionIsValid(process.env.RETHINKDB_3),
checkIfDBConnectionIsValid(process.env.RETHINKDB_4)
])
.then(hosts => {
res.end(j({
message: 'Hello: Succesfully connected to DB',
opts,
hosts
}))
})
.catch(function (err) {
console.log('Error: ', err)
res.end(j({
message: '!!! Error connecting to DB !!!',
opts,
err
}))
})
}).listen(process.env.PORT || 80)
console.log(`Server running at port ${process.env.PORT}`)