-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathcontroller.js
70 lines (57 loc) · 1.75 KB
/
controller.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
const Queue = require('bee-queue');
// const path = require('path');
// const fs = require('fs');
// const prettyBytes = require('pretty-bytes');
require("./worker");
const options = {
isWorker: false,
removeOnSuccess: true,
removeOnFailure: true,
redis: {
host: process.env.DB_HOST,
port: process.env.DB_PORT,
password: process.env.DB_PASS,
}
}
const downloadQueue = new Queue('kti_dwl', options);
const compressQueue = new Queue('kti_cmp', options);
const stagingQueue = new Queue('kti_stg', {
...options,
storeJobs: false,
activateDelayedJobs: true
});
const addDownload = (data, id) => {
return downloadQueue.createJob(data).setId(id).save();
}
downloadQueue.on('job succeeded', (jobId, result) => {
console.log(result);
console.log("\x1b[42m\x1b[30m%s\x1b[0m", `Download complete ${jobId}`);
addCompress(result, jobId);
})
downloadQueue.on("job failed", (jobId, err) => {
console.log("Download failed for", jobId, err);
})
const addCompress = (data, id) => {
return compressQueue.createJob(data).setId(id).save();
}
compressQueue.on('job succeeded', (jobId, result) => {
console.log('\x1b[43m\x1b[30m%s\x1b[0m', `Compression complete ${jobId}->${result}`);
addStaging(result, jobId);
})
compressQueue.on("job failed", (jobId, err) => {
console.log("Compression failed for ", jobId, err);
})
const addStaging = (data, id) => {
return stagingQueue.createJob(data).setId(id).delayUntil(Date.now() + 120 * 60 * 1000).save();
}
const shutdown = () => {
console.log("Controller shutting down");
downloadQueue.close(5);
compressQueue.close(5);
stagingQueue.close(5);
}
process.on('SIGINT', shutdown);
process.on('SIGTERM', shutdown);
module.exports = {
addDownload
}