-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathapi-load-balancer.js
67 lines (48 loc) · 1.66 KB
/
api-load-balancer.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
require('dotenv').config();
var fs = require('fs');
var https = require('https');
var axios = require('axios');
var PRIVATEKEY = process.env.PRIVATEKEY || '<local_da_chave>';
var FULLCHAIN = process.env.FULLCHAIN || '<local_da_chave>';
var CHAIN = process.env.CHAIN || '<local_da_chave>';
var privateKey = fs.readFileSync(PRIVATEKEY, 'utf8');
var certificate = fs.readFileSync(FULLCHAIN, 'utf8');
var chain = fs.readFileSync(CHAIN, 'utf8');
var credentials = {key: privateKey, cert: certificate, ca: chain};
var express = require('express');
var bodyParser = require('body-parser');
var cors = require('cors');
var app = express();
app.use(cors());
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({extended:true}));
const servers = process.env.SERVERS.split(',');
let cur = 0;
app.get('/vmstatus', async (req, res) => {
let msg = '';
for(var i = 0; i < servers.length; ++i) {
msg += servers[i];
try {
const response = await axios.get(servers[i]);
msg += response.status === 200 ? ': OK' : ': DOWN\n';
msg += '<br/>'
} catch(e) {
msg += ': DOWN<br/>';
}
}
res.send(msg);
})
app.post('/apiCompiler', (req, res) => {
const options = {
headers: {
'Content-Type': 'application/json',
},
timeout: 10000,
}
axios.post(`${servers[cur]}/apiCompiler`, req.body, options)
.then((response) => res.send(response.data))
.catch((e) => res.send(e));
cur = (cur + 1) % servers.length;
})
var httpsServer = https.createServer(credentials, app);
httpsServer.listen(process.env.LOAD_BALANCER_PORT, () => console.log(`https on ${process.env.LOAD_BALANCER_PORT}`));