-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserver.js
More file actions
90 lines (70 loc) · 2.54 KB
/
server.js
File metadata and controls
90 lines (70 loc) · 2.54 KB
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
const Http = require('http');
const Redis = require('ioredis');
const Url = require('url');
const Fs = require('fs');
const Docker = require('dockerode');
const Colors = require('colors/safe');
// redis config
const redisConfig = require('./redis-config.json');
const redis = new Redis(redisConfig);
const sub = new Redis(redisConfig);
const pub = new Redis(redisConfig);
// start redis subscribe channel
sub.subscribe('waiting-queue-event', 'processing-queue-event', (err, count) => {
// no event
if(err)
console.log(err+':'+count);
});
// server config
const proxyPort = process.env.ACCEPTED_PORT || 9001;
// http condfig
const defaultHeader = require('./default-http-header.json');
const getFromClient = (request, response) => {
let urlParts = Url.parse(request.url, true);
switch(urlParts.pathname){
case '':
case '/':
response.writeHead(200, defaultHeader);
response.end(JSON.stringify({'status': 'success'}));
break;
case '/add':
if (request.method === 'POST') {
let data = '';
request.on('data', (chunk) => {
data += chunk;
});
request.on('end', () => {
console.log('hello');
const jsonData = JSON.parse(data);
if (jsonData.xid && jsonData.code) {
Fs.writeFileSync('./examples/' + jsonData.xid + '.py', jsonData.code);
// TODO:add error handler
pub.pipeline().publish('waiting-queue-event', JSON.stringify({'event': 'add', 'id': jsonData.xid})).exec();
redis.pipeline().rpush('waitQueue', jsonData.xid).exec();
//response.writeHead(200, defaultHeader);
response.end(JSON.stringify({'status': 'success'}));
} else {
response.writeHead(400, defaultHeader);
response.end(JSON.stringify({'status': 'bad request.'}));
}
});
} else {
response.writeHead(405, defaultHeader);
response.end(JSON.stringify({'status': 'Method Not Allowed.'}));
}
break;
default:
response.writeHead(404, defaultHeader);
response.end(JSON.stringify({'status': 'not found'}));
break;
}
};
// launch proxy server
const server = Http.createServer(getFromClient);
server.listen(proxyPort);
// print server status
console.log('Accepted-Proxy running on ' + proxyPort + ' with node: ' + process.version + ' pid: ' + process.pid + ' ... ' + Colors.green('[OK]'));
console.log('Redis connect ... ' + Colors.green('[OK]'));
console.log('Redis subscribe connect ... ' + Colors.green('[OK]'));
console.log('Redis publish connect ... ' + Colors.green('[OK]'));
console.log('To shutdown, press <CTRL>+C at any time.');